Berack96
f0193b94bb
- 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/
80 lines
3.1 KiB
Python
80 lines
3.1 KiB
Python
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)
|