Update app.py

app.py

@@ -11,31 +11,33 @@ st.title("🧠 Codebase RAG Assistant")
 repo_url = st.text_input("GitHub Repository URL")
 
 if st.button("Index Repository"):
-    response = requests.post(
+    r = requests.post(
         f"{BACKEND_URL}/load",
         params={"repo_url": repo_url}
     )
-    if response.status_code == 200:
+    if r.status_code == 200:
         st.success("Repository indexed successfully!")
     else:
-        st.error("Failed to index repository")
+        st.error("Indexing failed")
 
 question = st.text_input("Ask a question about the codebase")
 
-if st.button("Ask"):
-    response = requests.get(
+ask_clicked = st.button("Ask")
+
+if ask_clicked:
+    r = requests.get(
         f"{BACKEND_URL}/ask",
         params={"question": question}
     )
 
-    if response.status_code == 200:
-        data = response.json()
+    if r.status_code != 200:
+        st.error("Backend error")
+    else:
+        data = r.json()
 
-        st.write("### Answer")
+        st.markdown("### Answer")
         st.write(data["answer"])
 
-        st.write("### Sources")
+        st.markdown("### Sources")
         for src in data["sources"]:
             st.write(src)
-    else:
-        st.error("Failed to get answer from backend")