Update backend/ai/main.py

backend/ai/main.py

@@ -4,7 +4,7 @@ from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 import os
-from backend.ai.database import db  # singleton database
+import sqlite3
 
 app = FastAPI(title="VNPT AI Call Center MVP")
 
@@ -14,15 +14,32 @@ app.mount("/static", StaticFiles(directory="frontend"), name="static")
 # --- Templates ---
 templates = Jinja2Templates(directory="frontend")
 
-# --- Entry point: base.html với 3 tab ---
+# --- SQLite local database ---
+DB_FILE = os.path.join(os.path.dirname(__file__), "backend/ai/db.sqlite")
+
+# Tạo database nếu chưa có
+conn = sqlite3.connect(DB_FILE, check_same_thread=False)
+cursor = conn.cursor()
+cursor.execute("""
+    CREATE TABLE IF NOT EXISTS calls (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        name TEXT,
+        duration INTEGER
+    )
+""")
+conn.commit()
+
+# --- Database helper ---
+def get_all_calls():
+    cursor.execute("SELECT * FROM calls")
+    rows = cursor.fetchall()
+    return [{"id": r[0], "name": r[1], "duration": r[2]} for r in rows]
+
+# --- Routes HTML ---
 @app.get("/", response_class=HTMLResponse)
 async def base(request: Request):
-    """
-    Trang chính của app Space: base.html với 3 tab
-    """
     return templates.TemplateResponse("base.html", {"request": request})
 
-# --- Các tab HTML ---
 @app.get("/call_desktop", response_class=HTMLResponse)
 async def call_desktop(request: Request):
     return templates.TemplateResponse("call_desktop.html", {"request": request})
@@ -35,15 +52,13 @@ async def call_mobile(request: Request):
 async def dashboard(request: Request):
     return templates.TemplateResponse("dashboard.html", {"request": request})
 
-# --- API dashboard ---
+# --- API ---
 @app.get("/api/dashboard-stats")
 async def dashboard_stats():
     try:
-        calls = db.get_all_calls()
-        print(f"[DEBUG API] Lấy {len(calls)} bản ghi")
+        calls = get_all_calls()
         return {"status": "success", "data": calls}
     except Exception as e:
-        print(f"[DEBUG API Error] {e}")
         return {"status": "error", "message": str(e)}
 
 # --- Run app ---