QueryGen

查询生成器

QueryGen 是一个 Streamlit 应用程序，允许您根据自然语言提示和 CSV 文件输入生成 SQL 查询。它使用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. 该应用程序将在您的默认网络浏览器中打开。

4. 使用侧边栏中的文件上传器上传一个或多个 CSV 文件。

5. 在文本输入字段中输入您的自然语言查询。

6. 生成的 SQL 查询及其结果将显示在页面上。

文件结构

• app.py ：处理用户界面和文件上传的主 Streamlit 应用程序文件。
• csv_db.py ：一个实用程序模块，用于从上传的 CSV 文件创建 SQLite 数据库。
• GenAi.py ：该模块包含使用 LangChain 和 ChatGroq 根据自然语言提示生成 SQL 查询的函数。