update the app.py to handle the clear cache logic + show

app.py

@@ -4,12 +4,44 @@ from typing import Union, List
 from predict_dept_model import DepartmentPredictor
 from contextlib import asynccontextmanager
 from response_schema import ClassificationOutput, TextInput 
+from huggingface_hub import scan_cache_dir, HfApi
+
+# hf api
+api = HfApi()
+
+def clear_hf_cache(model_repo: str) -> None:
+    """
+    Clears the Hugging Face model cache for the specified repository.
+
+    Args:
+        model_repo (str): The model repository identifier (e.g., 'mr-kush/sambodhan-department-classification-model').
+    """
+    # Scan the cache directory
+    cache_info = scan_cache_dir()
+
+    # Find the repository in the cache
+    repo_info = next((repo for repo in cache_info.cached_repos if repo.repo_id == model_repo), None)
+
+    if repo_info:
+        # Delete all revisions of the repository
+        delete_strategy = cache_info.delete_revisions(*repo_info.revisions)
+        print(f"Will free {delete_strategy.expected_freed_size_str}")
+        delete_strategy.execute()
+        print("Cache deletion done.")
+    else:
+        print(f"Repository '{model_repo}' not found in cache.")
+
 
 
 
 # Setting up startup and shutdown logic
 @asynccontextmanager
 async def lifespan(app: FastAPI):
+
+    # Clear the Hugging Face model cache
+    clear_hf_cache("mr-kush/sambodhan-department-classification-model")
+
+    # Load the model 
     global predictor
     predictor = DepartmentPredictor()
     yield
@@ -39,7 +71,22 @@ def predict_department(input_data: TextInput):
 
 @app.get("/")
 def root():
-    return {"message": "Sambodhan Department Classification API is running."}
+    # Define the model repository ID
+    model_repo = "mr-kush/sambodhan-department-classification-model"
+
+    
+    # Fetch the latest commit hash (revision) from the model repository
+    latest_tag = api.list_repo_refs(repo_id=model_repo, repo_type="model").tags[0].name
+
+    return {
+        "message": "Sambodhan Department Classification API is running.",
+        "status": "Active" if predictor  else "Inactive"
+        "model_version": latest_tag
+    }
+
+
+
+    
 
 # if __name__ == "__main__":
 #     # Important for Hugging Face Spaces (port detection)