Update app.py

app.py

@@ -11,13 +11,14 @@ client_db = PersistentClient(path="./chromadb_directory/chromadb_file")
 # Load your collection
 collection = client_db.get_collection("my_collection")
 
-# Initialize the Hugging Face Inference Client
-inference_client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-
+# Function to retrieve documents from ChromaDB, ensuring results are strings
 def retrieve_from_chromadb(query):
     results = collection.query(query_texts=query, n_results=5)  # Adjust n_results as needed
-    return results['documents']
+    # Ensure each document is a string
+    documents = [str(doc) for doc in results['documents']]
+    return documents
 
+# Respond function for the chatbot
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -38,6 +39,8 @@ def respond(
 
     # Retrieve relevant documents from ChromaDB
     retrieved_docs = retrieve_from_chromadb(message)
+    
+    # Join the documents to create a context for the user query
     context = "\n".join(retrieved_docs) + "\nUser: " + message
     messages.append({"role": "user", "content": context})