Update app.py

app.py

@@ -1,74 +1,39 @@
 import streamlit as st
-import requests
-import time
+import os
+from ingest import load_repo, ingest_repo
+from vectorstore import create_vectorstore
+from rag_chain import build_rag_chain
 
-BACKEND_URL = "http://127.0.0.1:8000"
-HEALTH_URL = f"{BACKEND_URL}/health"
+st.set_page_config(page_title="Codebase RAG Assistant")
 
-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)
+st.title("🧠 Codebase RAG Assistant")
 
-st.title("🧠 Codebase RAG Assistant (Demo)")
+if "qa_chain" not in st.session_state:
+    st.session_state.qa_chain = None
 
-st.info("Waiting for backend... (this may take a few seconds after startup)")
-ready = wait_for_backend(timeout=45, interval=1)
-
-if not ready:
-    st.error("Backend not available. Check space logs. Try restarting the Space.")
-    st.stop()
-else:
-    st.success("Backend available ✅")
-
-repo_url = st.text_input("GitHub Repository URL", placeholder="https://github.com/your/repo")
+repo_url = st.text_input("GitHub Repository URL")
 
 if st.button("Index Repository"):
-    if not repo_url:
-        st.warning("Enter GitHub repo URL first")
-    else:
-        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}")
+    with st.spinner("Indexing repository..."):
+        path = load_repo(repo_url)
+        docs = ingest_repo(path)
+        vectorstore = create_vectorstore(docs)
+        st.session_state.qa_chain = build_rag_chain(
+            vectorstore,
+            os.getenv("GROQ_API_KEY")
+        )
+    st.success("Repository indexed successfully!")
 
 question = st.text_input("Ask a question about the codebase")
 
 if st.button("Ask"):
-    if not question:
-        st.warning("Type a question first")
+    if st.session_state.qa_chain is None:
+        st.warning("Please index a repository 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}")
+        with st.spinner("Thinking..."):
+            result = st.session_state.qa_chain(question)
+            st.markdown("### Answer")
+            st.write(result["result"])
+            st.markdown("### Sources")
+            for doc in result["source_documents"]:
+                st.write(doc.metadata["file"])