Update app.py

app.py

@@ -57,8 +57,17 @@ def plot_anomalies(df_test_value, data, anomalies):
     ax.set_xlabel("Time")
     ax.set_ylabel("Value")
     ax.set_title("Anomalous Data Points")
-    return fig, anomalous_data_indices
-                                      
+    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 master(file):
     # read file
     data = pd.read_csv(file, parse_dates=True, index_col="timestamp")
@@ -68,17 +77,11 @@ def master(file):
     # predict
     anomalies = get_anomalies(df_test_value)
     # plot anomalous data points
-    plot2, anomalous_data_indices = plot_anomalies(df_test_value, data, anomalies)
-    # convert indices to string
-    anomalous_data_indices_str = [str(idx) for idx in anomalous_data_indices]
+    plot2 = plot_anomalies(df_test_value, data, anomalies)
+    # format output
+    anomalous_data_indices_str = ", ".join(map(str, np.where(anomalies)[0]))
     return format_output(plot2, anomalous_data_indices_str)
 
-def format_output(plot, indices):
-    # Combine the plot and indices into a single output
-    fig, ax = plot
-    ax.text(0.5, -0.2, f"Anomalous Data Indices: {', '.join(indices)}", transform=ax.transAxes, fontsize=12, ha='center')
-    return fig
-
 outputs = gr.outputs.Image()
 
 iface = gr.Interface(