try chroma direct loading and wait and run

app.py

@@ -125,33 +125,30 @@ def load_documents(file_path: str, mode: str = "elements"):
     docs = loader.load()  
     return [doc.page_content for doc in docs]  
 
-def wait_for_chroma_server(host, port, retries=5, delay=5):  
-    url = f"http://{host}:{port}/"  
+def wait_for_chroma_server(client, retries=10, delay=0.5):  
     for _ in range(retries):  
         try:  
-            response = httpx.get(url)  
-            if response.status_code == 200:  
-                print("Chroma server is up and running!")  
-                return True  
-        except httpx.HTTPStatusError as e:  
-            print(f"Attempt to connect to Chroma server failed with status code: {e.response.status_code}")  
-        except httpx.RequestError as e:  
+            client.heartbeat()  
+            print("Chroma server is up and running!")  
+            return True  
+        except Exception as e:  
             print(f"Attempt to connect to Chroma server failed: {e}")  
         time.sleep(delay)  
     print("Failed to connect to Chroma server after multiple attempts.")  
     return False  
-    
-
+  
 def initialize_chroma(collection_name: str, embedding_function: MyEmbeddingFunction):  
     host = 'localhost'  
     port = 8000  
-    if not wait_for_chroma_server(host, port):  
-        raise ConnectionError("Could not connect to Chroma server. Ensure it is running.")  
   
     client = HttpClient(host=host, port=port, settings=Settings(allow_reset=True, anonymized_telemetry=False))  
-    client.reset() 
+      
+    if not wait_for_chroma_server(client):  
+        raise ConnectionError("Could not connect to Chroma server. Ensure it is running.")  
+      
+    client.reset()  # Empties and completely resets the database  
     collection = client.create_collection(collection_name)  
-    return client, collection
+    return client, collection  
   
 def add_documents_to_chroma(client, collection, documents: list, embedding_function: MyEmbeddingFunction):  
     for doc in documents: