Update main.py

main.py

@@ -65,20 +65,14 @@ def generate_search(query: str, stream: bool = True) -> str:
                 json_modified_value = json.loads(value[6:])
                 content = json_modified_value.get("choices", [{}])[0].get("delta", {}).get("content", "")
                 if stream:
-                    # Don't double-encode the JSON
-                    yield f"data: {json_modified_value}\n\n"
+                    yield f"data: {json.dumps(json_modified_value)}\n\n"
                 streaming_text += content
             except json.JSONDecodeError:
                 continue
     
     if not stream:
-        # Return a proper JSON response without double encoding
-        yield {
-            "choices": [{
-                "message": {"content": streaming_text},
-                "finish_reason": "stop"
-            }]
-        }
+        # For non-streaming, just yield the text content
+        yield streaming_text
 
 @app.get("/searchgpt")
 async def search_gpt(q: str, stream: Optional[bool] = False):
@@ -91,10 +85,10 @@ async def search_gpt(q: str, stream: Optional[bool] = False):
             media_type="text/event-stream"
         )
     else:
-        # For non-streaming response, collect all content and return as JSON
+        # For non-streaming, collect the text and return as JSON response
         response_text = "".join([chunk for chunk in generate_search(q, stream=False)])
         return JSONResponse(content={"response": response_text})
-
+        
 @app.get("/", response_class=HTMLResponse)
 async def root():
     # Open and read the content of index.html (in the same folder as the app)