Update app.py

app.py

@@ -60,21 +60,32 @@ def plot_anomalies(df_test_value, data, anomalies):
     return fig
 
 def format_output(plot, indices):
+    if plot is None:
+        return None, None
+
     # 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]})
+    fig_combined = plt.figure(figsize=(12, 8))
+    gs = fig_combined.add_gridspec(2, 1, height_ratios=[6, 1])
 
     # Add the plot to the top axis
-    fig, ax = plot
-    ax_plot.imshow(fig.canvas.renderer.buffer_rgba())
+    ax_plot = fig_combined.add_subplot(gs[0])
+    if isinstance(plot, tuple):
+        fig, ax = plot
+        ax.plot()
+        ax_plot.imshow(fig.canvas.renderer.buffer_rgba())
+    else:
+        ax_plot.imshow(plot)
     ax_plot.axis('off')
 
     # Add the text to the bottom axis
+    ax_text = fig_combined.add_subplot(gs[1])
     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
     data = pd.read_csv(file, parse_dates=True, index_col="timestamp")