Update main.py

main.py

@@ -39,6 +39,9 @@ async def favicon():
     favicon_path = Path(__file__).parent / "favicon.ico"
     return FileResponse(favicon_path, media_type="image/x-icon")
     
+import requests
+import json
+
 def generate_search(query: str, stream: bool = True) -> str:
     headers = {"User-Agent": ""}
     prompt = [
@@ -65,19 +68,20 @@ def generate_search(query: str, stream: bool = True) -> str:
         # Ensure the value starts with 'data: ' and process it
         if value.startswith("data: "):  
             try:
-                json_modified_value = json.loads(value[6:])  # Remove 'data: ' prefix
-                content = json_modified_value
-                # Include everything, even if it's just whitespace
-                if stream:
-                    yield f"data: {content}\n"
-                streaming_text += content
+                # Parse JSON and remove 'data: ' prefix
+                json_modified_value = json.loads(value[6:])
+                content = json_modified_value.get("content", "")  # Get the actual content
+                
+                if content.strip():  # Ensure non-empty content
+                    if stream:
+                        yield content  # Pass content directly, removing 'data: ' and JSON overhead
+                    streaming_text += content  # Accumulate content for non-streaming case
             except json.JSONDecodeError:
                 continue  # Skip lines that are not valid JSON
     
     # If not streaming, yield the full collected content
     if not stream:
         yield streaming_text
-        
 @app.get("/searchgpt")
 async def search_gpt(q: str, stream: Optional[bool] = False):
     if not q: