Update app.py

app.py

@@ -35,8 +35,17 @@ def create_db(splits, collection_name, db_type):
         embedding = HuggingFaceEmbeddings(model_name="dbmdz/bert-base-italian-xxl-uncased")
     
     new_client = chromadb.EphemeralClient()
-    return Chroma.from_documents(documents=splits, embedding=embedding, client=new_client, collection_name=collection_name)
-
+    
+    # Check if collection already exists
+    try:
+        existing_collection = new_client.get_collection(collection_name)
+        # If collection exists, use the same embedding model
+        existing_embedding = existing_collection.embedding_function
+        return Chroma.from_documents(documents=splits, embedding=existing_embedding, client=new_client, collection_name=collection_name)
+    except ValueError:
+        # If collection doesn't exist, create a new one with the selected embedding
+        return Chroma.from_documents(documents=splits, embedding=embedding, client=new_client, collection_name=collection_name)
+        
 def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, progress=gr.Progress()):
     progress(0.5, desc="Initializing HF Hub...")
     llm = HuggingFaceEndpoint(