Update app.py

app.py

@@ -58,34 +58,7 @@ def plot_anomalies(df_test_value, data, anomalies):
     ax.set_ylabel("Value")
     ax.set_title("Anomalous Data Points")
     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 = plt.figure(figsize=(12, 8))
-    gs = fig_combined.add_gridspec(2, 1, height_ratios=[6, 1])
-
-    # Add the plot to the top axis
-    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")
@@ -94,11 +67,9 @@ def master(file):
     plot1 = plot_test_data(df_test_value)
     # predict
     anomalies = get_anomalies(df_test_value)
-    # plot anomalous data points
+    #plot anomalous data points
     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)
+    return plot2
 
 outputs = gr.outputs.Image()
 
@@ -111,4 +82,4 @@ iface = gr.Interface(
     description="Anomaly detection of timeseries data."
 )
 
-iface.launch()
+iface.launch()