Update app.py

app.py

@@ -88,10 +88,10 @@ def process_pdf(file):
     chunks = text_splitter.split_text(text)
     
     # Create embeddings for each chunk
-    embeddings = embedding_function(chunks)
-    
+    #embeddings = embedding_function(chunks)
+    embeddings = embedding_function(text)
     # Store each chunk in ChromaDB
-    for i, chunk in enumerate(chunks):
+    for i, chunk in enumerate(text):
         collection.add(
             documents=[chunk],
             metadatas=[{"filename": file.name, "chunk_id": i}],
@@ -106,7 +106,7 @@ def search_similar_documents(prompt):
     # Führe die Ähnlichkeitssuche durch
     results = collection.query(
         query_embeddings=[query_embedding],
-        n_results=50
+        n_results=20
     )
     
     # Formatiere die Ergebnisse