Update app/main.py

app/main.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, HTTPException, Header, Request
 from fastapi.responses import JSONResponse, HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
+from loguru import logger
 import sqlite3
 import sqlparse
 import os
@@ -104,20 +105,52 @@ async def create_session():
     sessions[session_id] = {"conn": create_session_db(), "domain": None}
     return {"session_id": session_id}
 
+
 @app.get("/api/databases")
 async def get_databases():
     questions_dir = os.path.join(BASE_DIR, "questions")
-    return {"databases": [f.replace(".json", "") for f in os.listdir(questions_dir) if f.endswith(".json")] if os.path.exists(questions_dir) else []}
+    logger.debug(f"Checking databases in directory: {questions_dir}")
+    if not os.path.exists(questions_dir):
+        logger.error(f"Questions directory not found: {questions_dir}")
+        return {"databases": []}
+    databases = [f.replace(".json", "") for f in os.listdir(questions_dir) if f.endswith(".json")]
+    logger.debug(f"Found databases: {databases}")
+    return {"databases": databases}
+
+from loguru import logger
+
+@app.get("/api/databases")
+async def get_databases():
+    questions_dir = os.path.join(BASE_DIR, "questions")
+    logger.debug(f"Checking databases in directory: {questions_dir}")
+    if not os.path.exists(questions_dir):
+        logger.error(f"Questions directory not found: {questions_dir}")
+        return {"databases": []}
+    databases = [f.replace(".json", "") for f in os.listdir(questions_dir) if f.endswith(".json")]
+    logger.debug(f"Found databases: {databases}")
+    return {"databases": databases}
 
 @app.post("/api/load-schema/{domain}")
 async def load_schema(domain: str, session_id: str = Header(...)):
-    if session_id not in sessions: raise HTTPException(status_code=401, detail="Invalid session")
+    logger.debug(f"Loading schema for domain: {domain}, session_id: {session_id}")
+    if session_id not in sessions:
+        logger.error(f"Invalid session: {session_id}")
+        raise HTTPException(status_code=401, detail="Invalid session")
     sessions[session_id] = {"conn": create_session_db(), "domain": domain}
     try:
-        sessions[session_id]["conn"].executescript(load_schema_sql(domain))
+        schema_sql = load_schema_sql(domain)
+        logger.debug(f"Schema SQL loaded for {domain}")
+        sessions[session_id]["conn"].executescript(schema_sql)
         sessions[session_id]["conn"].commit()
+        logger.info(f"Schema loaded successfully for {domain}")
+    except FileNotFoundError as e:
+        logger.error(f"Schema file not found: {str(e)}")
+        close_session_db(sessions[session_id]["conn"])
+        del sessions[session_id]
+        raise HTTPException(status_code=500, detail=str(e))
     except sqlite3.Error as e:
-        close_session_db(sessions[session_id]["conn"])  # Cleanup on error
+        logger.error(f"Database error: {str(e)}")
+        close_session_db(sessions[session_id]["conn"])
         del sessions[session_id]
         raise HTTPException(status_code=500, detail=f"Database error: {str(e)}")
     return {"message": f"Database {domain} loaded"}