Update app.py

app.py

@@ -400,7 +400,10 @@ def compare_embeddings(file, query, model_types, model_names, split_strategy, ch
         # Storing embeddings into the results for future use
         for doc in results_raw:
             print(doc)  # or print(dir(doc)) to see available attributes
-        result_embeddings = [doc.metadata['embedding'] for doc in results_raw]  # Adjust this based on the actual attribute names
+
+        #embedding = doc.metadata.get('embedding', None)  # Use .get() to avoid KeyError
+
+        result_embeddings = [doc.metadata.get('embedding', None) for doc in results_raw]  # Adjust this based on the actual attribute names
 #        result_embeddings = [doc['embedding'] for doc in results_raw]  # Assuming each result has an embedding
 
         stats = calculate_statistics(results_raw, search_time, vector_store, num_tokens, embedding_model, query, top_k)