Update app.py

app.py

@@ -53,11 +53,11 @@ def plot_anomalies(df_test_value, data, anomalies):
     df_subset = data.iloc[anomalous_data_indices]
     fig, ax = plt.subplots(figsize=(12, 6))
     data.plot(legend=False, ax=ax)
-    df_subset.plot(legend=False, ax=ax, color="r")
+    df_subset.plot(legend=False, ax=ax, color="r", marker="o", linestyle="None")
     ax.set_xlabel("Time")
     ax.set_ylabel("Value")
     ax.set_title("Anomalous Data Points")
-    return fig
+    return fig, anomalous_data_indices
                                       
 def master(file):
     # read file
@@ -77,10 +77,10 @@ def master(file):
     anomalies = get_anomalies(df_test_value)
     
     # plot anomalous data points
-    plot2 = plot_anomalies(df_test_value, data, anomalies)
+    plot2, anomalous_indices = plot_anomalies(df_test_value, data, anomalies)
     
     # Update Gradio interface to show the resulting plot
-    return plot2
+    return plot2, anomalous_indices
 
 outputs = gr.outputs.Image()
 
@@ -93,4 +93,5 @@ iface = gr.Interface(
     description="Anomaly detection of timeseries data."
 )
 
-iface.launch()
+iface.launch()
+