Update app.py

app.py

@@ -1,35 +1,42 @@
-
+import streamlit as st
+import requests
 import time
+
+# Wait for FastAPI backend to start
 time.sleep(3)
 
-import streamlit as st
-import requests
+BACKEND_URL = "http://127.0.0.1:8000"
 
 st.title("🧠 Codebase RAG Assistant")
 
 repo_url = st.text_input("GitHub Repository URL")
 
 if st.button("Index Repository"):
-    res = requests.post(
-    f"{BACKEND_URL}/load",
-    params={"repo_url": repo_url}
-)
-
-    st.success("Repository indexed!")
+    response = requests.post(
+        f"{BACKEND_URL}/load",
+        params={"repo_url": repo_url}
+    )
+    if response.status_code == 200:
+        st.success("Repository indexed successfully!")
+    else:
+        st.error("Failed to index repository")
 
 question = st.text_input("Ask a question about the codebase")
 
 if st.button("Ask"):
-    BACKEND_URL = "http://127.0.0.1:8000"
-
-res = requests.get(
-    f"{BACKEND_URL}/ask",
-    params={"question": question}
-).json()
-
-    st.write("### Answer")
-    st.write(res["answer"])
-
-    st.write("### Sources")
-    for src in res["sources"]:
-        st.write(src)
+    response = requests.get(
+        f"{BACKEND_URL}/ask",
+        params={"question": question}
+    )
+
+    if response.status_code == 200:
+        data = response.json()
+
+        st.write("### Answer")
+        st.write(data["answer"])
+
+        st.write("### Sources")
+        for src in data["sources"]:
+            st.write(src)
+    else:
+        st.error("Failed to get answer from backend")