Update app.py

app.py

@@ -107,13 +107,20 @@ def get_roc_curve():
 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 (mixed records)
-    mid_start = len(sorted_df) // 2 - 5
-    middle_10 = sorted_df.iloc[mid_start: mid_start + 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
 
 # Create Gradio interface