Restore full brain logic and fix dashboard URL

dashboard.html

@@ -412,7 +412,7 @@ SKU: JK-2024-WTR"></textarea>
 
             try {
                 // Determine API URL - expecting localhost for this demo
-                const response = await fetch('http://localhost:8000/generate-catalog', {
+                const response = await fetch('/generate-catalog', {
                     method: 'POST',
                     body: formData
                 });

main.py

@@ -1,43 +1,72 @@
 import os
-from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import HTMLResponse, JSONResponse
-from fastapi.staticfiles import StaticFiles
-from agents.visual_analyst import VisualAnalyst
 from dotenv import load_dotenv
-# Load environment variables
+
+# Import Agents
+from agents.visual_analyst import VisualAnalyst
+from agents.memory_agent import MemoryAgent
+from agents.writer_agent import WriterAgent
+
 load_dotenv()
 app = FastAPI()
-# Initialize Agent
-visual_agent = VisualAnalyst()
-# 1. READ THE DASHBOARD HTML FILE INTO MEMORY
+
+# Initialize All Agents
 try:
-    with open("dashboard.html", "r") as f:
-        dashboard_html = f.read()
-except FileNotFoundError:
-    dashboard_html = "<h1>Error: dashboard.html not found. Please ensure the file exists.</h1>"
-# 2. SERVE DASHBOARD AT ROOT (Home Page)
+    visual_agent = VisualAnalyst()
+    memory_agent = MemoryAgent()
+    writer_agent = WriterAgent()
+    
+    # Seed memory on startup
+    memory_agent.seed_database()
+    print("✅ All Agents Online")
+except Exception as e:
+    print(f"⚠️ Warning: Some agents failed to load: {e}")
+
+# 1. SERVE DASHBOARD AT ROOT
 @app.get("/", response_class=HTMLResponse)
 async def read_root():
-    return dashboard_html
-# 3. KEEP /dashboard ROUTE AS BACKUP
-@app.get("/dashboard", response_class=HTMLResponse)
-async def read_dashboard():
-    return dashboard_html
-@app.post("/analyze")
-async def analyze_merch(file: UploadFile = File(...)):
     try:
+        with open("dashboard.html", "r") as f:
+            return f.read()
+    except FileNotFoundError:
+        return "Error: dashboard.html not found"
+
+# 2. THE MAIN ORCHESTRATOR ENDPOINT
+@app.post("/generate-catalog")
+async def generate_catalog(file: UploadFile = File(...)):
+    try:
+        # A. Save Temp File
         os.makedirs("uploads", exist_ok=True)
         file_path = f"uploads/{file.filename}"
         with open(file_path, "wb") as f:
             f.write(await file.read())
-        result = await visual_agent.analyze_image(file_path)
+
+        # B. Visual Analysis (The Eyes)
+        visual_data = await visual_agent.analyze_image(file_path)
+        
+        # C. Memory Search (The Context)
+        # Create a search query from visual data
+        query = f"{visual_data.get('main_color', '')} {visual_data.get('product_type', 'product')}"
+        seo_keywords = memory_agent.retrieve_keywords(query)
+        
+        # D. Write Copy (The Brain)
+        listing = writer_agent.write_listing(visual_data, seo_keywords)
         
+        # Cleanup
         if os.path.exists(file_path):
             os.remove(file_path)
             
-        return JSONResponse(content=result)
+        # Return Full Data Structure
+        return JSONResponse(content={
+            "visual_data": visual_data,
+            "seo_keywords": seo_keywords,
+            "listing": listing
+        })
     except Exception as e:
+        print(f"Error: {e}")
         return JSONResponse(content={"error": str(e)}, status_code=500)
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)