Update main.py

main.py

@@ -1,49 +1,52 @@
 import httpx
+import json
 from fastapi import FastAPI, HTTPException, Query
 from typing import Optional
 
 # Initialize the FastAPI app
 app = FastAPI(
-    title="FastAPI DuckDuckGo Lite Proxy",
-    description="A FastAPI proxy for the DuckDuckGo Lite search engine.",
+    title="FastAPI DuckDuckGo API Proxy",
+    description="A FastAPI proxy for the DuckDuckGo Instant Answer API.",
     version="1.0.0",
 )
 
-# Define the base URL for the DuckDuckGo Lite API
-DUCKDUCKGO_LITE_URL = "https://lite.duckduckgo.com/lite/"
+# Define the base URL for the DuckDuckGo Instant Answer API (for JSON results)
+DUCKDUCKGO_API_URL = "https://api.duckduckgo.com/"
 
 @app.get("/search", tags=["Search"])
 async def search_duckduckgo(
     q: str = Query(..., description="The search query."),
-    s: Optional[int] = Query(0, description="Can be `0`."),
-    o: Optional[str] = Query("json", description="Set to `json` for JSON output."),
-    kl: Optional[str] = Query("wt-wt", description="Region, e.g., us-en, uk-en."),
-    bing_market: Optional[str] = Query("wt-wt", description="Bing market region.")
 ):
     """
-    Performs a search using the DuckDuckGo Lite API and returns the results.
+    Performs a search using the DuckDuckGo Instant Answer API and returns the results in JSON format.
     """
     params = {
         "q": q,
-        "s": s,
-        "o": o,
-        "kl": kl,
-        "bing_market": bing_market,
+        "format": "json",  # Request JSON format
+        "no_html": 1,      # Disable HTML in the response
     }
 
-    # Use httpx for asynchronous requests
     async with httpx.AsyncClient() as client:
         try:
-            # The OpenAPI spec indicates a POST, but parameters are in the query.
-            # A GET request is more appropriate for a search with query parameters.
-            response = await client.get(DUCKDUCKGO_LITE_URL, params=params)
+            response = await client.get(DUCKDUCKGO_API_URL, params=params)
             
             # Raise an exception for bad status codes (4xx or 5xx)
             response.raise_for_status()
             
-            # Return the JSON response from the API
+            # The API returns an empty string for no results, which is not valid JSON.
+            # Handle this case explicitly.
+            if not response.text:
+                return {"results": [], "message": "No results found."}
+
+            # Attempt to parse the JSON response
             return response.json()
 
+        except json.JSONDecodeError:
+            # If the response is not valid JSON, raise an internal server error
+            raise HTTPException(
+                status_code=500,
+                detail=f"Failed to decode JSON from DuckDuckGo API. Response text: {response.text}",
+            )
         except httpx.HTTPStatusError as e:
             raise HTTPException(
                 status_code=e.response.status_code,
@@ -55,7 +58,6 @@ async def search_duckduckgo(
                 detail=f"Failed to connect to DuckDuckGo API: {str(e)}",
             )
 
-# Optional: Add a root endpoint for health checks or basic info
 @app.get("/", tags=["Root"])
 async def read_root():
-    return {"message": "Welcome to the DuckDuckGo Lite API proxy!"}
+    return {"message": "Welcome to the DuckDuckGo API proxy!"}