Update main.py

main.py

@@ -40,12 +40,13 @@ async def favicon():
     return FileResponse(favicon_path, media_type="image/x-icon")
     
 
-def generate_search(query: str, stream: bool = True) -> str:
-    headers = {"User-Agent": ""}
+def generate_search(query: str, stream: bool = True):
+    headers = {"User-Agent": "Your-User-Agent"}
     prompt = [
         {"role": "user", "content": query},
     ]
     
+    # Adding system message at the start
     prompt.insert(0, {"content": "Be Helpful and Friendly", "role": "system"})
     
     payload = {
@@ -62,16 +63,21 @@ def generate_search(query: str, stream: bool = True) -> str:
     for value in response.iter_lines(decode_unicode=True):
         if value.startswith("data: "):  
             try:
+                # Parse JSON and get the desired structure
                 json_modified_value = json.loads(value[6:])
-                content = json_modified_value.get("choices", [{}])[0].get("delta", {}).get("content", "")
+                content = json_modified_value.get("choices", [{}])[0].get("message", {}).get("content", "")
+                
+                # Stream the response with 'data:' prefix if streaming is enabled
                 if stream:
                     yield f"data: {json.dumps(json_modified_value)}\n\n"
+                
+                # Accumulate content for non-streaming
                 streaming_text += content
             except json.JSONDecodeError:
                 continue
     
     if not stream:
-        # For non-streaming, just yield the text content
+        # For non-streaming, return the accumulated text content
         yield streaming_text
 
 @app.get("/searchgpt")
@@ -80,6 +86,7 @@ async def search_gpt(q: str, stream: Optional[bool] = False):
         raise HTTPException(status_code=400, detail="Query parameter 'q' is required")
     
     if stream:
+        # Stream the response with 'data:' prefix
         return StreamingResponse(
             generate_search(q, stream=True),
             media_type="text/event-stream"