Update app.py

app.py

@@ -140,17 +140,24 @@ def plot_interactive_feature_scatter(input_data, feature_x, feature_y, n_samples
 
 
 # Function for anomaly examples (Optional feature row)
-def plot_anomaly_examples(input_data, n_samples, outliers_fraction):
-    n_outliers = int(outliers_fraction * n_samples)
-    rng = np.random.RandomState(42)
-    data, _ = make_blobs(n_samples=n_samples, random_state=0)
-    outliers = rng.uniform(low=-6, high=6, size=(n_outliers, 2))
-    plt.figure(figsize=(5, 5))
-    plt.scatter(data[:, 0], data[:, 1], alpha=0.8, c="blue", s=20, label="Inliers")
-    plt.scatter(outliers[:, 0], outliers[:, 1], alpha=0.8, c="red", s=20, label="Outliers")
-    plt.title("Anomaly Examples")
-    plt.legend()
-    return plt.gcf()
+def get_anomaly_samples():
+    """Returns formatted top, middle, and bottom 10 records based on anomaly score."""
+    sorted_df = df.sort_values("Anomaly_Score", ascending=False)
+    
+    # Top 10 anomalies
+    top_10 = sorted_df[sorted_df["Anomaly_Label"] == "Anomaly"].head(10)
+    
+    # Middle 10 (mix of anomalies and normal)
+    mid_start = len(sorted_df) // 2 - 50  # Get a broader middle slice
+    middle_section = sorted_df.iloc[mid_start: mid_start + 100]  # Consider a larger middle slice
+    middle_anomalies = middle_section[middle_section["Anomaly_Label"] == "Anomaly"].sample(n=5, random_state=42)
+    middle_normals = middle_section[middle_section["Anomaly_Label"] == "Normal"].sample(n=5, random_state=42)
+    middle_10 = pd.concat([middle_anomalies, middle_normals]).sort_values("Anomaly_Score", ascending=False)
+    
+    # Bottom 10 normal records
+    bottom_10 = sorted_df[sorted_df["Anomaly_Label"] == "Normal"].tail(10)
+    
+    return top_10, middle_10, bottom_10
 
 
 # Gradio Interface
@@ -206,15 +213,21 @@ with gr.Blocks() as demo:
         outputs=scatter_plot,
     )
 
-    # Anomaly Examples Visualization
-    gr.Markdown("### 3. Anomaly Examples Visualization")
-    anomaly_examples_button = gr.Button("Generate Anomaly Examples")
-    anomaly_plot = gr.Plot(label="Anomaly Examples")
-
-    anomaly_examples_button.click(
-        fn=plot_anomaly_examples,
-        inputs=[input_data, n_samples, outliers_fraction],
-        outputs=anomaly_plot,
+    with gr.Tab("Anomaly Samples"):
+        gr.HTML("<h3 style='text-align: center; font-size: 18px; font-weight: bold;'>Top 10 Records (Anomalies)</h3>")
+        top_table = gr.Dataframe(label="Top 10 Records")
+        
+        gr.HTML("<h3 style='text-align: center; font-size: 18px; font-weight: bold;'>Middle 10 Records (Mixed)</h3>")
+        middle_table = gr.Dataframe(label="Middle 10 Records")
+        
+        gr.HTML("<h3 style='text-align: center; font-size: 18px; font-weight: bold;'>Bottom 10 Records (Normal)</h3>")
+        bottom_table = gr.Dataframe(label="Bottom 10 Records")
+        
+        anomaly_samples_button = gr.Button("Show Anomaly Samples")
+        anomaly_samples_button.click(
+            get_anomaly_samples, 
+            outputs=[top_table, middle_table, bottom_table]
+        )
     )
 
 demo.launch(debug=True)