Update app.py

app.py

@@ -76,16 +76,31 @@ async def setup_kb(kb_file: UploadFile = File(...)):
         # Prepare texts, ids, and metadata
         texts, ids, metadatas = [], [], []
         for i, item in enumerate(data):
-            text = item.get("text") or item.get("content") or ""
+            # Try multiple possible field names for text content
+            text = (item.get("answer") or 
+                   item.get("text") or 
+                   item.get("content") or "")
+            
             item_id = item.get("id") or str(i)
+            category = item.get("category", "")
             
             if not text:
                 print(f"⚠️ Skipping item {i} - no text content")
                 continue
+            
+            # Create a richer text for embedding by combining relevant fields
+            # This helps with better semantic search
+            combined_text = text
+            if category:
+                combined_text = f"Category: {category}. {text}"
                 
-            texts.append(text)
+            texts.append(combined_text)
             ids.append(str(item_id))
-            metadatas.append({"id": str(item_id), "original_index": i})
+            metadatas.append({
+                "id": str(item_id), 
+                "category": category,
+                "original_index": i
+            })
 
         if not texts:
             raise HTTPException(status_code=400, detail="No valid text content found in JSON.")