Update app.py

app.py

@@ -1,18 +1,21 @@
-from fastapi import FastAPI, Request
-from transformers import pipeline
+from fastapi import FastAPI, Query
+from fastapi.middleware.cors import CORSMiddleware
+from duckduckgo_search import DDGS
 
 app = FastAPI()
 
-# Load the model
-generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
+# Enable CORS for all origins (allows API to be used from any website)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
-@app.get("/")
-def read_root():
-    return {"message": "Welcome to the free LLM API"}
-
-@app.post("/generate")
-async def generate_text(request: Request):
-    data = await request.json()
-    prompt = data.get("prompt", "")
-    response = generator(prompt, max_length=200, do_sample=True)
-    return {"response": response[0]['generated_text']}
+@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}