Update app.py

app.py

@@ -1,5 +1,5 @@
 """
-OpenRouter Chatbot – con storico conversazione
+OpenRouter Chatbot – con storico conversazione + gruppi modello
 Run: gradio app.py
 """
 
@@ -17,12 +17,12 @@ OPENROUTER_API_KEY = os.getenv(
 )
 
 # ------------------------------------------------------------------
-# Utility per ottenere l’elenco modelli
+# Utility per ottenere l’elenco modelli raggruppati
 # ------------------------------------------------------------------
 @functools.lru_cache(maxsize=1)
-def fetch_models() -> list[str]:
+def fetch_models_grouped() -> dict:
     """
-    Restituisce la lista completa di modelli offerti da OpenRouter.
+    Restituisce i modelli raggruppati tra 'con ragionamento' e 'senza ragionamento'.
     """
     headers = {"Authorization": f"Bearer {OPENROUTER_API_KEY}"}
     try:
@@ -33,11 +33,26 @@ def fetch_models() -> list[str]:
         )
         resp.raise_for_status()
         data = resp.json()
-        models = sorted(m["id"] for m in data["data"])
-        return models
+        reasoning_models = []
+        casual_models = []
+
+        for m in data["data"]:
+            model_id = m["id"].lower()
+            if any(key in model_id for key in ["gpt", "claude", "llama", "mistral", "mixtral", "gemini", "command-r", "qwen"]):
+                reasoning_models.append(m["id"])
+            else:
+                casual_models.append(m["id"])
+
+        return {
+            "🧠 Con Ragionamento": sorted(reasoning_models),
+            "⚡️ Generici / Casual": sorted(casual_models)
+        }
     except Exception as e:
-        gr.Warning(f"Impossibile scaricare l’elenco modelli: {e}")
-        return ["openai/gpt-4-turbo"]  # fallback statico
+        gr.Warning(f"Impossibile ottenere modelli: {e}")
+        return {
+            "🧠 Con Ragionamento": ["openai/gpt-4-turbo"],
+            "⚡️ Generici / Casual": []
+        }
 
 # ------------------------------------------------------------------
 # Funzione per formattare la cronologia
@@ -55,6 +70,11 @@ def format_history(history: list) -> str:
 # Funzione di chiamata al modello
 # ------------------------------------------------------------------
 def chat_with_openrouter(prompt: str, model: str, history: list):
+    if model.startswith("🔷"):
+        msg = "⚠️ Seleziona un **modello valido**, non un'intestazione di gruppo."
+        history.append({"role": "assistant", "content": msg})
+        return history, format_history(history)
+
     headers = {
         "Authorization": f"Bearer {OPENROUTER_API_KEY}",
         "Content-Type": "application/json"
@@ -86,10 +106,20 @@ def chat_with_openrouter(prompt: str, model: str, history: list):
         return history, format_history(history)
 
 # ------------------------------------------------------------------
-# Interfaccia Gradio con memoria
+# Interfaccia Gradio con dropdown raggruppato
 # ------------------------------------------------------------------
 def build_interface():
-    models = fetch_models()
+    models_grouped = fetch_models_grouped()
+
+    # Crea lista per dropdown con intestazioni simulate
+    grouped_choices = []
+    first_valid_model = None
+
+    for group_name, group_models in models_grouped.items():
+        grouped_choices.append(f"🔷 {group_name}")
+        grouped_choices.extend(group_models)
+        if not first_valid_model and group_models:
+            first_valid_model = group_models[0]
 
     with gr.Blocks(title="NotExistChatter – Chat con memoria") as demo:
         gr.Markdown("## 🤖 project Adam – Chat dinamica")
@@ -97,8 +127,8 @@ def build_interface():
 
         with gr.Row():
             model_dropdown = gr.Dropdown(
-                choices=models,
-                value=models[0] if models else None,
+                choices=grouped_choices,
+                value=first_valid_model,
                 label="Modello",
                 allow_custom_value=False,
                 interactive=True