Update app.py

app.py

@@ -8,10 +8,20 @@ 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):#prommpt->modol->
+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",  # or another model of your choice
+        # 根据选择使用不同的客户端和模型
+        if model_provider == "OpenAI":
+            client = openai_client
+            model_name = "gpt-3.5-turbo"
+        elif model_provider == "DeepSeek":
+            client = deepseek_client
+            model_name = "deepseek-chat"  # 替换成你想用的 DeepSeek 模型名称
+        else:
+            return "❌ 无效的模型提供者选择"
+
+        response = client.chat.completions.create(
+            model=model_name,
             messages=[{"role": "user", "content": prompt}],
             temperature=temperature,
             top_p=top_p,
@@ -21,11 +31,12 @@ 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 Error: {str(e)}"
-#
+        return f"❌ API 调用出错：{str(e)}"
+
 iface = gr.Interface(
     fn=generate_response,
     inputs=[
+        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"),
@@ -33,8 +44,8 @@ iface = gr.Interface(
         gr.Slider(minimum=1.0, maximum=2.0, value=1.1, step=0.1, label="Repetition Penalty")
     ],
     outputs="text",
-    title="🧠 DeepSeek LLM Chat with Parameter Tuning",
+    title="🧠 DeepSeek & OpenAI Chat with Parameter Tuning",
     theme=gr.themes.Soft()
 )
 
-iface.launch()
+iface.launch()