Update main.py

main.py

@@ -1,74 +1,61 @@
 import httpx
-from bs4 import BeautifulSoup
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-import uvicorn
-
-app = FastAPI()
-
-class SearchQuery(BaseModel):
-    query: str
-
-class BraveSearch:
-    def __init__(self, query):
-        self.query = query
-        self.url = f"https://search.brave.com/search?q={query.replace(' ', '+')}&source=web"
-
-    async def fetch_results(self):
+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.",
+    version="1.0.0",
+)
+
+# Define the base URL for the DuckDuckGo Lite API
+DUCKDUCKGO_LITE_URL = "https://lite.duckduckgo.com/lite/"
+
+@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.
+    """
+    params = {
+        "q": q,
+        "s": s,
+        "o": o,
+        "kl": kl,
+        "bing_market": bing_market,
+    }
+
+    # Use httpx for asynchronous requests
+    async with httpx.AsyncClient() as client:
         try:
-            async with httpx.AsyncClient() as client:
-                response = await client.get(self.url)
-                response.raise_for_status()
-
-            soup = BeautifulSoup(response.text, "html.parser")
-            results = {
-                "searchResults": [],
-                "additionalData": [],
-                "faq": []
-            }
-
-            for el in soup.select(".snippet"):
-                title = el.select_one(".title")
-                description = el.select_one(".snippet-description")
-                link = el.select_one("a")
-                results["searchResults"].append({
-                    "title": title.text.strip() if title else "No Title",
-                    "description": description.text.strip() if description else "No Description",
-                    "link": link["href"] if link else "#"
-                })
+            # 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)
+            
+            # Raise an exception for bad status codes (4xx or 5xx)
+            response.raise_for_status()
+            
+            # Return the JSON response from the API
+            return response.json()
 
-            for el in soup.select(".t-tertiary.svelte-1yt5tdo"):
-                attribution = el.select_one(".attribution")
-                citation_link = el.select_one("cite a")
-                results["additionalData"].append({
-                    "attribution": attribution.text.strip() if attribution else "No Attribution",
-                    "citationLink": citation_link["href"] if citation_link else "#"
-                })
-
-            for el in soup.select(".fq-item"):
-                question = el.select_one(".faq-q")
-                answer = el.select_one(".faq-a")
-                faq_link = el.select_one("a")
-                results["faq"].append({
-                    "question": question.text.strip() if question else "No Question",
-                    "answer": answer.text.strip() if answer else "No Answer",
-                    "faqLink": faq_link["href"] if faq_link else "#"
-                })
-
-            return results
         except httpx.HTTPStatusError as e:
-            raise HTTPException(status_code=e.response.status_code, detail=f"HTTP error occurred: {e}")
-        except Exception as e:
-            raise HTTPException(status_code=500, detail=f"An error occurred: {e}")
-
-@app.post("/search")
-async def search(search_query: SearchQuery):
-    if not search_query.query:
-        raise HTTPException(status_code=400, detail="Query is required")
-
-    search = BraveSearch(search_query.query)
-    data = await search.fetch_results()
-    return data
-
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+            raise HTTPException(
+                status_code=e.response.status_code,
+                detail=f"Error from DuckDuckGo API: {e.response.text}",
+            )
+        except httpx.RequestError as e:
+            raise HTTPException(
+                status_code=500,
+                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!"}