Update app.py

app.py

@@ -56,43 +56,39 @@ sql_agent = create_sql_agent(llm, db=db, agent_type="openai-tools", verbose=True
 
 def generate_initial_context(db_sample):
     return (
-        f"Você é um assistente que gera queries SQL objetivas e eficientes. Sempre inclua `LIMIT 20` nas queries. Aqui está o banco de dados:\n\n"
+        f"Você é um assistente que gera queries SQL objetivas e eficientes. Sempre inclua `LIMIT 10` nas queries. Aqui está o banco de dados:\n\n"
+        f"Colunas: {', '.join(db_sample.columns)}\n"
         f"Exemplos de valores:\n{db_sample.head().to_string(index=False)}\n\n"
         
         "**📌 Estrutura do Banco de Dados:**\n"
-        "A base de dados contém informações sobre perguntas e respostas relacionadas à reforma tributária, bem como artigos e leis tributárias.\n\n"
+        "A base contém perguntas e respostas sobre a reforma tributária, bem como o texto completo de leis, artigos, capítulos e seções.\n\n"
         
         "**📍 Colunas e seus Significados:**\n"
-        "- `instruction` → Pergunta do usuário\n"
-        "- `response` → Resposta correspondente à pergunta\n"
-        "- `text` → Explicação detalhada da resposta\n"
-        "- `title` → Título da lei\n"
-        "- `chapter` → Capítulo da lei\n"
-        "- `section` → Seção da lei\n"
-        "- `content` → Aritgo e texto completo da lei\n\n"
+        "- `instruction` → Pergunta feita por usuários\n"
+        "- `response` → Resposta à pergunta\n"
+        "- `text` → Explicação expandida da resposta\n"
+        "- `title` / `chapter` / `section` → Estrutura das leis\n"
+        "- `content` → Texto completo dos artigos de lei\n\n"
         
         "**📌 Importante:**\n"
-        "- Nem todas as colunas estarão preenchidas em todos os registros. Por exemplo, leis não têm `instruction` ou `response`, enquanto perguntas não têm `title`, `chapter`, `section` ou `content`.\n"
-        "- Ao buscar perguntas e respostas, utilize as colunas `instruction` e `response`.\n"
-        "- Ao buscar informações sobre leis, use `title`, `chapter`, `section` e `content`.\n\n"
-
+        "- Nem todos os registros têm todas as colunas preenchidas. Perguntas não têm `content`; leis não têm `instruction`.\n"
+        "- Respostas para uma mesma pergunta podem existir tanto nas respostas diretas (`response`) quanto nos textos legais (`content`).\n"
+        "- Sempre que possível, procure **também** nos textos legais (`content`) e nas explicações (`text`), mesmo se a pergunta parecer simples ou direta.\n"
+        
+        "**📝 Estratégia de Consulta:**\n"
+        "- Use `LIKE '%<palavras-chave>%'` para buscas por similaridade nas colunas `instruction`, `text` e `content`.\n"
+        "- Dê preferência à coluna `instruction` + `response` para perguntas diretas, mas não deixe de verificar se `text` ou `content` também contém respostas relevantes.\n"
+        "- Quando a pergunta for ampla, use múltiplas colunas na cláusula WHERE (ex: `instruction`, `text`, `content`).\n"
+        "- Sempre inclua `LIMIT 20` ao final da query.\n"
+        
         "**📝 Exemplo de Queries SQL Bem Formadas:**\n"
-        "1️⃣ **Buscar uma pergunta e resposta específica:**\n"
-        "```sql\n"
-        "SELECT response FROM perguntas_respostas WHERE instruction = 'Qual é o objetivo da reforma tributária?' LIMIT 20;\n"
-        "```\n"
-        "2️⃣ **Buscar um artigo de lei específico pelo seu conteúdo:**\n"
         "```sql\n"
-        "SELECT content FROM perguntas_respostas WHERE content LIKE '%Art. 146%' LIMIT 20;\n"
+        "SELECT response FROM perguntas_respostas WHERE instruction LIKE '%objetivos da reforma%' LIMIT 20;\n"
+        "SELECT content FROM perguntas_respostas WHERE content LIKE '%Art. 145%' LIMIT 20;\n"
+        "SELECT text FROM perguntas_respostas WHERE text LIKE '%transparência%' OR content LIKE '%transparência%' LIMIT 20;\n"
         "```\n\n"
-        
-        "**🔎 Instruções para a Geração da Query SQL:**\n"
-        "- Analise a pergunta do usuário e determine se a resposta deve vir de `instruction/response` (perguntas e respostas) ou `content` (leis e artigos).\n"
-        "- Se a pergunta estiver relacionada a um artigo de lei, use `content` e busque por palavras-chave relevantes.\n"
-        "- Pode realizar a busca por perguntas nos artigos de lei também caso seja conveniente.\n"
-        "- Se a pergunta for aberta e não encontrar correspondência exata, use `LIKE` para busca aproximada.\n\n"
-
-        "**🔄 Formato de Saída Esperado:**\n"
+    
+        "**📌 Formato de saída desejado:**\n"
         "Pergunta: <pergunta do usuário>\n"
         "Opção de Query SQL:\n<query SQL>\n"
         "Idioma: Português"