Update main.py

main.py

@@ -7,8 +7,7 @@ from huggingface_hub import snapshot_download
 from app.api.routes import router
 from app.core.config import settings
 
-# ====== Step 1: Safe Hugging Face cache location ======
-# For Hugging Face Spaces, use /tmp to ensure write access
+
 HF_CACHE_DIR = "/tmp/hf_cache"
 os.makedirs(HF_CACHE_DIR, exist_ok=True)
 
@@ -18,17 +17,16 @@ os.environ["HF_HUB_CACHE"] = HF_CACHE_DIR
 os.environ["HF_DATASETS_CACHE"] = HF_CACHE_DIR
 os.environ["HF_METRICS_CACHE"] = HF_CACHE_DIR
 
-# ====== Step 2: Logging setup ======
+
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 logger.info("Downloading model and dataset from Hugging Face Hub...")
 
-# ====== Step 3: Load your Hugging Face token from env ======
 HF_TOKEN = os.getenv("HF_TOKEN")
 if not HF_TOKEN:
     raise RuntimeError("HF_TOKEN environment variable not set")
 
-# ====== Step 4: Download model and FAISS index into /tmp ======
+
 MODEL_PATH = snapshot_download(
     repo_id="negi2725/legalBert",
     token=HF_TOKEN,
@@ -38,7 +36,7 @@ MODEL_PATH = snapshot_download(
 
 FAISS_INDEX_PATH = snapshot_download(
     repo_id="negi2725/dataRag",
-    repo_type="dataset",  # <- CRUCIAL!
+    repo_type="dataset", 
     token=HF_TOKEN,
     local_dir="/tmp/faiss_indexes"
 )
@@ -46,11 +44,11 @@ FAISS_INDEX_PATH = snapshot_download(
 logger.info(f"FAISS index files downloaded to: {FAISS_INDEX_PATH}")
 logger.info(f"Model files downloaded to: {MODEL_PATH}")
 
-# ====== Step 5: Make paths globally available ======
+
 os.environ["MODEL_PATH"] = MODEL_PATH
 os.environ["FAISS_INDEX_PATH"] = FAISS_INDEX_PATH
 
-# ====== Step 6: Initialize FastAPI ======
+
 app = FastAPI(
     title="Legal RAG Analysis API",
     description="FastAPI backend for legal case analysis using RAG system with LegalBERT predictions and Gemini AI evaluation",
@@ -65,7 +63,7 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Register API routes
+
 app.include_router(router, prefix="/api/v1")
 
 @app.get("/")
@@ -76,7 +74,6 @@ async def root():
 async def health_check():
     return {"status": "healthy", "message": "API is running"}
 
-# ====== Step 7: Run locally if needed ======
 if __name__ == "__main__":
     uvicorn.run(
         "main:app",