Updated with Storage Monitor

app.py

@@ -1,59 +1,107 @@
-import sys
-from pathlib import Path
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.staticfiles import StaticFiles
-from fastapi.responses import FileResponse
-
-# add project root to Python path
-BASE_DIR = Path(__file__).resolve().parent
-sys.path.append(str(BASE_DIR))
-
-# import backend modules
-from backend.models import ChatRequest, ChatResponse
-from src.chatbot import RAGChatBot
-
-# Initialize chatbot
-bot = RAGChatBot()
-
-# Log FAISS index size safely
-try:
-    size = getattr(bot, "index_size", None)
-    if size is None and hasattr(bot, "index"):
-        # if RAGChatBot has .index (FAISS object) instead
-        size = len(bot.index) if hasattr(bot.index, "__len__") else "unknown"
-    print(f"✅ FAISS index size: {size}")
-except Exception as e:
-    print(f"⚠️ Could not determine FAISS index size: {e}")
-
-# Create FastAPI app
-app = FastAPI(title="Odisha Disaster Chatbot")
-
-# Enable CORS
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# Serve frontend static files
-frontend_dir = BASE_DIR / "frontend"
-app.mount("/static", StaticFiles(directory=str(frontend_dir)), name="static")
-
-# Serve index.html at root
-@app.get("/")
-def read_index():
-    return FileResponse(frontend_dir / "index.html")
-
-# Backend chat API
-@app.post("/api/chat", response_model=ChatResponse)
-def chat(request: ChatRequest):
-    answer = bot.chat(request.query)
-    return ChatResponse(answer=answer)
-
-# Optional health check
-@app.get("/health")
-def health():
-    return {"status": "ok"}
+import os
+import shutil
+
+# ===== STORAGE MONITOR =====
+def monitor_storage():
+    """Monitor disk usage on startup"""
+    try:
+        total, used, free = shutil.disk_usage("/")
+        used_gb = used // (2**30)
+        free_gb = free // (2**30)
+        total_gb = total // (2**30)
+        
+        print(f"\n{'='*60}")
+        print(f"💾 DISK USAGE:")
+        print(f"   Total: {total_gb} GB")
+        print(f"   Used:  {used_gb} GB")
+        print(f"   Free:  {free_gb} GB")
+        print(f"{'='*60}")
+        
+        # Check cache directories
+        cache_paths = {
+            "/app/.huggingface": "App Cache",
+            "/tmp/.huggingface": "Temp Cache",
+            "/app": "App Directory"
+        }
+        
+        for path, name in cache_paths.items():
+            if os.path.exists(path):
+                try:
+                    size = sum(
+                        os.path.getsize(os.path.join(root, file))
+                        for root, _, files in os.walk(path)
+                        for file in files
+                    )
+                    size_gb = size / (2**30)
+                    print(f"📦 {name} ({path}): {size_gb:.2f} GB")
+                except Exception as e:
+                    print(f"⚠️  Could not measure {name}: {e}")
+        
+        print(f"{'='*60}\n")
+        
+    except Exception as e:
+        print(f"⚠️  Storage monitoring failed: {e}")
+
+# Run storage check
+monitor_storage()
+# ===== END STORAGE MONITOR =====
+
+import sys
+from pathlib import Path
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+
+# add project root to Python path
+BASE_DIR = Path(__file__).resolve().parent
+sys.path.append(str(BASE_DIR))
+
+# import backend modules
+from backend.models import ChatRequest, ChatResponse
+from src.chatbot import RAGChatBot
+
+# Initialize chatbot
+bot = RAGChatBot()
+
+# Log FAISS index size safely
+try:
+    size = getattr(bot, "index_size", None)
+    if size is None and hasattr(bot, "index"):
+        # if RAGChatBot has .index (FAISS object) instead
+        size = len(bot.index) if hasattr(bot.index, "__len__") else "unknown"
+    print(f"✅ FAISS index size: {size}")
+except Exception as e:
+    print(f"⚠️ Could not determine FAISS index size: {e}")
+
+# Create FastAPI app
+app = FastAPI(title="Odisha Disaster Chatbot")
+
+# Enable CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Serve frontend static files
+frontend_dir = BASE_DIR / "frontend"
+app.mount("/static", StaticFiles(directory=str(frontend_dir)), name="static")
+
+# Serve index.html at root
+@app.get("/")
+def read_index():
+    return FileResponse(frontend_dir / "index.html")
+
+# Backend chat API
+@app.post("/api/chat", response_model=ChatResponse)
+def chat(request: ChatRequest):
+    answer = bot.chat(request.query)
+    return ChatResponse(answer=answer)
+
+# Optional health check
+@app.get("/health")
+def health():
+    return {"status": "ok"}