Update app.py

app.py

@@ -94,10 +94,11 @@ def handle_csv_upload(file):
             return_intermediate_steps=True
         )
         
-        logging.info("[UPLOAD] Novo agente SQL recriado com sucesso.")
-        
+        logging.info("[UPLOAD] Novo banco carregado e agente recriado. Cache limpo.")
+        return "✅ CSV carregado com sucesso!"
     except Exception as e:
         logging.error(f"[ERRO] Falha ao processar novo CSV: {e}")
+        return f"❌ Erro ao processar CSV: {e}"
 
 # === Inicialização ===
 engine = create_engine_and_load_db(get_active_csv_path(), SQL_DB_PATH)
@@ -105,9 +106,35 @@ db = SQLDatabase(engine=engine)
 llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
 sql_agent = create_sql_agent(llm, db=db, agent_type="openai-tools", verbose=True, max_iterations=40, return_intermediate_steps=True)
 
+def reset_app():
+    global engine, db, sql_agent, query_cache, history_log, recent_history
+    try:
+        if os.path.exists(UPLOADED_CSV_PATH):
+            os.remove(UPLOADED_CSV_PATH)
+            logging.info("[RESET] CSV personalizado removido.")
+        engine = create_engine_and_load_db(DEFAULT_CSV_PATH, SQL_DB_PATH)
+        db = SQLDatabase(engine=engine)
+        sql_agent = create_sql_agent(ChatOpenAI(model="gpt-4o-mini", temperature=0), db=db, agent_type="openai-tools")
+        query_cache.clear()
+        history_log.clear()
+        recent_history.clear()
+        return "🔄 Sistema resetado para o estado inicial."
+    except Exception as e:
+        return f"❌ Erro ao resetar: {e}"
+
+def export_history_json():
+    path = "history_log.json"
+    pd.DataFrame(history_log).to_json(path, orient="records", indent=2)
+    return path
+
+def export_history_csv():
+    path = "history_log.csv"
+    pd.DataFrame(history_log).to_csv(path, index=False)
+    return path
+
 def generate_initial_context(db_sample):
     return (
-        f"Você é um assistente que gera queries SQL objetivas e eficientes. Sempre inclua LIMIT 15 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"Exemplos do banco de dados:\n{db_sample.head().to_string(index=False)}\n\n"
         "\n***IMPORTANTE***: Detecte automaticamente o idioma da pergunta do usuário e responda sempre no mesmo idioma.\n"
         "Essa base de dados representa o sellout de 2025, janeiro, fevereiro e março até dia 11, de uma farmácia.\n"
@@ -213,12 +240,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("## ⚙️ Configurações")
-            model_selector = gr.Dropdown(
-                choices=list(LLAMA_MODELS.keys()),
-                label="Escolha o Modelo LLM para gerar a query SQL",
-                value="LLaMA 70B"
-            )
+            model_selector = gr.Dropdown(list(LLAMA_MODELS.keys()), label="Modelo LLM", value="LLaMA 70B")
             csv_file = gr.File(label="📂 Enviar novo CSV", file_types=[".csv"])
+            upload_feedback = gr.Markdown()
+            reset_btn = gr.Button("🔄 Resetar")
+            export_json = gr.Button("📤 Exportar JSON")
+            export_csv = gr.Button("📤 Exportar CSV")
+            download_output = gr.File()
 
         with gr.Column(scale=4):
             gr.Markdown("# 🧠 Anomalia Agent")
@@ -226,19 +254,21 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             msg = gr.Textbox(placeholder="Digite sua pergunta aqui...", label=" ", lines=1)
             btn = gr.Button("Enviar", variant="primary")
             history_btn = gr.Button("Histórico", variant="secondary")
+            history_output = gr.JSON()
 
-            def respond(message, chat_history, selected_model_name):
-                response = chatbot_response(message, selected_model_name)
+            def respond(message, chat_history, selected_model):
+                response = chatbot_response(message, selected_model)
                 chat_history.append((message, response))
                 return "", chat_history
 
             msg.submit(respond, [msg, chatbot, model_selector], [msg, chatbot])
             btn.click(respond, [msg, chatbot, model_selector], [msg, chatbot])
+            history_btn.click(toggle_history, outputs=history_output)
 
-            history_output = gr.JSON()
-            history_btn.click(toggle_history, inputs=[], outputs=history_output)
-    
-        csv_file.change(handle_csv_upload, inputs=csv_file, outputs=csv_file)
+        csv_file.change(handle_csv_upload, inputs=csv_file, outputs=upload_feedback)
+        reset_btn.click(reset_app, outputs=upload_feedback)
+        export_json.click(export_history_json, outputs=download_output)
+        export_csv.click(export_history_csv, outputs=download_output)
 
 if __name__ == "__main__":
     demo.launch(share=False)