Update main.py

main.py

@@ -40,6 +40,16 @@ async def favicon():
     return FileResponse(favicon_path, media_type="image/x-icon")
     
 
+import json
+import requests
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import StreamingResponse, JSONResponse
+from typing import Optional
+
+app = FastAPI()
+
+secret_api_endpoint_3 = "your_secret_endpoint_here"  # Replace with your actual endpoint
+
 def generate_search(query: str, stream: bool = True) -> str:
     headers = {"User-Agent": ""}
     prompt = [
@@ -64,14 +74,30 @@ def generate_search(query: str, stream: bool = True) -> str:
             try:
                 json_modified_value = json.loads(value[6:])
                 content = json_modified_value.get("choices", [{}])[0].get("delta", {}).get("content", "")
-                if stream:
-                    yield f"data: {json.dumps(json_modified_value)}\n\n"
-                streaming_text += content
+
+                if content.strip():  # Only process non-empty content
+                    cleaned_response = {
+                        "created": json_modified_value.get("created"),
+                        "id": json_modified_value.get("id"),
+                        "model": "searchgpt",
+                        "object": "chat.completion",
+                        "choices": [
+                            {
+                                "message": {
+                                    "content": content
+                                }
+                            }
+                        ]
+                    }
+                    
+                    if stream:
+                        yield f"data: {json.dumps(cleaned_response)}\n\n"
+                    
+                    streaming_text += content
             except json.JSONDecodeError:
                 continue
     
     if not stream:
-        # For non-streaming, just yield the text content
         yield streaming_text
 
 @app.get("/searchgpt")
@@ -88,7 +114,7 @@ async def search_gpt(q: str, stream: Optional[bool] = False):
         # 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)