Refactor and update structure (#20)

* Aggiorna gli agenti e il modello del team per utilizzare OLLAMA_QWEN_1B * Riorganizza e rinomina funzioni di estrazione in moduli di mercato e notizie; migliora la gestione delle importazioni * Spostato main nel corretto file __main__ e aggiornato il README.md * Aggiunta cartella per i modelli, agenti e team * Aggiornata la posizione delle istruzioni * Rimossi TODO e Aggiunto documentazione per metodi aggregated * Aggiornate le istruzioni del coordinatore del team * utils type checks * Rinominato BaseWrapper in MarketWrapper e fix type check markets * fix type checks di notizie e social. * Aggiunti type hints finali * Riorganizzati gli import * Refactoring architetturale e spostamento classi base - Eliminazione del file __init__.py obsoleto che importava ChatManager e Pipeline - Spostamento della classe Pipeline in agents/pipeline.py - Spostamento della classe ChatManager in utils/chat_manager.py - Aggiornamento di __main__.py per importare da app.utils e app.agents, e modifica della logica per utilizzare Pipeline invece di chat per la selezione di provider e stile - Creazione della cartella base con classi base comuni: markets.py (ProductInfo, Price, MarketWrapper), news.py (Article, NewsWrapper), social.py (SocialPost, SocialComment, SocialWrapper) - Aggiornamento di tutti gli import nel progetto (markets/, news/, social/, utils/, tests/) per utilizzare la nuova struttura base/ * Aggiornato Readme * Corretto il valore predefinito della valuta in BinanceWrapper da "USDT" a "USD" * fix type in tests * fix type per models * Rinominato 'quote_currency' in 'currency' e aggiornato il trattamento del timestamp in Price * fix errors found by Copilot * WrapperHandler: semplificata la logica di chiamata delle funzioni sui wrapper * fix docs * fix demos, semplificata logica lista ollama
2025-10-08 16:21:09 +02:00
parent 85153c405b
commit 517842c834
42 changed files with 696 additions and 644 deletions
--- a/src/app/main.py
+++ b/src/app/main.py
@@ -0,0 +1,79 @@
+import gradio as gr
+from dotenv import load_dotenv
+from agno.utils.log import log_info #type: ignore
+from app.utils import ChatManager
+from app.agents import Pipeline
+
+
+if __name__ == "__main__":
+    # Inizializzazioni
+    load_dotenv()
+    pipeline = Pipeline()
+    chat = ChatManager()
+
+    ########################################
+    # Funzioni Gradio
+    ########################################
+    def respond(message: str, history: list[dict[str, str]]) -> tuple[list[dict[str, str]], list[dict[str, str]], str]:
+        chat.send_message(message)
+        response = pipeline.interact(message)
+        chat.receive_message(response)
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": response})
+        return history, history, ""
+
+    def save_current_chat() -> str:
+        chat.save_chat("chat.json")
+        return "💾 Chat salvata in chat.json"
+
+    def load_previous_chat() -> tuple[list[dict[str, str]], list[dict[str, str]]]:
+        chat.load_chat("chat.json")
+        history: list[dict[str, str]] = []
+        for m in chat.get_history():
+            history.append({"role": m["role"], "content": m["content"]})
+        return history, history
+
+    def reset_chat() -> tuple[list[dict[str, str]], list[dict[str, str]]]:
+        chat.reset_chat()
+        return [], []
+
+    ########################################
+    # Interfaccia Gradio
+    ########################################
+    with gr.Blocks() as demo:
+        gr.Markdown("# 🤖 Agente di Analisi e Consulenza Crypto (Chat)")
+
+        # Dropdown provider e stile
+        with gr.Row():
+            provider = gr.Dropdown(
+                choices=pipeline.list_providers(),
+                type="index",
+                label="Modello da usare"
+            )
+            provider.change(fn=pipeline.choose_predictor, inputs=provider, outputs=None)
+
+            style = gr.Dropdown(
+                choices=pipeline.list_styles(),
+                type="index",
+                label="Stile di investimento"
+            )
+            style.change(fn=pipeline.choose_style, inputs=style, outputs=None)
+
+        chatbot = gr.Chatbot(label="Conversazione", height=500, type="messages")
+        msg = gr.Textbox(label="Scrivi la tua richiesta", placeholder="Es: Quali sono le crypto interessanti oggi?")
+
+        with gr.Row():
+            clear_btn = gr.Button("🗑️ Reset Chat")
+            save_btn = gr.Button("💾 Salva Chat")
+            load_btn = gr.Button("📂 Carica Chat")
+
+        # Eventi e interazioni
+        msg.submit(respond, inputs=[msg, chatbot], outputs=[chatbot, chatbot, msg])
+        clear_btn.click(reset_chat, inputs=None, outputs=[chatbot, chatbot])
+        save_btn.click(save_current_chat, inputs=None, outputs=None)
+        load_btn.click(load_previous_chat, inputs=None, outputs=[chatbot, chatbot])
+
+    server, port = ("0.0.0.0", 8000) # 0.0.0.0 per accesso esterno (Docker)
+    server_log = "localhost" if server == "0.0.0.0" else server
+    log_info(f"Starting UPO AppAI Chat on http://{server_log}:{port}") # noqa
+    demo.launch(server_name=server, server_port=port, quiet=True)