QueryGen

QueryGen

QueryGen — это приложение Streamlit, которое позволяет генерировать SQL-запросы на основе подсказок на естественном языке и входных данных из файла CSV. Он использует возможности генерации SQL-запросов LangChain и языковую модель ChatGroq для интерпретации запросов пользователя и создания оптимизированных SQL-запросов, адаптированных к загруженным данным CSV.

Демо: Ссылка

Функции

• Загрузить несколько файлов CSV
• Вводите подсказки на естественном языке для генерации SQL-запросов.
• Автоматически создавать базу данных SQLite из загруженных файлов CSV.
• Отображение сгенерированного SQL-запроса и его результата.
• Обработка ошибок неверных запросов или проблем с базой данных

Установка

1. Клонируем репозиторий:

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

2. Установите необходимые зависимости:

 pip install -r requirements.txt

3. Получите ключ API от ChatGroq и замените заполнители в GenAi.py своим фактическим ключом API.

Использование

1. Перейдите в каталог проекта:

 cd QueryGen

2. Запустите приложение Streamlit:

 streamlit run app.py

3. Приложение откроется в веб-браузере по умолчанию.

4. Загрузите один или несколько файлов CSV с помощью инструмента загрузки файлов на боковой панели.

5. Введите запрос на естественном языке в поле ввода текста.

6. Сгенерированный SQL-запрос и его результат будут отображены на странице.

Структура файла

• app.py : основной файл приложения Streamlit, который управляет пользовательским интерфейсом и загрузкой файлов.
• csv_db.py : служебный модуль, который создает базу данных SQLite из загруженных файлов CSV.
• GenAi.py : этот модуль содержит функции для генерации SQL-запросов из подсказок на естественном языке с использованием LangChain и ChatGroq.