Update app.py

app.py

@@ -2,42 +2,73 @@ import streamlit as st
 import requests
 import time
 
-time.sleep(3)
-
 BACKEND_URL = "http://127.0.0.1:8000"
+HEALTH_URL = f"{BACKEND_URL}/health"
 
-st.title("🧠 Codebase RAG Assistant")
-
-repo_url = st.text_input("GitHub Repository URL")
+def wait_for_backend(timeout=30, interval=1):
+    """Wait until backend /health returns 200 or timeout (seconds)."""
+    t0 = time.time()
+    while True:
+        try:
+            r = requests.get(HEALTH_URL, timeout=1)
+            if r.status_code == 200:
+                return True
+        except Exception:
+            pass
+        if time.time() - t0 > timeout:
+            return False
+        time.sleep(interval)
 
-if st.button("Index Repository"):
-    r = requests.post(
-        f"{BACKEND_URL}/load",
-        params={"repo_url": repo_url}
-    )
-    if r.status_code == 200:
-        st.success("Repository indexed successfully!")
-    else:
-        st.error("Indexing failed")
+st.title("🧠 Codebase RAG Assistant (Demo)")
 
-question = st.text_input("Ask a question about the codebase")
+st.info("Waiting for backend... (this may take a few seconds after startup)")
+ready = wait_for_backend(timeout=45, interval=1)
 
-ask_clicked = st.button("Ask")
+if not ready:
+    st.error("Backend not available. Check space logs. Try restarting the Space.")
+    st.stop()
+else:
+    st.success("Backend available ✅")
 
-if ask_clicked:
-    r = requests.get(
-        f"{BACKEND_URL}/ask",
-        params={"question": question}
-    )
+repo_url = st.text_input("GitHub Repository URL", placeholder="https://github.com/your/repo")
 
-    if r.status_code != 200:
-        st.error("Backend error")
+if st.button("Index Repository"):
+    if not repo_url:
+        st.warning("Enter GitHub repo URL first")
     else:
-        data = r.json()
+        try:
+            r = requests.post(
+                f"{BACKEND_URL}/load",
+                params={"repo_url": repo_url},
+                timeout=120  # indexing can take time
+            )
+            if r.status_code == 200:
+                st.success("Repository indexed successfully!")
+            else:
+                st.error(f"Indexing failed: {r.status_code} {r.text}")
+        except requests.exceptions.RequestException as e:
+            st.error(f"Request failed: {e}")
 
-        st.markdown("### Answer")
-        st.write(data["answer"])
+question = st.text_input("Ask a question about the codebase")
 
-        st.markdown("### Sources")
-        for src in data["sources"]:
-            st.write(src)
+if st.button("Ask"):
+    if not question:
+        st.warning("Type a question first")
+    else:
+        try:
+            r = requests.get(
+                f"{BACKEND_URL}/ask",
+                params={"question": question},
+                timeout=60
+            )
+            if r.status_code == 200:
+                data = r.json()
+                st.markdown("### Answer")
+                st.write(data.get("answer", "No answer returned"))
+                st.markdown("### Sources")
+                for src in data.get("sources", []):
+                    st.write(src)
+            else:
+                st.error(f"Backend error: {r.status_code} {r.text}")
+        except requests.exceptions.RequestException as e:
+            st.error(f"Request failed: {e}")