Update app.py

app.py

@@ -29,25 +29,40 @@ config = Settings(
     chroma_db_impl="sqlite",
 )
 
-# Initialize ChromaDB client
+
+import chromadb
+
+# Initialize persistent client with SQLite
 def setup_chromadb():
-    try:
-        client = chromadb.Client(config)
-        collections = client.list_collections()
-        print(f"Existing collections: {collections}")
-        if "pdf_data" in [c.name for c in collections]:
-            client.delete_collection("pdf_data")
-            print("Existing collection 'pdf_data' deleted.")
-        collection = client.create_collection(
-            "pdf_data",
-            embedding_function=chromadb.utils.embedding_functions.SentenceTransformerEmbeddingFunction(
-                model_name="sentence-transformers/all-MiniLM-L6-v2"
-            ),
-        )
-        return client, collection
-    except Exception as e:
-        print("Error setting up ChromaDB:", e)
-        raise e
+    client = chromadb.PersistentClient(path="./chromadb_data")
+    collection = client.get_or_create_collection(
+        name="pdf_data",
+        embedding_function=chromadb.utils.embedding_functions.SentenceTransformerEmbeddingFunction(
+            model_name="sentence-transformers/all-MiniLM-L6-v2"
+        ),
+    )
+    return client, collection
+
+
+# Initialize ChromaDB client
+# def setup_chromadb():
+#     try:
+#         client = chromadb.Client(config)
+#         collections = client.list_collections()
+#         print(f"Existing collections: {collections}")
+#         if "pdf_data" in [c.name for c in collections]:
+#             client.delete_collection("pdf_data")
+#             print("Existing collection 'pdf_data' deleted.")
+#         collection = client.create_collection(
+#             "pdf_data",
+#             embedding_function=chromadb.utils.embedding_functions.SentenceTransformerEmbeddingFunction(
+#                 model_name="sentence-transformers/all-MiniLM-L6-v2"
+#             ),
+#         )
+#         return client, collection
+#     except Exception as e:
+#         print("Error setting up ChromaDB:", e)
+#         raise e
 
 
 # Step 2: Extract Text from PDF