Update app.py

app.py

@@ -1,5 +1,5 @@
 """
-OpenRouter Chatbot – con storico conversazione + gruppi modello
+OpenRouter Chatbot – con storico conversazione + modelli separati
 Run: gradio app.py
 """
 
@@ -17,22 +17,16 @@ OPENROUTER_API_KEY = os.getenv(
 )
 
 # ------------------------------------------------------------------
-# Utility per ottenere l’elenco modelli raggruppati
+# Ottieni modelli raggruppati
 # ------------------------------------------------------------------
 @functools.lru_cache(maxsize=1)
 def fetch_models_grouped() -> dict:
-    """
-    Restituisce i modelli raggruppati tra 'con ragionamento' e 'senza ragionamento'.
-    """
     headers = {"Authorization": f"Bearer {OPENROUTER_API_KEY}"}
     try:
-        resp = requests.get(
-            "https://openrouter.ai/api/v1/models",
-            headers=headers,
-            timeout=15
-        )
+        resp = requests.get("https://openrouter.ai/api/v1/models", headers=headers, timeout=15)
         resp.raise_for_status()
         data = resp.json()
+
         reasoning_models = []
         casual_models = []
 
@@ -44,18 +38,17 @@ def fetch_models_grouped() -> dict:
                 casual_models.append(m["id"])
 
         return {
-            "🧠 Con Ragionamento": sorted(reasoning_models),
-            "⚡️ Generici / Casual": sorted(casual_models)
+            "reasoning": sorted(reasoning_models),
+            "casual": sorted(casual_models)
         }
     except Exception as e:
-        gr.Warning(f"Impossibile ottenere modelli: {e}")
         return {
-            "🧠 Con Ragionamento": ["openai/gpt-4-turbo"],
-            "⚡️ Generici / Casual": []
+            "reasoning": ["openai/gpt-4-turbo"],
+            "casual": []
         }
 
 # ------------------------------------------------------------------
-# Funzione per formattare la cronologia
+# Format cronologia
 # ------------------------------------------------------------------
 def format_history(history: list) -> str:
     output = ""
@@ -67,14 +60,9 @@ def format_history(history: list) -> str:
     return output.strip()
 
 # ------------------------------------------------------------------
-# Funzione di chiamata al modello
+# Chat
 # ------------------------------------------------------------------
-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)
-
+def chat_with_openrouter(prompt: str, selected_model: str, history: list):
     headers = {
         "Authorization": f"Bearer {OPENROUTER_API_KEY}",
         "Content-Type": "application/json"
@@ -83,19 +71,14 @@ def chat_with_openrouter(prompt: str, model: str, history: list):
     history.append({"role": "user", "content": prompt})
 
     payload = {
-        "model": model,
+        "model": selected_model,
         "messages": history,
         "max_tokens": 4096,
         "temperature": 0.7,
     }
 
     try:
-        resp = requests.post(
-            "https://openrouter.ai/api/v1/chat/completions",
-            headers=headers,
-            json=payload,
-            timeout=60
-        )
+        resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=60)
         resp.raise_for_status()
         reply = resp.json()["choices"][0]["message"]["content"]
         history.append({"role": "assistant", "content": reply})
@@ -106,31 +89,26 @@ def chat_with_openrouter(prompt: str, model: str, history: list):
         return history, format_history(history)
 
 # ------------------------------------------------------------------
-# Interfaccia Gradio con dropdown raggruppato
+# Interfaccia con dropdown separati
 # ------------------------------------------------------------------
 def build_interface():
-    models_grouped = fetch_models_grouped()
+    grouped_models = fetch_models_grouped()
 
-    # Crea lista per dropdown con intestazioni simulate
-    grouped_choices = []
-    first_valid_model = None
+    default_model = grouped_models["reasoning"][0] if grouped_models["reasoning"] else ""
 
-    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")
-        gr.Markdown("Usa **qualsiasi modello** di OpenRouter con cronologia visibile.")
+    with gr.Blocks(title="NotExistChatter – Chat con modelli") as demo:
+        gr.Markdown("## 🤖 Project Adam – Chat dinamica con modelli OpenRouter")
+        gr.Markdown("Seleziona un modello da uno dei due gruppi.")
 
         with gr.Row():
-            model_dropdown = gr.Dropdown(
-                choices=grouped_choices,
-                value=first_valid_model,
-                label="Modello",
-                allow_custom_value=False,
+            reasoning_dropdown = gr.Dropdown(
+                choices=grouped_models["reasoning"],
+                label="🧠 Modelli con Ragionamento",
+                interactive=True
+            )
+            casual_dropdown = gr.Dropdown(
+                choices=grouped_models["casual"],
+                label="⚡️ Modelli Generici / Casual",
                 interactive=True
             )
 
@@ -151,19 +129,25 @@ def build_interface():
         )
 
         send_btn = gr.Button("Invia", variant="primary")
-
-        # Stato della cronologia
         chat_history = gr.State([])
 
+        # Seleziona modello valido (da uno dei due)
+        def resolve_model(prompt, reasoning_model, casual_model, history):
+            model = reasoning_model or casual_model
+            if not model:
+                history.append({"role": "assistant", "content": "⚠️ Seleziona almeno un modello."})
+                return history, format_history(history)
+            return chat_with_openrouter(prompt, model, history)
+
         send_btn.click(
-            fn=chat_with_openrouter,
-            inputs=[prompt_box, model_dropdown, chat_history],
+            fn=resolve_model,
+            inputs=[prompt_box, reasoning_dropdown, casual_dropdown, chat_history],
             outputs=[chat_history, output_box]
         )
 
         prompt_box.submit(
-            fn=chat_with_openrouter,
-            inputs=[prompt_box, model_dropdown, chat_history],
+            fn=resolve_model,
+            inputs=[prompt_box, reasoning_dropdown, casual_dropdown, chat_history],
             outputs=[chat_history, output_box]
         )
 
@@ -176,7 +160,7 @@ def build_interface():
     return demo
 
 # ------------------------------------------------------------------
-# Avvio app
+# Avvio
 # ------------------------------------------------------------------
 if __name__ == "__main__":
     build_interface().launch()