Update app.py

app.py

@@ -1,33 +1,58 @@
 import gradio as gr
 import os
 from openai import OpenAI
+import requests
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 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")
+deepseek_base_url = "https://api.deepseek.com"  # assuming DeepSeek uses a REST API, you can adjust as needed
 
-
-def generate_response(prompt,temperature, top_p, max_tokens, repetition_penalty):
+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
-            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()
-    except Exception as e:
-        return f"OpenAI API Error: {str(e)}"
+        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 = {
+                "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."
+
+    except Exception as e:
+        return f"Error: {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"),
@@ -41,3 +66,4 @@ iface = gr.Interface(
 
 iface.launch()
 
+