Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import json
 import requests
-from typing import Iterator, List, Dict
+from typing import List, Dict
 
 #   ____                              
 #  / __ \                             
@@ -16,27 +16,12 @@ from typing import Iterator, List, Dict
 
 API_URL = "https://app.oxyapi.uk/free/v1/chat/completions"
 
-def create_stream_response(response: requests.Response) -> Iterator[str]:
-    for line in response.iter_lines():
-        if not line or line.decode('utf-8').startswith("data: [DONE]"):
-            continue
-        try:
-            line_text = line.decode('utf-8')
-            if line_text.startswith("data: "):
-                json_response = json.loads(line_text[6:])
-                if 'choices' in json_response:
-                    delta = json_response['choices'][0].get('delta', {})
-                    if 'content' in delta:
-                        yield delta['content']
-        except json.JSONDecodeError:
-            continue
-
 def predict(
     message: str,
     chat_history: List[Dict[str, str]],
     temperature: float,
     top_p: float,
-) -> Iterator[List[Dict[str, str]]]:
+) -> List[Dict[str, str]]:
     messages = [
         {
             "role": "system",
@@ -47,8 +32,7 @@ def predict(
     payload = {
         "messages": messages,
         "temperature": temperature,
-        "top_p": top_p,
-        "stream": True
+        "top_p": top_p
     }
 
     headers = {
@@ -56,25 +40,27 @@ def predict(
         "Authorization": "Bearer oxy-1-small-gradio"
     }
 
-    chat_history = chat_history + [{"role": "user", "content": message}, {"role": "assistant", "content": ""}]
+    chat_history = chat_history + [{"role": "user", "content": message}]
     try:
         response = requests.post(
             API_URL,
             headers=headers,
-            json=payload,
-            stream=True
+            json=payload
         )
         response.raise_for_status()
         
-        assistant_content = ""
-        for delta in create_stream_response(response):
-            assistant_content += delta
-            chat_history[-1]["content"] = assistant_content
-            yield chat_history
-                
+        json_response = response.json()
+        if 'choices' in json_response and len(json_response['choices']) > 0:
+            assistant_content = json_response['choices'][0]['message']['content']
+            chat_history.append({"role": "assistant", "content": assistant_content})
+        else:
+            chat_history.append({"role": "assistant", "content": "Error: No response from assistant."})
+
+        return chat_history
+            
     except Exception as e:
-        chat_history[-1]["content"] = f"Error: {str(e)}"
-        yield chat_history
+        chat_history.append({"role": "assistant", "content": f"Error: {str(e)}"})
+        return chat_history
 
 css = """
 .gradio-container {