QueryGen

QueryGen

QueryGen é um aplicativo Streamlit que permite gerar consultas SQL com base em prompts de linguagem natural e entradas de arquivos CSV. Ele usa os recursos de geração de consultas SQL do LangChain e o modelo de linguagem ChatGroq para interpretar prompts do usuário e gerar consultas SQL otimizadas adaptadas aos dados CSV carregados.

Demonstração: Link

Características

• Carregar vários arquivos CSV
• Insira prompts em linguagem natural para gerar consultas SQL
• Crie automaticamente um banco de dados SQLite a partir dos arquivos CSV carregados
• Exibir a consulta SQL gerada e seu resultado
• Tratamento de erros para prompts inválidos ou problemas de banco de dados

Instalação

1. Clone o repositório:

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

2. Instale as dependências necessárias:

 pip install -r requirements.txt

3. Obtenha uma chave de API do ChatGroq e substitua os espaços reservados em GenAi.py pela sua chave de API real.

Uso

1. Navegue até o diretório do projeto:

 cd QueryGen

2. Execute o aplicativo Streamlit:

 streamlit run app.py

3. O aplicativo será aberto em seu navegador padrão.

4. Carregue um ou mais arquivos CSV usando o uploader de arquivos na barra lateral.

5. Insira sua consulta em linguagem natural no campo de entrada de texto.

6. A consulta SQL gerada e seu resultado serão exibidos na página.

Estrutura de arquivo

• app.py : o arquivo principal do aplicativo Streamlit que lida com a interface do usuário e uploads de arquivos.
• csv_db.py : um módulo utilitário que cria um banco de dados SQLite a partir dos arquivos CSV carregados.
• GenAi.py : Este módulo contém funções para gerar consultas SQL a partir de prompts em linguagem natural usando LangChain e ChatGroq.