Update app.py

app.py

@@ -8,33 +8,38 @@ st.set_page_config(page_title="Codebase RAG Assistant")
 
 st.title("🧠 Codebase RAG Assistant")
 
+# Session state
 if "qa_chain" not in st.session_state:
     st.session_state.qa_chain = None
 
+# Input: GitHub repo
 repo_url = st.text_input("GitHub Repository URL")
 
 if st.button("Index Repository"):
-    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!")
-
+    if not repo_url:
+        st.warning("Please enter a GitHub repository URL")
+    else:
+        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!")
+
+# Input: question
 question = st.text_input("Ask a question about the codebase")
 
 if st.button("Ask"):
     if st.session_state.qa_chain is None:
         st.warning("Please index a repository first")
+    elif not question:
+        st.warning("Please enter a question")
     else:
         with st.spinner("Thinking..."):
-            result = st.session_state.qa_chain.invoke(question)
+            answer = st.session_state.qa_chain.invoke(question)
 
-            st.markdown("### Answer")
-            st.write(answer)
-            st.markdown("### Sources")
-            for doc in result["source_documents"]:
-                st.write(doc.metadata["file"])
+        st.markdown("### Answer")
+        st.write(answer)