Update app.py

app.py

@@ -84,7 +84,6 @@ def handle_csv_upload(file):
         db = SQLDatabase(engine=engine)
         logging.info("[UPLOAD] Novo banco carregado e DB atualizado.")
 
-        # Recria o agent com o novo DB
         sql_agent = create_sql_agent(
             ChatOpenAI(model="gpt-4o-mini", temperature=0),
             db=db,
@@ -95,7 +94,12 @@ def handle_csv_upload(file):
         )
         
         logging.info("[UPLOAD] Novo banco carregado e agente recriado. Cache limpo.")
+        query_cache.clear()
+        history_log.clear()
+        recent_history.clear()
+
         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}"
@@ -118,19 +122,21 @@ def reset_app():
         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
+    return gr.File.update(value=path, visible=True)
 
 def export_history_csv():
     path = "history_log.csv"
-    pd.DataFrame(history_log).to_csv(path, index=False)
-    return path
+    pd.DataFrame(history_log).to_csv(path, index=False, sep=";")
+    return gr.File.update(value=path, visible=True)
 
 def generate_initial_context(db_sample):
     return (
@@ -241,12 +247,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=1):
             gr.Markdown("## ⚙️ Configurações")
             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"])
+            csv_file = gr.File(label="📂 Enviar novo CSV", file_types=[".csv"], file_types_strict=True)
             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()
+            download_output = gr.File(visible=False)
 
         with gr.Column(scale=4):
             gr.Markdown("# 🧠 Anomalia Agent")
@@ -265,8 +271,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             btn.click(respond, [msg, chatbot, model_selector], [msg, chatbot])
             history_btn.click(toggle_history, outputs=history_output)
 
-        csv_file.change(handle_csv_upload, inputs=csv_file, outputs=upload_feedback)
-        reset_btn.click(reset_app, outputs=upload_feedback)
+        csv_file.change(lambda f: (handle_csv_upload(f), gr.update(value=None), []), inputs=csv_file, outputs=[upload_feedback, csv_file, chatbot])
+        reset_btn.click(lambda: (reset_app(), gr.update(value=None), []), outputs=[upload_feedback, csv_file, chatbot])
         export_json.click(export_history_json, outputs=download_output)
         export_csv.click(export_history_csv, outputs=download_output)