Update app.py

app.py

@@ -1,6 +1,6 @@
 """
-🧬 Gemma 4 Playground — Demo Space
-Dual model (31B / 26B-A4B) · ZeroGPU · Vision · Thinking Mode
+Darwin-31B-Opus — Demo Space
+Diagnostic-Guided Evolutionary Merge · Gemma 4 31B · Thinking Mode · Vision
 """
 import sys
 print(f"[BOOT] Python {sys.version}", flush=True)
@@ -42,27 +42,28 @@ from transformers.generation.streamers import TextIteratorStreamer
 
 
 # ══════════════════════════════════════════════════════════════════════════════
-# 1.  MODEL CONFIG — Gemma 4 Dual Model
+# 1.  MODEL CONFIG — Darwin-31B-Opus (Single Model)
 # ══════════════════════════════════════════════════════════════════════════════
+MODEL_ID = "FINAL-Bench/Darwin-31B-Opus"
+MODEL_NAME = "Darwin-31B-Opus"
+MODEL_DESC = "Gemma 4 31B + Claude Opus Distill — Darwin V6 diagnostic-guided evolutionary merge"
+MODEL_INFO = {
+    "arch": "Dense", "total": "30.7B", "active": "30.7B",
+    "ctx": "256K", "vision": True,
+}
+
 MODELS = {
-    "Gemma-4-31B-it": {
-        "id": "google/gemma-4-31b-it",
-        "arch": "Dense", "total": "30.7B", "active": "30.7B",
-        "ctx": "256K", "vision": True, "audio": False,
-        "desc": "Dense 31B — 최고 품질, AIME 89.2%, Codeforces 2150",
-    },
-    "Gemma-4-26B-A4B-it": {
-        "id": "google/gemma-4-26B-A4B-it",
-        "arch": "MoE", "total": "25.2B", "active": "3.8B",
-        "ctx": "256K", "vision": True, "audio": False,
-        "desc": "MoE 26B (3.8B active) — 31B의 95% 성능, 추론 ~8배 빠름",
+    MODEL_NAME: {
+        "id": MODEL_ID,
+        **MODEL_INFO,
+        "desc": MODEL_DESC,
     },
 }
 
-DEFAULT_MODEL = "Gemma-4-26B-A4B-it"  # MoE가 ZeroGPU에서 더 적합
+DEFAULT_MODEL = MODEL_NAME
 
 PRESETS = {
-    "general":   "You are Gemma 4, a highly capable multimodal AI assistant by Google DeepMind. Think step by step for complex questions.",
+    "general":   "You are Darwin-31B-Opus, a reasoning-enhanced AI assistant created by VIDRAFT using diagnostic-guided evolutionary merge. Think step by step for complex questions.",
     "code":      "You are an expert software engineer. Write clean, efficient, well-commented code. Explain your approach before writing. Use modern best practices.",
     "math":      "You are a world-class mathematician. Break problems step-by-step. Show full working. Use LaTeX where helpful.",
     "creative":  "You are a brilliant creative writer. Be imaginative, vivid, and engaging. Adapt tone and style to the request.",
@@ -317,33 +318,28 @@ footer { display: none !important; }
 
 def _model_info_html(name):
     m = MODELS.get(name, MODELS[DEFAULT_MODEL])
-    icon = "⚡" if m["arch"] == "MoE" else "🏆"
     return (
         f'<div class="model-box">'
-        f'<b>{icon} {name}</b> '
-        f'<span style="font-size:9px;padding:2px 6px;border-radius:6px;background:rgba(109,40,217,.08);color:#6d28d9;font-weight:700">{m["arch"]}</span><br>'
+        f'<b>🧬 {name}</b> '
+        f'<span style="font-size:9px;padding:2px 6px;border-radius:6px;background:rgba(109,40,217,.08);color:#6d28d9;font-weight:700">{m["arch"]} {m["active"]}</span><br>'
         f'<div class="st">{m["active"]} active / {m["total"]} total · 👁️ Vision · {m["ctx"]} context</div>'
         f'<div class="st">{m["desc"]}</div>'
         f'<div class="st" style="margin-top:6px">'
         f'<a href="https://huggingface.co/{m["id"]}" target="_blank" style="color:#6d28d9;font-weight:700;text-decoration:none">🤗 Model Card ↗</a> · '
-        f'<a href="https://deepmind.google/models/gemma/gemma-4/" target="_blank" style="color:#059669;font-weight:700;text-decoration:none">🔬 DeepMind ↗</a>'
+        f'<a href="https://huggingface.co/FINAL-Bench" target="_blank" style="color:#059669;font-weight:700;text-decoration:none">🧬 FINAL-Bench ↗</a>'
         f'</div></div>'
     )
 
-with gr.Blocks(title="Gemma 4 Playground") as demo:
+with gr.Blocks(title="Darwin-31B-Opus") as demo:
 
     with gr.Row():
-        gr.Markdown("## 💎 Gemma 4 Playground\nGoogle DeepMind · Apache 2.0 · Vision · Thinking")
+        gr.Markdown("## 🧬 Darwin-31B-Opus\nVIDRAFT · Diagnostic-Guided Evolutionary Merge · Apache 2.0 · Vision · Thinking")
         with gr.Column(scale=0, min_width=120):
             gr.LoginButton(size="sm")
 
     with gr.Row():
         # ── Sidebar ──
         with gr.Column(scale=0, min_width=280):
-            model_dd = gr.Dropdown(
-                choices=list(MODELS.keys()), value=DEFAULT_MODEL, label="Model",
-                info="⚡MoE=Fast | 🏆Dense=Best quality (전환 시 1-2분)",
-            )
             model_info = gr.HTML(value=_model_info_html(DEFAULT_MODEL))
             image_input = gr.Image(label="👁️ Image (Vision)", type="filepath", height=140)
             thinking_radio = gr.Radio(["⚡ Fast", "🧠 Thinking"], value="⚡ Fast", label="Mode")
@@ -360,15 +356,16 @@ with gr.Blocks(title="Gemma 4 Playground") as demo:
             chatbot = gr.Chatbot(elem_id="chatbot", show_label=False, height=600)
             with gr.Row():
                 chat_input = gr.Textbox(
-                    placeholder="Message Gemma 4…",
+                    placeholder="Message Darwin-31B-Opus…",
                     show_label=False, scale=7, autofocus=True, lines=1, max_lines=4,
                 )
                 send_btn = gr.Button("↑", variant="primary", scale=0, min_width=48, elem_id="send-btn")
 
     # ── Events ──
-    model_dd.change(fn=_model_info_html, inputs=[model_dd], outputs=[model_info])
     preset_dd.change(fn=lambda k: PRESETS.get(k, PRESETS["general"]), inputs=[preset_dd], outputs=[sys_prompt])
 
+    model_dd_hidden = gr.Textbox(value=DEFAULT_MODEL, visible=False)
+
     def user_msg(msg, hist):
         if not msg.strip(): return "", hist
         return "", hist + [{"role": "user", "content": msg}]
@@ -381,7 +378,7 @@ with gr.Blocks(title="Gemma 4 Playground") as demo:
             hist[-1]["content"] = chunk
             yield hist
 
-    ins = [chatbot, thinking_radio, image_input, sys_prompt, max_tok, temp, topp, model_dd]
+    ins = [chatbot, thinking_radio, image_input, sys_prompt, max_tok, temp, topp, model_dd_hidden]
     send_btn.click(user_msg, [chat_input, chatbot], [chat_input, chatbot], queue=False).then(bot_reply, ins, chatbot)
     chat_input.submit(user_msg, [chat_input, chatbot], [chat_input, chatbot], queue=False).then(bot_reply, ins, chatbot)
     clear_btn.click(lambda: [], None, chatbot, queue=False)
@@ -391,5 +388,5 @@ with gr.Blocks(title="Gemma 4 Playground") as demo:
 # 7.  LAUNCH
 # ══════════════════════════════════════════════════════════════════════════════
 if __name__ == "__main__":
-    print(f"[BOOT] Gemma 4 Playground · Model: {DEFAULT_MODEL}", flush=True)
+    print(f"[BOOT] Darwin-31B-Opus Playground · VIDRAFT", flush=True)
     demo.launch(server_name="0.0.0.0", server_port=7860, css=CSS, ssr_mode=False)