Update app.py

app.py

@@ -10,8 +10,6 @@
 
 import os
 import re
-import json
-import time
 import asyncio
 import httpx
 import gradio as gr
@@ -31,7 +29,11 @@ SIMSIMI_BAD_MAX  = float(os.getenv("SIMSIMI_BAD_MAX", "0.30"))
 # ---------------------------
 # Load Qwen
 # ---------------------------
-dtype = torch.float16 if torch.cuda.is_available() else torch.bfloat16 if torch.backends.mps.is_available() else torch.float32
+dtype = (
+    torch.float16 if torch.cuda.is_available()
+    else torch.bfloat16 if torch.backends.mps.is_available()
+    else torch.float32
+)
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_ID,
@@ -174,13 +176,13 @@ async def hybrid_reply(history_messages, user_text, mode: str, lang: str, bad_ma
 with gr.Blocks(css="""
 #chatbot {height: 560px}
 """) as demo:
-    gr.Markdown("## Qwen × SimSimi Hybrid Chat\n小贴士：在 Settings → Repository secrets 设置 SIMSIMI_API_KEY 后即可使用 SimSimi。")
+    gr.Markdown("## Qwen × SimSimi Hybrid Chat\n")
 
-    # 只保留“对话模式”一个选项（映射到内部的“Auto 混合”）
+    # ✅ 页面只保留一个下拉框（含三种模式），以及提交/清空按钮
     mode_dd = gr.Dropdown(
-        choices=["对话模式"],
-        value="对话模式",
-        label="模式"
+        choices=["Auto 混合", "只用 Qwen", "只用 SimSimi"],
+        value="Auto 混合",
+        label="对话模式"
     )
 
     chatbox = gr.Chatbot(elem_id="chatbot")
@@ -191,23 +193,18 @@ with gr.Blocks(css="""
     # 内部状态：存储 Qwen 用的 messages
     state_msgs = gr.State([{"role": "system", "content": SYSTEM_PROMPT}])
 
-    async def respond(user_text, history, messages, mode_label):
+    async def respond(user_text, history, messages, mode):
         user_text = (user_text or "").strip()
         if not user_text:
             return gr.update(), messages, ""
-
-        # 将“对话模式”映射到内部“Auto 混合”
-        internal_mode = "Auto 混合"
-
-        # 使用环境变量/默认值作为 SimSimi 参数
+        # 使用环境变量作为 SimSimi 参数（不在页面暴露）
         lang = SIMSIMI_LANG
         bad_max = SIMSIMI_BAD_MAX
 
-        # 更新 message 历史
         messages = list(messages) if messages else [{"role": "system", "content": SYSTEM_PROMPT}]
         messages.append({"role": "user", "content": user_text})
 
-        reply = await hybrid_reply(messages, user_text, mode=internal_mode, lang=lang, bad_max=bad_max)
+        reply = await hybrid_reply(messages, user_text, mode=mode, lang=lang, bad_max=bad_max)
 
         messages.append({"role": "assistant", "content": reply})
         history = (history or []) + [[user_text, reply]]
@@ -216,13 +213,12 @@ with gr.Blocks(css="""
     def clear_all():
         return [], [{"role": "system", "content": SYSTEM_PROMPT}]
 
-    # 仅“提交”按钮触发发送；移除回车提交
+    # 仅“提交”按钮触发发送；如需回车发送，可再加 user_in.submit(...)
     submit_btn.click(
         respond,
         inputs=[user_in, chatbox, state_msgs, mode_dd],
         outputs=[chatbox, state_msgs, user_in]
     )
-
     clear_btn.click(
         clear_all,
         inputs=None,