Update app.py

app.py

@@ -52,3 +52,43 @@ async def search(q: Query):
         return {"answer": response.choices[0].message.content}
     except Exception as e:
         return {"answer": f"Error: {str(e)}"}
+
+
+def search(prompt):
+    results = []
+    with DDGS() as ddgs:
+        for r in ddgs.text(prompt, max_results=3):
+            if r.get("href"):
+                results.append(r["href"])
+            if len(results) == 3:
+                break
+
+    data = {}
+    headers = {"User-Agent": "Mozilla/5.0"}
+    for url in results:
+        try:
+            res = requests.get(url, headers=headers, timeout=10)
+            soup = BeautifulSoup(res.content, "html.parser")
+            text = soup.get_text(separator=' ', strip=True)
+            data[url] = text[:5000]
+        except Exception as e:
+            data[url] = f"Error retrieving content: {e}"
+    return data
+
+@app.get("/ask")
+def ask(request: Request):
+    query_param = request.query_params.get("prompt")
+    if not query_param:
+        return {"error": "No prompt provided"}
+
+    sources = search(query_param)
+
+    content_blocks = "\n\n".join([f"Source: {url}\n{text}" for url, text in sources.items()])
+    combined_prompt = f"{content_blocks}\n\nUser: {query_param}"
+
+    response = client.chat.completions.create(
+        model="meta-llama/Llama-3.3-70B-Instruct-Turbo-Free",
+        messages=[{"role": "user", "content": combined_prompt}]
+    )
+
+    return {"answer": response.choices[0].message.content}