Create main.py

main.py

@@ -0,0 +1,54 @@
+# app.py
+from fastapi import FastAPI, Request
+from pydantic import BaseModel
+from duckduckgo_search import DDGS
+import requests
+from bs4 import BeautifulSoup
+from together import Together
+import os
+
+# Your Together API Key
+os.environ["TOGETHER_API_KEY"] = "tgp_v1_fviFjIVOiQFgzx40nMG-q_8mrWpocP8VzWBZMLH1FbA"
+
+app = FastAPI()
+client = Together()
+
+class Query(BaseModel):
+    query: str
+
+def get_web_context(prompt: str) -> dict:
+    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
+
+    context = ""
+    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)
+            context += f"\nFrom {url}:\n{text[:2000]}\n"
+        except Exception as e:
+            context += f"\nFrom {url}:\nError retrieving content: {e}\n"
+    return context.strip()
+
+@app.post("/api/search")
+async def search(q: Query):
+    web_data = get_web_context(q.query)
+    messages = [
+        {"role": "system", "content": "You are a helpful search assistant that reads from the following web sources to answer user questions."},
+        {"role": "user", "content": f"{q.query}\n\nHere is some context:\n{web_data}"}
+    ]
+    try:
+        response = client.chat.completions.create(
+            model="meta-llama/Llama-3.3-70B-Instruct-Turbo-Free",
+            messages=messages
+        )
+        return {"answer": response.choices[0].message.content}
+    except Exception as e:
+        return {"answer": f"Error: {str(e)}"}