Update main.py

main.py

@@ -1,13 +1,10 @@
+from contextlib import asynccontextmanager
 from fastapi import FastAPI
-from pydantic import BaseModel
-from typing import List, Dict, Any, Optional
-import os
-import uvicorn
-import time
-import hashlib
 from document_processor import DocumentProcessor
 from models import *
-from contextlib import asynccontextmanager
+import time
+import hashlib
+
 # Initialize document processor
 processor = DocumentProcessor()
 
@@ -20,19 +17,14 @@ async def lifespan(app: FastAPI):
     yield
     # Shutdown events (if you need any cleanup)
     print("🛑 Shutting down application...")
-# Initialize FastAPI app
+
+# Create FastAPI app with lifespan handler
 app = FastAPI(
     title="Legal Document Analysis API", 
     version="1.0.0", 
     lifespan=lifespan  # Pass the lifespan handler here
 )
 
-
-@app.on_event("startup")
-async def startup_event():
-    """Initialize models on startup"""
-    await processor.initialize()
-
 @app.post("/analyze_document")
 async def analyze_document(data: AnalyzeDocumentInput):
     """Unified endpoint for complete document analysis"""
@@ -69,3 +61,4 @@ def summarize_batch(data: SummarizeBatchInput):
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)
+