Update app.py

app.py

@@ -1,17 +1,18 @@
-from fastapi import FastAPI, Query
-from fastapi.middleware.cors import CORSMiddleware
+from fastapi import FastAPI
+from fastapi.responses import JSONResponse
 from duckduckgo_search import DDGS
-import uvicorn
+from fastapi.middleware.cors import CORSMiddleware
 
+# Initialize FastAPI app
 app = FastAPI()
 
-# Enable CORS for all origins (fixes CORS issues)
+# CORS Middleware to allow cross-origin requests
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
+    allow_origins=["*"],  # Allow all origins
     allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
+    allow_methods=["*"],  # Allow all methods (GET, POST, etc.)
+    allow_headers=["*"],  # Allow all headers
 )
 
 @app.get("/")
@@ -19,12 +20,19 @@ def home():
     return {"message": "DuckDuckGo Search API is running!"}
 
 @app.get("/search")
-def search(query: str = Query(..., title="Search Query"), max_results: int = 10):
-    """ DuckDuckGo Search API - Returns top search results. """
-    with DDGS() as ddgs:
-        results = list(ddgs.text(query, max_results=max_results))
-    return {"query": query, "results": results}
+def search(query: str, max_results: int = 5):
+    try:
+        # Use headers to mimic a real browser
+        headers = {
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/110.0"
+        }
+        
+        # Use DuckDuckGo Search API with HTML backend
+        with DDGS(headers=headers) as ddgs:
+            results = list(ddgs.text(query, max_results=max_results, backend="html"))
+
+        return JSONResponse(content={"query": query, "results": results})
+    
+    except Exception as e:
+        return JSONResponse(content={"error": str(e)}, status_code=500)
 
-# Ensure the API runs properly
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)