Update backend/ai/main.py

backend/ai/main.py

@@ -1,38 +1,87 @@
-from fastapi import FastAPI, Request
-from fastapi.responses import HTMLResponse, FileResponse
-from fastapi.templating import Jinja2Templates
-from fastapi.middleware.cors import CORSMiddleware
+# main.py
 import os
+from fastapi import FastAPI, UploadFile, File, Form
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import FileResponse, JSONResponse, RedirectResponse
+import sys
+
+# ===== Thêm backend/ai vào Python path =====
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(BASE_DIR)
 
-app = FastAPI()
+try:
+    from core.logic_flow import agent
+    from core.database import db
+except ImportError as e:
+    print("❌ LỖI IMPORT: Không tìm thấy core.logic_flow hoặc core.database")
+    print(f"Chi tiết lỗi: {e}")
+    agent = None
+    class FakeDB:
+        def get_all_calls(self): return []
+        def update_call_rating(self, cid, s, n): pass
+    db = FakeDB()
 
+# ===== FastAPI app =====
+app = FastAPI(title="PINE-AI MVP")
+
+# ===== CORS =====
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"], 
+    allow_origins=["*"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# Template
-templates = Jinja2Templates(directory="./frontend")
+# ===== Mount frontend =====
+FRONTEND_DIR = os.path.join(os.path.dirname(BASE_DIR), "frontend")
+
+# ===== Root redirect =====
+@app.get("/")
+def root_redirect():
+    return RedirectResponse(url="/desktop")
+
+# ===== Tab routes =====
+@app.get("/desktop")
+def desktop():
+    return FileResponse(os.path.join(FRONTEND_DIR, "call_desktop.html"))
 
-FRONTEND_DIR = "./frontend"
+@app.get("/mobile")
+def mobile():
+    return FileResponse(os.path.join(FRONTEND_DIR, "call_mobile.html"))
 
-# --- Trang chủ Base ---
-@app.get("/", response_class=HTMLResponse)
-async def home(request: Request):
-    return templates.TemplateResponse("base.html", {"request": request})
+@app.get("/dashboard")
+def dashboard():
+    return FileResponse(os.path.join(FRONTEND_DIR, "dashboard.html"))
 
-# --- Các tab ---
-@app.get("/call_desktop", response_class=HTMLResponse)
-async def call_desktop(request: Request):
-    return templates.TemplateResponse("call_desktop.html", {"request": request})
+# ===== API endpoint cho Dashboard =====
+@app.get("/api/dashboard-stats")
+def dashboard_stats():
+    try:
+        calls_data = db.get_all_calls()
+        return JSONResponse({"status": "success", "data": calls_data})
+    except Exception as e:
+        return JSONResponse({"status": "error", "message": str(e)}, status_code=500)
 
-@app.get("/call_mobile", response_class=HTMLResponse)
-async def call_mobile(request: Request):
-    return templates.TemplateResponse("call_mobile.html", {"request": request})
+# ===== API endpoints cho call =====
+@app.post("/start-call")
+async def start_call(customer_id: str = Form(...)):
+    if not agent:
+        return {"error": "Chưa có logic_flow.py"}
+    return StreamingResponse(
+        agent.process_stream(customer_id, None),
+        media_type="application/x-ndjson"
+    )
 
-@app.get("/dashboard", response_class=HTMLResponse)
-async def dashboard(request: Request):
-    return templates.TemplateResponse("dashboard.html", {"request": request})
+@app.post("/chat-voice")
+async def chat_voice(
+    customer_id: str = Form(...),
+    file: UploadFile = File(...)
+):
+    if not agent:
+        return {"error": "Chưa có logic_flow.py"}
+    audio_bytes = await file.read()
+    return StreamingResponse(
+        agent.process_stream(customer_id, audio_bytes),
+        media_type="application/x-ndjson"
+    )