Update app.py

app.py

@@ -524,25 +524,27 @@ def page3():
             st.error("File not found. Please check the file path.")
             return
     
+
         # Initialize CromA client
         chroma_client = chromadb.Client()
     
-        # Check if the collection already exists
-        existing_collections = chroma_client.list_collections()
-        if "Kosten_Strukturdaten0602204" not in existing_collections:
-            # Create the collection if it does not exist
+        try:
+            # Attempt to create the collection
             collection = chroma_client.create_collection(name="Kosten_Strukturdaten0602204")
-        else:
-            # Use the existing collection if it already exists
-            collection = chroma_client.get_collection(name="Kosten_Strukturdaten0602204")
-
-
-        # Add the extracted text from PDF to the Chroma collection
-        collection.add(
-            documents=[pdf_text],  
-            metadatas=[{"source": pdf_path3}],  # Add any relevant metadata for your document
-            ids=[("Kosten_Strukturdaten0602204")]  
-        )
+        except Exception as e:
+            # If the collection already exists, fetch the existing collection
+            if 'already exists' in str(e):
+                collection = chroma_client.get_collection(name="Kosten_Strukturdaten0602204")
+            else:
+                # If the error is due to some other reason, re-raise the exception
+                raise e
+    
+            # Add the extracted text from PDF to the Chroma collection
+            collection.add(
+                documents=[pdf_text],  
+                metadatas=[{"source": pdf_path3}],  # Add any relevant metadata for your document
+                ids=[("Kosten_Strukturdaten0602204")]  
+            )
 
         display_chat_history(st.session_state['chat_history_page3'])