Update app.py

app.py

@@ -86,12 +86,12 @@ def get_scatter_plot(feature1, feature2):
     return "scatter_plot.png"
 
 def get_anomaly_samples():
-    """Returns top, middle, and bottom 10 records based on anomaly score."""
+    """Returns formatted top, middle, and bottom 10 records based on anomaly score."""
     sorted_df = df.sort_values("Anomaly_Score", ascending=False)
-    top_10 = sorted_df.head(10).to_dict(orient="records")
-    middle_10 = sorted_df.iloc[len(sorted_df) // 2 - 5: len(sorted_df) // 2 + 5].to_dict(orient="records")
-    bottom_10 = sorted_df.tail(10).to_dict(orient="records")
-    return {"Top 10 Anomalies": top_10, "Middle 10 Records": middle_10, "Bottom 10 Scores": bottom_10}
+    top_10 = sorted_df.head(10)
+    middle_10 = sorted_df.iloc[len(sorted_df) // 2 - 5: len(sorted_df) // 2 + 5]
+    bottom_10 = sorted_df.tail(10)
+    return top_10, middle_10, bottom_10
 
 # Create Gradio interface
 with gr.Blocks() as demo:
@@ -119,10 +119,15 @@ with gr.Blocks() as demo:
         scatter_button.click(get_scatter_plot, inputs=[feature1_dropdown, feature2_dropdown], outputs=scatter_image)
 
     with gr.Tab("Anomaly Samples"):
-        gr.Markdown("### View Anomaly Scores")
-        anomaly_samples_button = gr.Button("Get Anomaly Samples")
-        anomaly_samples_output = gr.JSON()
-        anomaly_samples_button.click(get_anomaly_samples, outputs=anomaly_samples_output)
+        gr.Markdown("### Anomaly Samples")
+        anomaly_samples_button = gr.Button("Show Anomaly Samples")
+        top_table = gr.Dataframe(label="Top 10 Records")
+        middle_table = gr.Dataframe(label="Middle 10 Records")
+        bottom_table = gr.Dataframe(label="Bottom 10 Records")
+        anomaly_samples_button.click(
+            get_anomaly_samples, 
+            outputs=[top_table, middle_table, bottom_table]
+        )
 
 # Launch the Gradio app
 demo.launch()