Update app.py

app.py

@@ -60,4 +60,28 @@ IMAGE_FOLDER = 'images'
 image_uris = sorted([os.path.join(IMAGE_FOLDER, image_name) for image_name in os.listdir(IMAGE_FOLDER) if not image_name.endswith('.txt')])
 ids = [str(i) for i in range(len(image_uris))]
 
-collection_images.add(ids=ids, uris=image_uris)
+collection_images.add(ids=ids, uris=image_uris)
+
+# Path to the backup file
+load_path = 'text_collection_backup.json'
+
+# Load the data from the JSON file
+with open(load_path, 'r') as f:
+    loaded_data = json.load(f)
+
+# Extract the documents and IDs
+loaded_documents = loaded_data['documents']
+loaded_ids = loaded_data['ids']
+
+# Assuming 'client' is already set up for ChromaDB
+# Create or get a collection to store the loaded data
+# collection_text = client.create_collection(
+#     name='collection_text',  # Ensure the collection name is consistent if required
+#     embedding_function=default_ef  # Use the same embedding function as before
+# )
+# Add data to the collection
+collection_text.add(
+    documents=loaded_documents,
+    ids=loaded_ids
+)
+