Update app.py

app.py

@@ -1,38 +1,59 @@
-from flask import Flask, render_template, request, jsonify
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import JSONResponse
+from fastapi.encoders import jsonable_encoder
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 import requests
 import os
 from webscout import WEBS, transcriber
 
-app = Flask(__name__)
+app = FastAPI()
 
 BASE_URL = "https://oevortex-webscout-api.hf.space"
 
-@app.route("/", methods=["GET", "POST"])
-def index():
-    return render_template("index.html")
-
-@app.route("/api/search", methods=["GET"])
-def search():
-    q = request.args.get("q")
-    max_results = int(request.args.get("max_results", 10))
-    timelimit = request.args.get("timelimit") # Timelimit is optional
-    safesearch = request.args.get("safesearch", "moderate")
-    region = request.args.get("region", "wt-wt")
-    backend = request.args.get("backend", "api")
+app.mount("/static", StaticFiles(directory="static"), name="static")
+templates = Jinja2Templates(directory="templates")
+
+@app.get("/", response_class=HTMLResponse)
+async def index(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request})
+
+@app.get("/api/suggestions")
+async def get_suggestions(q: str):
+    api_endpoint = f"{BASE_URL}/api/suggestions?q={q}"
+    response = requests.get(api_endpoint)
+    return response.json()
+
+@app.get("/api/search")
+async def search(
+    q: str,
+    max_results: int = 10,
+    timelimit: Optional[str] = None,
+    safesearch: str = "moderate",
+    region: str = "wt-wt",
+    backend: str = "api"
+):
+    """Perform a text search."""
     try:
         with WEBS() as webs:
             results = webs.text(keywords=q, region=region, safesearch=safesearch, timelimit=timelimit, backend=backend, max_results=max_results)
-            # Extract data you want to display
+            # Extract the data you want to display
             search_results = [
                 {
                     'title': result.title,
                     'description': result.description
                 } for result in results
             ]
-            return jsonify(search_results)  # Return as JSON
+            return JSONResponse(content=jsonable_encoder(search_results))
     except Exception as e:
-        return jsonify({"error": f"Error during search: {e}"})
+        raise HTTPException(status_code=500, detail=f"Error during search: {e}")
+
+@app.get("/api/answers")
+async def get_people_also_search(q: str):
+    api_endpoint = f"{BASE_URL}/api/answers?q={q}"
+    response = requests.get(api_endpoint)
+    return response.json()
 
 if __name__ == "__main__":
-    port = int(os.environ.get('PORT', 7860)) 
-    app.run(host='0.0.0.0', port=port)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860, reload=True)