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):
+def generate_response(prompt, temperature, top_p, max_tokens, repetition_penalty, provider="openai"):
     try:
-        response = openai_client.chat.completions.create(
-           model="gpt-3.5-turbo" ===>  model="deepseek-chat",  # or another model of your choice
+        # 动态选择模型和客户端
+        if provider == "openai":
+            model = "gpt-3.5-turbo"
+            client = openai_client  # 假设你已配置好 OpenAI 客户端
+        elif provider == "deepseek":
+            model = "deepseek-chat"
+            client = deepseek_client  # 假设你已配置好 DeepSeek 客户端（兼容 OpenAI 接口）
+        else:
+            return f"Invalid provider: {provider}"
+
+        response = client.chat.completions.create(
+            model=model,
             messages=[{"role": "user", "content": prompt}],
             temperature=temperature,
             top_p=top_p,
@@ -20,8 +30,10 @@ def generate_response(prompt, temperature, top_p, max_tokens, repetition_penalty
             stream=False
         )
         return response.choices[0].message.content.strip()
+
     except Exception as e:
-        return f"OpenAI API Error: {str(e)}"
+        return f"{provider.title()} API Error: {str(e)}"
+
 
 
 iface = gr.Interface(