QueryGen

クエリジェネ

QueryGen は、自然言語プロンプトと CSV ファイル入力に基づいて SQL クエリを生成できる Streamlit アプリケーションです。 LangChain の SQL クエリ生成機能と ChatGroq 言語モデルを使用して、ユーザー プロンプトを解釈し、アップロードされた CSV データに合わせて最適化された SQL クエリを生成します。

デモ: リンク

特徴

• 複数のCSVファイルをアップロードする
• 自然言語プロンプトを入力して SQL クエリを生成します
• アップロードされた CSV ファイルから SQLite データベースを自動的に作成します
• 生成された SQL クエリとその結果を表示する
• 無効なプロンプトまたはデータベースの問題のエラー処理

インストール

1. リポジトリのクローンを作成します。

 https://github.com/Pragateeshwaran/QueryGen.git

2. 必要な依存関係をインストールします。

 pip install -r requirements.txt

3. ChatGroq から API キーを取得し、 GenAi.py内のプレースホルダーを実際の API キーに置き換えます。

使用法

1. プロジェクト ディレクトリに移動します。

 cd QueryGen

2. Streamlit アプリケーションを実行します。

 streamlit run app.py

3. アプリケーションはデフォルトの Web ブラウザで開きます。

4. サイドバーのファイル アップローダーを使用して、1 つ以上の CSV ファイルをアップロードします。

5. テキスト入力フィールドに自然言語クエリを入力します。

6. 生成された SQL クエリとその結果がページに表示されます。

ファイル構造

• app.py : ユーザー インターフェイスとファイルのアップロードを処理するメインの Streamlit アプリケーション ファイル。
• csv_db.py : アップロードされた CSV ファイルから SQLite データベースを作成するユーティリティ モジュール。
• GenAi.py : このモジュールには、LangChain と ChatGroq を使用して自然言語プロンプトから SQL クエリを生成するための関数が含まれています。