Update app.py

app.py

@@ -10,6 +10,8 @@ from datetime import datetime
 import traceback
 import logging
 from typing import List, Optional
+import time
+from fastapi.responses import JSONResponse
 
 # Setup logging
 logging.basicConfig(level=logging.INFO)
@@ -31,6 +33,7 @@ app.add_middleware(
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
+    max_age=600,
 )
 
 # Import processing functions
@@ -44,19 +47,46 @@ except ImportError as e:
     logger.error(f"Import error: {e}")
     DEPENDENCIES_LOADED = False
 
+@app.on_event("startup")
+async def startup_event():
+    """Handle startup events"""
+    logger.info("Application startup initiated")
+    # Load model on startup to avoid cold start delays
+    try:
+        from summarizer import get_summarizer
+        get_summarizer()  # Pre-load the model
+        logger.info("Models pre-loaded successfully")
+    except Exception as e:
+        logger.warning(f"Model pre-loading failed: {e}")    
+
 @app.get("/")
 async def root():
     return {"message": "Material Summarizer API", "status": "running"}
 
 @app.get("/health")
 async def health_check():
+    """Health check endpoint specifically for Hugging Face Spaces"""
     status = "healthy" if DEPENDENCIES_LOADED else "missing_dependencies"
-    return {
-        "status": status,
-        "service": "material-summarizer",
-        "dependencies_loaded": DEPENDENCIES_LOADED
-    }
+    return JSONResponse(
+        content={
+            "status": status,
+            "service": "material-summarizer",
+            "dependencies_loaded": DEPENDENCIES_LOADED,
+            "timestamp": time.time()
+        },
+        status_code=200 if DEPENDENCIES_LOADED else 503
+    )
+
+
+@app.get("/ping")
+async def ping():
+    """Simple ping endpoint for load balancers"""
+    return JSONResponse(
+        content={"status": "ok", "timestamp": time.time()},
+        status_code=200
+    )
 
+    
 @app.post("/summarize-document")
 async def summarize_document(
     file: UploadFile = File(...),