Update index.py

index.py

@@ -4,7 +4,7 @@ sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
 
 from fastapi import FastAPI, Form
 from fastapi.middleware.cors import CORSMiddleware
-from .cura import github_ingestion, vector_store, chatbot
+from cura import github_ingestion, vector_store, chatbot
 
 app = FastAPI(
     title="Mindify Chat API", description="API for Mindify Chat", version="0.1"
@@ -28,8 +28,7 @@ def query_chat_route(query: str = Form(...), repo_name: str = Form(...), token:
     collection_name = repo_name.replace("/", "_")
     if repo_name:
         files = github_ingestion.ingest_github_repo(repo_name=repo_name, access_token=token)
-        collection = vector_store.set_up_chromadb(collection_name=collection_name)
-        collection = vector_store.index_vector_store_chroma(collection=collection, files=files)
+        collection = vector_store.index_vector_store_chroma(collection_name=collection_name, files=files)
         if collection != None:
             query = vector_store.query_vector_store_chroma(collection=collection, query=query)
     else:
@@ -38,8 +37,8 @@ def query_chat_route(query: str = Form(...), repo_name: str = Form(...), token:
 
     return {"status": "success", "response": query}
 
+
 if __name__ == "__main__":
     import uvicorn
 
     uvicorn.run(app)
-