Update app.py

app.py

@@ -9,7 +9,7 @@ set_seed(42)
 
 # 从环境变量获取 token
 HF_TOKEN = os.environ.get('HF_TOKEN')
-print(f"Token 状态: {'已设置' if HF_TOKEN else '未设置'}")
+token_status = "✅ 已设置" if HF_TOKEN else "❌ 未设置"
 
 # 初始化模型管道
 try:
@@ -20,17 +20,22 @@ try:
         torch_dtype="auto"
     )
     MODEL_LOADED = True
-    print("✅ 本地模型加载成功")
+    model_status = "✅ 本地模型加载成功"
+    print(model_status)
 except Exception as e:
     print(f"❌ 本地模型加载失败: {e}")
     MODEL_LOADED = False
     protein_generator = None
+    model_status = "❌ 本地模型加载失败，使用API模式"
+
+# 确定运行模式
+if MODEL_LOADED:
+    run_mode = "🧠 本地模型"
+else:
+    run_mode = "🌐 API调用"
 
 def generate_with_local_model(instruction, max_length=100):
     """使用本地加载的模型生成蛋白质序列"""
-    if not MODEL_LOADED or protein_generator is None:
-        return "❌ 模型未正确加载，请检查控制台日志"
-    
     try:
         result = protein_generator(
             instruction,
@@ -72,7 +77,6 @@ def generate_with_api(instruction, max_length=100):
             }
         }
         
-        print("🔄 正在调用 API...")
         response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
         
         if response.status_code == 200:
@@ -94,34 +98,33 @@ def generate_with_api(instruction, max_length=100):
 
 def generate_protein(instruction, max_length=100):
     """主生成函数"""
-    yield "🔄 正在处理您的请求..."
+    if not instruction or instruction.strip() == "":
+        return "❌ 请输入有效的蛋白质生成指令"
+    
+    # 显示处理状态
     time.sleep(0.5)
     
     try:
-        # 显示当前模式
-        mode = "本地模型" if MODEL_LOADED else "API调用"
-        yield f"🔄 使用 {mode} 处理中..."
-        
         # 优先使用本地模型，失败时使用API
         if MODEL_LOADED:
             result = generate_with_local_model(instruction, max_length)
         else:
             result = generate_with_api(instruction, max_length)
         
-        yield result
+        return result
         
     except Exception as e:
-        yield f"❌ 生成过程中出现未预期错误: {str(e)}"
+        return f"❌ 生成过程中出现未预期错误: {str(e)}"
 
-# 创建界面 - 移除了不兼容的参数
+# 创建界面
 with gr.Blocks(title="ProtTeX 蛋白质生成器") as demo:
     
-    gr.Markdown("""
+    gr.Markdown(f"""
     # 🧬 ProtTeX 蛋白质生成器
     
     **使用自然语言指令生成蛋白质序列**
     
-    *当前模式: {'🧠 本地模型' if MODEL_LOADED else '🌐 API 调用'} | Token: {'✅ 已设置' if HF_TOKEN else '❌ 未设置'}*
+    *当前模式: {run_mode} | Token状态: {token_status}*
     """)
     
     with gr.Row():
@@ -150,7 +153,6 @@ with gr.Blocks(title="ProtTeX 蛋白质生成器") as demo:
             output = gr.Textbox(
                 label="🧬 生成的蛋白质序列",
                 lines=8
-                # 移除了 show_copy_button 参数
             )
     
     # 示例部分
@@ -163,7 +165,10 @@ with gr.Blocks(title="ProtTeX 蛋白质生成器") as demo:
             ["Design a protein with enzymatic activity for hydrolysis"],
             ["Generate a stable protein for high temperature environments"]
         ],
-        inputs=[instruction]
+        inputs=[instruction],
+        outputs=[output],
+        fn=generate_protein,
+        cache_examples=False
     )
     
     # 连接按钮事件
@@ -177,8 +182,9 @@ with gr.Blocks(title="ProtTeX 蛋白质生成器") as demo:
     gr.Markdown(f"""
     ---
     **系统状态**:
-    - 运行模式: {'🧠 本地模型' if MODEL_LOADED else '🌐 API 调用'}
-    - Token 状态: {'✅ 已设置' if HF_TOKEN else '❌ 未设置'}
+    - 运行模式: {run_mode}
+    - Token状态: {token_status}
+    - 模型状态: {model_status}
     - 硬件: CPU Basic
     
     *基于 [mzcwd/ProtTeX](https://huggingface.co/mzcwd/ProtTeX) 模型*