LLM / app.py
hrsprojects's picture
Update app.py
3809b9b verified
raw
history blame
1.2 kB
from fastapi import FastAPI
from fastapi.responses import JSONResponse
from duckduckgo_search import DDGS
from fastapi.middleware.cors import CORSMiddleware
# Initialize FastAPI app
app = FastAPI()
# CORS Middleware to allow cross-origin requests
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], # Allow all origins
allow_credentials=True,
allow_methods=["*"], # Allow all methods (GET, POST, etc.)
allow_headers=["*"], # Allow all headers
)
@app.get("/")
def home():
return {"message": "DuckDuckGo Search API is running!"}
@app.get("/search")
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)