Update app.py

app.py

@@ -1,20 +1,34 @@
+import os
+import git
 import streamlit as st
 
-from agents.crew import build_crew
 from memory.code_indexer import index_codebase
 
 st.title("AI Codebase Intelligence")
 
-repo_path = st.text_input("Repository Path")
-query = st.text_area("Ask your coding question")
+repo_url = st.text_input("GitHub Repository URL")
 
-if st.button("Run Analysis"):
-    vector_db = index_codebase(repo_path)
+query = st.text_area("Ask Question")
 
-    crew = build_crew(vector_db)
 
-    result = crew.kickoff(
-        inputs={"query": query}
+if st.button("Analyze Repository"):
+
+    repo_name = repo_url.split("/")[-1]
+
+    if not os.path.exists(repo_name):
+        git.Repo.clone_from(
+            repo_url,
+            repo_name
+        )
+
+    st.success("Repository cloned")
+
+    vector_db = index_codebase(repo_name)
+
+    docs = vector_db.similarity_search(
+        query,
+        k=3
     )
 
-    st.write(result)
+    for doc in docs:
+        st.code(doc.page_content[:1500])