Upload main.py

main.py

@@ -18,7 +18,10 @@ app = FastAPI(title="Medical Policy RAG Chatbot API")
 # --------------------------
 
 REDIS_URL = os.getenv("REDIS_URL", "redis://localhost:6379/0")
-redis_client = redis.from_url(REDIS_URL)
+redis_client = redis.from_url(REDIS_URL, socket_connect_timeout=1)
+
+# Local fallback for session data if Redis is unavailable
+local_cache = {}
 
 INDEX_PATH = os.getenv("FAISS_INDEX_PATH", "data/faiss.index")
 METADATA_PATH = os.getenv("METADATA_PATH", "data/metadata.pkl")
@@ -84,16 +87,27 @@ def generate_answer(prompt: str) -> str:
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 def store_pending(session_id: str, data: dict):
-    redis_client.setex(f"session:{session_id}:pending", 600, json.dumps(data))
+    try:
+        redis_client.setex(f"session:{session_id}:pending", 600, json.dumps(data))
+    except Exception:
+        local_cache[f"session:{session_id}:pending"] = json.dumps(data)
 
 def get_pending(session_id: str) -> Optional[dict]:
-    raw = redis_client.get(f"session:{session_id}:pending")
+    try:
+        raw = redis_client.get(f"session:{session_id}:pending")
+    except Exception:
+        raw = local_cache.get(f"session:{session_id}:pending")
+
     if raw:
         return json.loads(raw)
     return None
 
 def clear_pending(session_id: str):
-    redis_client.delete(f"session:{session_id}:pending")
+    try:
+        redis_client.delete(f"session:{session_id}:pending")
+    except Exception:
+        pass
+    local_cache.pop(f"session:{session_id}:pending", None)
 
 # --------------------------
 # Core endpoint