Update app.py

app.py

@@ -59,14 +59,21 @@ def plot_anomalies(df_test_value, data, anomalies):
     ax.set_title("Anomalous Data Points")
     return fig
 
-def format_output(plot, anomalous_data_indices_str):
-    if isinstance(plot, tuple):
-        fig, ax = plot
-    else:
-        fig = plot
-        ax = fig.axes[0]
-    ax.text(0, -0.1, "Anomalous Data Indices: " + anomalous_data_indices_str, transform=ax.transAxes)
-    return fig
+def format_output(plot, indices):
+    # Create a new figure and axis for the combined output
+    fig_combined, (ax_plot, ax_text) = plt.subplots(nrows=2, figsize=(12, 8), gridspec_kw={'height_ratios': [6, 1]})
+
+    # Add the plot to the top axis
+    fig, ax = plot
+    ax_plot.imshow(fig.canvas.renderer.buffer_rgba())
+    ax_plot.axis('off')
+
+    # Add the text to the bottom axis
+    ax_text.text(0.5, 0, f"Anomalous Data Indices: {', '.join(indices)}", fontsize=12, ha='center')
+    ax_text.axis('off')
+
+    return fig_combined, None
+
 
 def master(file):
     # read file