Update app.py

app.py

@@ -164,21 +164,16 @@ SHOWCASE = [
 ]
 
 MODELS = {
-    "Groq": {"env":"GROQ_API_KEY","default":"openai/gpt-oss-120b",
-        "list":["openai/gpt-oss-120b","meta-llama/llama-4-scout-17b-16e-instruct",
-                "deepseek-r1-distill-llama-70b","llama-3.3-70b-versatile","qwen-qwq-32b"]},
-    "OpenAI": {"env":"OPENAI_API_KEY","default":"gpt-5.2",
-        "list":["gpt-5.2","gpt-5.2-chat-latest","gpt-5.4","gpt-5.4-pro","gpt-5.2-pro",
-                "gpt-5.2-codex","gpt-5","gpt-5-mini","gpt-4o","gpt-4o-mini","gpt-oss-120b"]},
+    "OpenAI": {"env":"OPENAI_API_KEY","default":"gpt-5.4",
+        "list":["gpt-5.4","gpt-5.4-pro","gpt-5.2","gpt-4o","gpt-4o-mini"]},
     "Anthropic": {"env":"ANTHROPIC_API_KEY","default":"claude-sonnet-4-6",
         "list":["claude-opus-4-6","claude-sonnet-4-6","claude-haiku-4-5-20251001"]},
-    "Google Gemini": {"env":"GOOGLE_API_KEY","default":"gemini-3.1-pro-preview",
-        "list":["gemini-3.1-pro-preview","gemini-3-flash","gemini-2.5-flash","gemini-2.5-pro"]},
+    "Google Gemini": {"env":"GOOGLE_API_KEY","default":"gemini-2.5-pro",
+        "list":["gemini-2.5-pro","gemini-2.5-flash"]},
     "DeepSeek": {"env":"DEEPSEEK_API_KEY","default":"deepseek-chat",
         "list":["deepseek-chat","deepseek-reasoner"]},
-    "xAI (Grok)": {"env":"XAI_API_KEY","default":"grok-4",
-        "list":["grok-4","grok-4-fast-reasoning","grok-4-1-fast-reasoning","grok-3-beta"]},
-    "Friendli": {"env":"FRIENDLI_TOKEN","default":"deppfs281rgffnk","list":["deppfs281rgffnk"]},
+    "xAI (Grok)": {"env":"XAI_API_KEY","default":"grok-3-beta",
+        "list":["grok-3-beta"]},
     "Ollama (Local)": {"env":"","default":"llama3.1",
         "list":["llama3.1","llama3.1:70b","qwen3.5:32b","deepseek-r1:8b","phi4:14b"]},
     "Custom (OpenAI-compatible)": {"env":"","default":"custom","list":["custom"]},
@@ -493,10 +488,7 @@ def _build(backend, api_key, model, base_url):
     k = api_key or (os.getenv(ek,"") if ek else "")
 
     try:
-        if backend == "Groq":
-            if not k: return None, "❌ GROQ_API_KEY required"
-            return Marl.from_openai_compatible("https://api.groq.com/openai/v1", k, model, cfg), "✅"
-        elif backend == "OpenAI":
+        if backend == "OpenAI":
             if not k: return None, "❌ OPENAI_API_KEY required"
             return Marl.from_openai(k, model, cfg), "✅"
         elif backend == "Anthropic":
@@ -512,9 +504,6 @@ def _build(backend, api_key, model, base_url):
         elif backend == "xAI (Grok)":
             if not k: return None, "❌ XAI_API_KEY required"
             return Marl.from_openai_compatible("https://api.x.ai/v1", k, model, cfg), "✅"
-        elif backend == "Friendli":
-            if not k: return None, "❌ FRIENDLI_TOKEN required"
-            return Marl.from_friendli(k, model, cfg), "✅"
         elif backend == "Ollama (Local)":
             return Marl.from_ollama(model, base_url or "http://localhost:11434", cfg), "✅"
         elif backend == "Custom (OpenAI-compatible)":
@@ -588,7 +577,7 @@ def run_marl_only(prompt, backend, api_key, model, base_url, budget, mode_sel, e
 # ════════════════════════════════════════════════════════════════
 
 def create_app():
-    init_m = MODELS["Groq"]["list"]
+    init_m = MODELS["OpenAI"]["list"]
 
     with gr.Blocks(title="MARL — Model-Agnostic Runtime Middleware") as app:
         gr.HTML(INDEX_HTML)
@@ -596,11 +585,11 @@ def create_app():
         with gr.Tabs():
           with gr.Tab("⚡ Playground"):
             with gr.Row():
-                backend = gr.Dropdown(label="Backend", choices=BACKEND_LIST, value="Groq", scale=2)
-                api_key = gr.Textbox(label="API Key", type="password", placeholder="ENV: GROQ_API_KEY",
-                                     value=os.getenv("GROQ_API_KEY",""), scale=3)
+                backend = gr.Dropdown(label="Backend", choices=BACKEND_LIST, value="OpenAI", scale=2)
+                api_key = gr.Textbox(label="API Key", type="password", placeholder="Enter your API key (required)",
+                                     value=os.getenv("OPENAI_API_KEY",""), scale=3)
             with gr.Row():
-                model = gr.Dropdown(label="Model", choices=init_m, value="openai/gpt-oss-120b",
+                model = gr.Dropdown(label="Model", choices=init_m, value="gpt-5.4",
                                     allow_custom_value=True, scale=3)
                 base_url = gr.Textbox(label="Base URL (Custom/Ollama)",
                                        placeholder="http://localhost:11434", scale=2)
@@ -770,7 +759,7 @@ ml = Marl.from_anthropic(<span style="color:#d97706">"sk-ant-..."</span>)
 ml = Marl.from_ollama(<span style="color:#d97706">"llama3.1"</span>)
 
 <span style="color:#94a3b8"># Any OpenAI-compatible</span>
-ml = Marl.from_openai_compatible(<span style="color:#d97706">"https://api.groq.com/openai/v1"</span>, <span style="color:#d97706">"key"</span>, <span style="color:#d97706">"gpt-oss-120b"</span>)</code></pre>
+ml = Marl.from_openai(<span style="color:#d97706">"sk-..."</span>, <span style="color:#d97706">"gpt-5.4"</span>)</code></pre>
 </div>
 
 <div style="background:var(--surface,#fff);border:1px solid #e2e5f0;border-radius:16px;padding:20px;margin-bottom:16px;box-shadow:0 1px 3px rgba(15,23,42,.04)">
@@ -811,12 +800,11 @@ ml = Marl.from_openai_compatible(<span style="color:#d97706">"https://api.groq.c
 <div style="overflow-x:auto">
 <table style="width:100%;border-collapse:separate;border-spacing:0;font-size:11px;border-radius:10px;overflow:hidden;border:1px solid #e2e5f0">
 <tr style="background:#f5f6fa"><th style="text-align:left;padding:8px 10px;font-family:JetBrains Mono,monospace;font-size:8px;color:#94a3b8;text-transform:uppercase;letter-spacing:.5px">Backend</th><th style="text-align:left;padding:8px 10px;font-family:JetBrains Mono,monospace;font-size:8px;color:#94a3b8;text-transform:uppercase;letter-spacing:.5px">Models</th></tr>
-<tr style="background:rgba(99,102,241,.06)"><td style="padding:8px 10px;border-top:1px solid #e2e5f0;font-weight:700;color:#6366f1">⭐ Groq (Default)</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0;font-weight:600">gpt-oss-120b, Llama 4, DeepSeek-R1, QwQ-32b</td></tr>
-<tr><td style="padding:8px 10px;border-top:1px solid #e2e5f0">OpenAI</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0">GPT-5.2, GPT-4.1, o4-mini</td></tr>
-<tr style="background:#fafbfe"><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Anthropic</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Claude Opus 4, Sonnet 4</td></tr>
+<tr style="background:rgba(99,102,241,.06)"><td style="padding:8px 10px;border-top:1px solid #e2e5f0;font-weight:700;color:#6366f1">⭐ OpenAI (Default)</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0;font-weight:600">GPT-5.4, GPT-5.4-pro, GPT-5.2, GPT-4o</td></tr>
+<tr style="background:#fafbfe"><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Anthropic</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Claude Opus 4.6, Sonnet 4.6, Haiku 4.5</td></tr>
 <tr><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Google Gemini</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Gemini 2.5 Pro / Flash</td></tr>
 <tr style="background:#fafbfe"><td style="padding:8px 10px;border-top:1px solid #e2e5f0">DeepSeek</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0">V3 / R1</td></tr>
-<tr><td style="padding:8px 10px;border-top:1px solid #e2e5f0">xAI</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Grok-3 / 4</td></tr>
+<tr><td style="padding:8px 10px;border-top:1px solid #e2e5f0">xAI</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Grok-3</td></tr>
 <tr style="background:#fafbfe"><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Ollama</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Llama, Mistral, Phi, Qwen</td></tr>
 <tr><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Custom</td><td style="padding:8px 10px;border-top:1px solid #e2e5f0">Any OpenAI-compatible endpoint</td></tr>
 </table>