Update app.py

app.py

@@ -9,10 +9,14 @@ openai_client = OpenAI(api_key=OPENAI_API_KEY)
 DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
 deepseek_client = OpenAI(api_key=DEEPSEEK_API_KEY, base_url="https://api.deepseek.com")
 
-def generate_response(prompt, temperature, top_p, max_tokens, repetition_penalty):  # 现在明确有5个参数
+def generate_response(model_provider, prompt, temperature, top_p, max_tokens, repetition_penalty):
     try:
-        response = openai_client.chat.completions.create(
-            model="gpt-3.5-turbo",
+        # 根据选择的模型提供商决定使用哪个客户端
+        client = deepseek_client if model_provider == "DeepSeek" else openai_client
+        model = "deepseek-chat" if model_provider == "DeepSeek" else "gpt-3.5-turbo"
+        
+        response = client.chat.completions.create(
+            model=model,
             messages=[{"role": "user", "content": prompt}],
             temperature=temperature,
             top_p=top_p,
@@ -22,20 +26,21 @@ def generate_response(prompt, temperature, top_p, max_tokens, repetition_penalty
         )
         return response.choices[0].message.content.strip()
     except Exception as e:
-        return f"OpenAI API错误: {str(e)}"
+        return f"API错误: {str(e)}"
 
 iface = gr.Interface(
     fn=generate_response,
     inputs=[
-        gr.Textbox(label="输入提示", lines=6, placeholder="请输入问题..."),
-        gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="温度参数"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p采样"),
-        gr.Slider(minimum=32, maximum=2048, value=512, step=32, label="最大生成长度"),
-        gr.Slider(minimum=1.0, maximum=2.0, value=1.1, step=0.1, label="重复惩罚系数")
+        gr.Dropdown(choices=["DeepSeek", "OpenAI"], value="DeepSeek", label="Model Provider"),
+        gr.Textbox(label="Prompt", lines=6, placeholder="Ask something..."),
+        gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="Temperature"),
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p"),
+        gr.Slider(minimum=32, maximum=2048, value=512, step=32, label="Max New Tokens"),
+        gr.Slider(minimum=1.0, maximum=2.0, value=1.1, step=0.1, label="Repetition Penalty")
     ],
     outputs="text",
     title="🧠 DeepSeek LLM 聊天演示（参数可调）",
     theme=gr.themes.Soft()
 )
 
-iface.launch()
+iface.launch()