Update app.py

app.py

@@ -11,14 +11,6 @@ client = OpenAI(
     api_key=os.environ.get("OPENROUTER_API_KEY"),
 )
 
-FREE_MODELS = [
-    "meta-llama/llama-3.3-70b-instruct:free",
-    "google/gemma-3-27b-it:free",
-    "qwen/qwen3-32b:free",
-    "deepseek/deepseek-r1:free",
-    "mistralai/mistral-small-3.1-24b-instruct:free",
-]
-
 def clean_response(text: str) -> str:
     lines = []
     for line in text.splitlines():
@@ -31,50 +23,42 @@ def clean_response(text: str) -> str:
 def chat(user_message: str, history: list):
     history = history or []
 
-    # Build messages for the API
     messages = [{"role": "system", "content": "You are a helpful assistant."}]
     for msg in history:
         messages.append({"role": msg["role"], "content": msg["content"]})
     messages.append({"role": "user", "content": user_message})
 
-    last_error = None
-
-    for model in FREE_MODELS:
-        try:
-            stream = client.chat.completions.create(
-                model=model,
-                messages=messages,
-                stream=True,
-                temperature=0.7,
-                max_tokens=4096,
-            )
-
-            response_text = ""
-            for chunk in stream:
-                if chunk.choices[0].delta.content:
-                    response_text += chunk.choices[0].delta.content
+    try:
+        # openrouter/free automatically picks a working free model
+        stream = client.chat.completions.create(
+            model="openrouter/free",
+            messages=messages,
+            stream=True,
+            temperature=0.7,
+            max_tokens=4096,
+        )
 
-            response_text = clean_response(response_text)
+        response_text = ""
+        for chunk in stream:
+            if chunk.choices[0].delta.content:
+                response_text += chunk.choices[0].delta.content
 
-            # Append in messages format
-            history.append({"role": "user", "content": user_message})
-            history.append({"role": "assistant", "content": response_text})
-            return history
+        response_text = clean_response(response_text)
 
-        except Exception as e:
-            last_error = str(e)
-            continue
+        history.append({"role": "user", "content": user_message})
+        history.append({"role": "assistant", "content": response_text})
+        return history
 
-    # All models failed
-    history.append({"role": "user", "content": user_message})
-    history.append({
-        "role": "assistant",
-        "content": f"All free models failed.\nLast error: {last_error}"
-    })
-    return history
+    except Exception as e:
+        history.append({"role": "user", "content": user_message})
+        history.append({
+            "role": "assistant",
+            "content": f"Error: {str(e)}"
+        })
+        return history
 
 with gr.Blocks(title="OpenRouter Free Chat") as demo:
-    gr.Markdown("## OpenRouter Free Chat (Auto Model Switching)")
+    gr.Markdown("## OpenRouter Free Chat (Auto Free Model)")
     
     chatbot = gr.Chatbot(
         label="Conversation",