Update app.py

app.py

@@ -1,6 +1,6 @@
 import json
 import re
-import os, requests
+import os, requests, time
 import fitz  # PyMuPDF  We use PyMuPDF (fitz) to capture hierarchy (section → subsection → subsubsection →  content/bullets).
 from collections import Counter
 from fastapi import FastAPI
@@ -591,11 +591,24 @@ def startup_event():
 #    return generate_rag_response(req.query, retriever, k=req.top_k)
 
 
+
+
+def wait_for_key(key_name="OPENAI_API_KEY", timeout=10):
+    for _ in range(timeout):
+        if os.getenv(key_name):
+            print(f"✅ {key_name} available.")
+            return True
+        print(f"⏳ Waiting for {key_name}...")
+        time.sleep(1)
+    print(f"❌ {key_name} not found after {timeout} seconds.")
+    return False
+    
 # =============================
 # Step 5: Chat Function
 # =============================
 def chat_fn(message, history):
     global retriever
+    wait_for_key()
     if retriever is None:
         return "⚠️ Retriever not initialized. Please rebuild or check vector DB."
     answer = generate_rag_response(message, retriever)