Update app.py

app.py

@@ -192,20 +192,30 @@ def compare_embeddings(file, query, model_types, model_names, split_strategy, ch
     return all_results + all_stats
 
 def format_results(results, stats):
-    df = pd.DataFrame([
-        {
-            "Content": doc.page_content, 
-            **doc.metadata
-
-            #"Source": doc.metadata.get("source", "Unknown"),
-            #"Relevance Score": doc.metadata.get("score", "N/A")
-        } for doc in results
-    ])
-    
+    # List to store the processed document data
+    data = []
+
+    # Extracting content and metadata from each document
+    for doc in results:
+        # Ensure metadata is a dictionary (if it's a custom object, convert it)
+        metadata_dict = dict(doc.metadata)
+
+        # Create a combined dictionary with 'Content' and all metadata fields
+        doc_data = {"Content": doc.page_content}
+        doc_data.update(metadata_dict)  # Add all metadata key-value pairs
+
+        # Append the processed document data to the list
+        data.append(doc_data)
+
+    # Convert the list of document data into a DataFrame
+    df = pd.DataFrame(data)
+
+    # Formatting stats as a DataFrame
     formatted_stats = pd.DataFrame([stats])
     
     return df, formatted_stats
 
+
 # Gradio interface
 iface = gr.Interface(
     fn=compare_embeddings,