Update app.py

app.py

@@ -32,23 +32,20 @@ COLLECTION_NAME = "knowledge_base"
 # KB Setup Endpoint
 # -------------------------------
 @app.post("/setup")
-def setup_kb(kb_file: UploadFile = File(...)):
+async def setup_kb(kb_file: UploadFile = File(...)):
     """
-    Loads a JSON file, generates embeddings, and populates a persistent ChromaDB knowledge base.
+    Uploads a JSON file, generates embeddings, and populates a persistent ChromaDB knowledge base.
     Only runs when explicitly called.
     """
-    if not os.path.exists(json_file_path):
-        raise HTTPException(status_code=404, detail=f"File not found: {json_file_path}")
-
     try:
-        # Load JSON data
-        with open(json_file_path, "r", encoding="utf-8") as f:
-            data = json.load(f)
+        # Read JSON directly from the uploaded file
+        content_bytes = await kb_file.read()
+        data = json.loads(content_bytes)
 
         if not isinstance(data, list):
             raise HTTPException(status_code=400, detail="JSON must contain a list of knowledge items.")
 
-        print(f"📘 Loaded {len(data)} items from {json_file_path}")
+        print(f"📘 Loaded {len(data)} items from uploaded file '{kb_file.filename}'")
 
         # Initialize encoder and Chroma client (only here)
         encoder = SentenceTransformer("all-MiniLM-L6-v2")
@@ -79,6 +76,8 @@ def setup_kb(kb_file: UploadFile = File(...)):
 
         return {"message": "Knowledge base successfully initialized.", "count": collection.count()}
 
+    except json.JSONDecodeError:
+        raise HTTPException(status_code=400, detail="Invalid JSON file.")
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Setup failed: {e}")