Update app.py

app.py

@@ -53,11 +53,14 @@ llm = Llama.from_pretrained(
     filename="unsloth.Q4_K_M.gguf",
 )
 
-# ✅ Function to Retrieve Relevant Knowledge
+# ✅ Function to Retrieve Relevant Knowledge (Fixed Nested List Issue)
 def retrieve_context(query):
     query_embedding = embedder.encode(query).tolist()
     results = collection.query(query_embeddings=[query_embedding], n_results=2)
-    retrieved_texts = [doc for doc in results.get("documents", []) if doc]
+
+    # Flatten nested lists and ensure only strings are returned
+    retrieved_texts = [doc for sublist in results.get("documents", []) for doc in sublist if isinstance(doc, str)]
+    
     return "\n".join(retrieved_texts) if retrieved_texts else "No relevant data found."
 
 # ✅ Chatbot Function with ChromaDB-RAG