Fix import

backend/main.py

@@ -1,29 +1,28 @@
-"""
-FastAPI backend for the AI Shopping Research Agent.
-
-Endpoints:
-  GET  /api/health              -> quick status check (keys configured? etc)
-  POST /api/recommend           -> runs the full multi-agent pipeline
-
-Also serves the static frontend (frontend/) directly, so running this one
-process gives you the whole working app -- no separate frontend server needed.
-"""
 import logging
 import os
 
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
+from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel, Field
 
 from backend.agents.orchestrator import run_pipeline
 from backend.config import GEMINI_API_KEY, SERPAPI_KEY
 
+# -------------------------------------------------------
+# Logging
+# -------------------------------------------------------
 logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger("main")
-
-app = FastAPI(title="AI Shopping Research Agent", version="1.0.0")
+logger = logging.getLogger("ShopPilot")
+
+# -------------------------------------------------------
+# App
+# -------------------------------------------------------
+app = FastAPI(
+    title="ShopPilot AI",
+    version="1.0.0"
+)
 
 app.add_middleware(
     CORSMiddleware,
@@ -33,14 +32,34 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-FRONTEND_DIR = os.path.join(os.path.dirname(__file__), "..", "frontend")
-
+# -------------------------------------------------------
+# Paths
+# -------------------------------------------------------
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+PROJECT_DIR = os.path.abspath(os.path.join(BASE_DIR, ".."))
+FRONTEND_DIR = os.path.join(PROJECT_DIR, "frontend")
+
+logger.info(f"Frontend directory: {FRONTEND_DIR}")
+
+# -------------------------------------------------------
+# Static Files
+# -------------------------------------------------------
+app.mount(
+    "/frontend",
+    StaticFiles(directory=FRONTEND_DIR),
+    name="frontend"
+)
 
+# -------------------------------------------------------
+# Request Model
+# -------------------------------------------------------
 class RecommendRequest(BaseModel):
-    query: str = Field(..., min_length=2, description="What the user wants to buy")
-    budget_max: float | None = Field(None, description="Optional maximum budget")
-
+    query: str = Field(..., min_length=2)
+    budget_max: float | None = None
 
+# -------------------------------------------------------
+# Health API
+# -------------------------------------------------------
 @app.get("/api/health")
 def health():
     return {
@@ -49,34 +68,64 @@ def health():
         "serpapi_configured": bool(SERPAPI_KEY),
     }
 
-
+# -------------------------------------------------------
+# Recommendation API
+# -------------------------------------------------------
 @app.post("/api/recommend")
 def recommend(payload: RecommendRequest):
+
     if not GEMINI_API_KEY:
         raise HTTPException(
             status_code=400,
-            detail="GEMINI_API_KEY is not configured on the server. "
-                   "Add a free key from https://aistudio.google.com/apikey to your .env file.",
+            detail="GEMINI_API_KEY not configured."
         )
-    try:
-        result = run_pipeline(payload.query, payload.budget_max)
-    except Exception as exc:  # noqa: BLE001
-        logger.exception("Pipeline failed")
-        raise HTTPException(status_code=500, detail=f"Pipeline error: {exc}") from exc
-
-    return {
-        "query": payload.query,
-        "used_live_data": result.get("used_live_data", False),
-        "errors": result.get("errors", []),
-        "results": result.get("ranked", []),
-    }
 
+    try:
+        result = run_pipeline(
+            payload.query,
+            payload.budget_max
+        )
 
-# --- Serve the static frontend ---
-FRONTEND_DIR = os.path.join(os.path.dirname(__file__), "..", "frontend")
+        return {
+            "query": payload.query,
+            "used_live_data": result.get("used_live_data", False),
+            "errors": result.get("errors", []),
+            "results": result.get("ranked", []),
+        }
 
-app.mount("/frontend", StaticFiles(directory=FRONTEND_DIR), name="frontend")
+    except Exception as e:
+        logger.exception("Pipeline failed")
+        raise HTTPException(
+            status_code=500,
+            detail=str(e)
+        )
 
+# -------------------------------------------------------
+# Frontend
+# -------------------------------------------------------
 @app.get("/")
-def home():
-    return FileResponse(os.path.join(FRONTEND_DIR, "index.html"))
+def index():
+    return FileResponse(
+        os.path.join(FRONTEND_DIR, "index.html"),
+        media_type="text/html",
+        headers={
+            "Cache-Control": "no-cache"
+        },
+    )
+
+# -------------------------------------------------------
+# Fallback routes (handles cached requests)
+# -------------------------------------------------------
+@app.get("/style.css")
+def style():
+    return FileResponse(
+        os.path.join(FRONTEND_DIR, "style.css"),
+        media_type="text/css"
+    )
+
+@app.get("/script.js")
+def script():
+    return FileResponse(
+        os.path.join(FRONTEND_DIR, "script.js"),
+        media_type="application/javascript"
+    )