Update app.py

app.py

@@ -192,6 +192,19 @@ def download_and_save_as_txt(url, account_sid, auth_token):
     except Exception as err:
         print(f"An error occurred: {err}")
 
+
+def initialize_chroma():
+    try:
+        # Initialize Chroma
+        db = Chroma(persist_directory=CHROMA_PATH, embedding_function=get_embedding_function())
+        # Perform an initial operation to ensure it works
+        db.similarity_search_with_score("test query", k=1)
+        print("Chroma initialized successfully.")
+    except Exception as e:
+        print(f"Error initializing Chroma: {e}")
+
+initialize_chroma()
+
 def query_rag(query_text: str):
     embedding_function = get_embedding_function()
     db = Chroma(persist_directory=CHROMA_PATH, embedding_function=embedding_function)
@@ -214,6 +227,7 @@ def query_rag(query_text: str):
             response += chunk.choices[0].delta.content
     
     return response.replace("###", '').replace('\nUser:', '')
+    
 def download_file(url, extension):
     try:
         response = requests.get(url)