Update main.py

main.py

@@ -1,4 +1,3 @@
-
 import os
 from dotenv import load_dotenv
 from fastapi import FastAPI, HTTPException, Request
@@ -59,17 +58,15 @@ def generate_search(query: str, stream: bool = True) -> str:
     # Collect streamed text content
     streaming_text = ""
     for value in response.iter_lines(decode_unicode=True, chunk_size=12):
-        if value:
+        modified_value = re.sub("data:", "", value)
+        if modified_value:
             try:
-                json_value = json.loads(value)  # Process the JSON content directly
-                content = json_value["choices"][0]["delta"]["content"]
-                
-                # Stream the raw content directly (no 'data:' prefix)
+                json_modified_value = json.loads(modified_value)
+                content = json_modified_value["choices"][0]["delta"]["content"]
                 if stream:
-                    yield content
-                
+                    yield f"data: {content}\n\n"
                 streaming_text += content
-            except json.JSONDecodeError:
+            except:
                 continue
     
     if not stream: