Update main.py

main.py

@@ -33,6 +33,9 @@ class Payload(BaseModel):
     messages: list
     stream: bool
 
+import requests
+import json
+
 def generate_search(query: str, stream: bool = True) -> str:
     headers = {"User-Agent": ""}
     prompt = [
@@ -56,20 +59,22 @@ 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):
-        modified_value = re.sub(value)
-        if modified_value:
+        if value:
             try:
-                json_modified_value = json.loads(modified_value)
-                content = json_modified_value["choices"][0]["delta"]["content"]
+                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)
                 if stream:
-                    yield f"data: {content}\n\n"
+                    yield content
+                
                 streaming_text += content
-            except:
+            except json.JSONDecodeError:
                 continue
     
     if not stream:
         yield streaming_text
-
+        
 @app.get("/searchgpt")
 async def search_gpt(q: str, stream: Optional[bool] = False):
     if not q: