Update app.py

app.py

@@ -9,47 +9,43 @@ openai_client = OpenAI(api_key=OPENAI_API_KEY)
 DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
 deepseek_base_url = "https://api.deepseek.com"  # assuming DeepSeek uses a REST API, you can adjust as needed
 
-def generate_response(model_provider, prompt, temperature, top_p, max_tokens, repetition_penalty):
+def generate_response(Model_provider, prompt, temperature, top_p, max_tokens, repetition_penalty):
     try:
-        if model_provider == "OpenAI":
-            response = openai_client.chat.completions.create(
-                model="gpt-3.5-turbo",  # or another model of your choice
-                messages=[{"role": "user", "content": prompt}],
-                temperature=temperature,
-                top_p=top_p,
-                max_tokens=max_tokens,
-                presence_penalty=repetition_penalty,
-                stream=False
-            )
-            return response.choices[0].message.content.strip()
-
-        elif model_provider == "DeepSeek":
-            headers = {
-                "Authorization": f"Bearer {DEEPSEEK_API_KEY}",
-                "Content-Type": "application/json"
-            }
-            payload = {
-                "model": "deepseek-chat",  # 使用 DeepSeek 的 "deepseek-chat" 模型
-                "prompt": prompt,
-                "temperature": temperature,
-                "top_p": top_p,
-                "max_tokens": max_tokens,
-                "repetition_penalty": repetition_penalty
-            }
-            # Assuming DeepSeek has a /generate endpoint, adjust URL and payload if needed
-            response = requests.post(f"{deepseek_base_url}/generate", headers=headers, json=payload)
-            
-            if response.status_code == 200:
-                return response.json().get("generated_text", "No response text found.")
-            else:
-                return f"DeepSeek API Error: {response.status_code}, {response.text}"
-        
-        else:
-            return "Invalid model provider selected."
-
+        response = deepseek_client.chat.completions.create（
+        model="deepseek-chat", #or "deepseek-reasoner" for R1 model
+        messages=[f"role":"user","content": prompt}],
+        temperature=temperature,
+        top_p=top_p,
+        max_tokens=max_tokens,
+        presence_penalty=repetition_penalty,
+        stream=False
+)
     except Exception as e:
-        return f"Error: {str(e)}"
-
+        return f"DeepSeek API Error: Istr(e)]"
+elif model_provider == "OpenAI":
+    try:
+        response = openai_client.chat.completions.create（
+        model="gpt-3.5-turbo", # or another model of your choice
+        messages=[f"role": "user","content":prompt}],
+        temperature=temperature,
+        top_p=top_P,
+        max_tokens=max_tokens,
+        presence_penalty=repetition_penalty,
+        stream=False
+)
+        return response.choices[o].message.content.strip()
+    except Exception as e:
+        return f"OpenAI API Error: [str(e)]"
+    else:
+        return "Invalid model provider selected."
+with gr.Blocks() as demo:
+    gr.Markdown("# LLM Chat Interface")
+    with gr.Row():
+        model_provider = gr.Dropdown(
+        choices=["DeepSeek", "OpenAI"],
+        value="DeepSeek",
+        label="Select Model Provider"
+        prompt = gr.Textbox(label="Enter your prompt", lines=4, placeholder="Type your message here..")
 iface = gr.Interface(
     fn=generate_response,
     inputs=[