Update app.py

app.py

@@ -8,11 +8,16 @@ 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,model_provider,temperature, top_p, max_tokens, repetition_penalty):
-    m={'DeepSeek':'deepseek-chat','OpenAI':'gpt-3.5-turbo'}
+def generate_response(prompt, model_provider, temperature, top_p, max_tokens, repetition_penalty):
+    models = {
+        "DeepSeek": ("deepseek-chat", deepseek_client),
+        "OpenAI": ("gpt-3.5-turbo", openai_client)
+    }
+    model, client = models[model_provider]
+    
     try:
-        response = openai_client.chat.completions.create(
-            model=m[model_provider],  # model="deepseek-chat"，
+        response = client.chat.completions.create(
+            model=model,
             messages=[{"role": "user", "content": prompt}],
             temperature=temperature,
             top_p=top_p,
@@ -22,8 +27,7 @@ def generate_response(prompt,model_provider,temperature, top_p, max_tokens, repe
         )
         return response.choices[0].message.content.strip()
     except Exception as e:
-        return f"OpenAI API Error: {str(e)}"
-
+        return f"{model_provider} API Error: {str(e)}"
 
 iface = gr.Interface(
     fn=generate_response,