Minor Bugs Fixed

back_end/config.py

@@ -364,4 +364,4 @@ SUPERVISOR_SYSTEM_PROMPT = """
 """
 
 
-MAX_FILES_TO_CREATE_VECTOR_DB = 6000
+MAX_FILES_TO_CREATE_VECTOR_DB = 6000

back_end/core/embeddings.py

@@ -28,10 +28,7 @@ class _SFRCodeEmbeddings(Embeddings):
     def __init__(self, model_path='Salesforce/SFR-Embedding-Code-400M_R'):
         print("Loading local SFR Code Model to GPU via ST...")
 
-        #self.model = SentenceTransformer(model_path, device='cuda', trust_remote_code=True)
-        # Automatically detect the hardware
-        hardware_device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.model = SentenceTransformer(model_path, device=hardware_device, trust_remote_code=True)
+        self.model = SentenceTransformer(model_path, device='cuda', trust_remote_code=True)
         self.model.max_seq_length = 1024
         self.model[0].auto_model.register_forward_pre_hook(_inject_position_ids_hook, with_kwargs=True)
 
@@ -115,4 +112,4 @@ def get_vector_db() -> Chroma:
         embedding_function=local_embedding_fn,
         collection_name=CHROMA_COLLECTION_NAME,
     )
-    return vector_db
+    return vector_db

back_end/core/splitter.py

@@ -193,4 +193,4 @@ def custom_splitter(docs: List[Document],current_dir: Path) -> List[Document]:
     print(f"Original Files Processed : {len(docs)}")
     print(f"Total Chunks Generated   : {len(all_chunks)}")
 
-    return all_chunks
+    return all_chunks

back_end/main.py

@@ -1,5 +1,4 @@
 # main.py
-from fastapi.staticfiles import StaticFiles
 from fastapi import FastAPI
 from fastapi.responses import StreamingResponse
 from fastapi.middleware.cors import CORSMiddleware
@@ -190,11 +189,4 @@ async def chat_stream(request: ChatRequest):
         yield format_chat_chunk("end", "[END]")
         await asyncio.sleep(0.01)
 
-    return StreamingResponse(generate_response(), media_type="text/event-stream")
-
-# 1. Find the path to the front_end folder
-current_dir = os.path.dirname(os.path.abspath(__file__))
-frontend_dir = os.path.join(current_dir, "..", "front_end")
-
-# 2. Tell FastAPI to serve your index.html when people visit the main URL
-app.mount("/", StaticFiles(directory=frontend_dir, html=True), name="frontend")
+    return StreamingResponse(generate_response(), media_type="text/event-stream")

front_end/app.js

@@ -138,8 +138,7 @@ function startAnalysis() {
 // ─────────────────────────────────────────────
 async function connectToBackend(repoUrl) {
   try {
-   //const response = await fetch('http://localhost:8000/init-repo', {
-    const response = await fetch('/init-repo', {
+    const response = await fetch('http://localhost:8000/init-repo', {
       method:  'POST',
       headers: { 'Content-Type': 'application/json' },
       body:    JSON.stringify({ url: repoUrl }),
@@ -418,8 +417,7 @@ async function sendMessage() {
   let fullMarkdown  = '';
 
   try {
-    //const response = await fetch('http://localhost:8000/chat', {
-    const response = await fetch('/chat', {
+    const response = await fetch('http://localhost:8000/chat', {
       method:  'POST',
       headers: { 'Content-Type': 'application/json' },
       body:    JSON.stringify({ message: text }),