Upload folder using huggingface_hub

main.py

@@ -23,17 +23,12 @@ logging.basicConfig(level=logging.INFO)
 
 @asynccontextmanager
 async def fn_lifespan(app: FastAPI):
-    # 🟢 Startup logic
     logging.info("🚀 Starting Firestore listener...")
     start_firestore_listener()
+    yield
+    logging.info("🛑 Firestore listener shutdown (no explicit cleanup needed).")
     
-    yield  # 👈 control returns to FastAPI app during runtime
-    
-    # 🔴 Shutdown logic (optional)
-    logging.info("🛑 Shutting down Firestore listener...")
-    # stop_firestore_listener() if you have one
-
-app = FastAPI(title="Sanatan AI Unified Server")
+app = FastAPI(title="Sanatan AI Unified Server",lifespan=fn_lifespan)
 limiter = Limiter(key_func=get_remote_address)
 app.state.limiter = limiter
 app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)

modules/firebase/listener.py

@@ -1,3 +1,5 @@
+import logging
+import threading
 import time
 from firebase_admin import firestore as admin_firestore
 import asyncio
@@ -51,13 +53,19 @@ def handle_changes(doc_snapshot, changes, read_time):
 
 
 def start_firestore_listener():
-    collection_ref = db.collection("work_updates")
-    collection_ref.on_snapshot(handle_changes)
-    print("👂 Listening for Firestore changes...")
-
-    # Keep this thread alive
-    while True:
-        time.sleep(60)
-
+    """Start Firestore listener in a background thread."""
+    def _listen():
+        collection_ref = db.collection("work_updates")
+        collection_ref.on_snapshot(handle_changes)
+        logging.info("👂 Firestore listener active.")
+        
+        # Keep the thread alive to maintain snapshot streaming
+        while True:
+            time.sleep(60)
+
+    thread = threading.Thread(target=_listen, daemon=True)
+    thread.start()
+    logging.info("🚀 Firestore listener thread started.")
+    
 # Run in background thread (so it doesn't block FastAPI)
 # threading.Thread(target=start_firestore_listener, daemon=True).start()