Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import numpy as np
 import cv2
 from PIL import Image
-import matplotlib.pyplot as plt
 import requests
 from datetime import datetime, timedelta
 import io
@@ -111,13 +110,12 @@ def create_gif(frames, output_path, duration=0.5):
     return output_path
 
 def handle_fetch(start_date, end_date, ident, size, tool):
-    """Fetch SDO images and return frames for preview."""
+    """Fetch SDO images and return frames for preview and state."""
     frames, error = fetch_sdo_images(start_date, end_date, ident, size, tool)
     if error:
-        return error, []
-    # Convert frames to PIL Images for preview
+        return error, [], frames
     preview_frames = [Image.fromarray(frame) for frame in frames]
-    return "Fetched {} images successfully.".format(len(frames)), preview_frames
+    return f"Fetched {len(frames)} images successfully.", preview_frames, frames
 
 def analyze_images(frames, lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode):
     """Analyze frames for concentric circles, highlighting growing series."""
@@ -171,10 +169,8 @@ def analyze_images(frames, lower_bound, upper_bound, param1, param2, center_tole
 
         # Prepare output based on display mode
         if display_mode == "All Frames":
-            # Show pure, unprocessed frames
             results = [Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_GRAY2RGB)) for frame in frames]
         elif display_mode == "Detected Frames":
-            # Show only frames with detected circles
             for c in all_circle_data:
                 output_frame = cv2.cvtColor(c["output_frame"], cv2.COLOR_GRAY2RGB)
                 cv2.circle(output_frame, c["center"], c["radius"], (0, 255, 0), 2)  # Green for detected
@@ -182,7 +178,6 @@ def analyze_images(frames, lower_bound, upper_bound, param1, param2, center_tole
                     cv2.circle(output_frame, c["center"], c["radius"] + 2, (255, 165, 0), 2)  # Orange for growing
                 results.append(Image.fromarray(output_frame))
         elif display_mode == "Both (Detected Replaces Original)":
-            # Show all frames, replacing detected frames with circles
             for i, frame in enumerate(frames):
                 if i + 1 in [c["frame"] for c in all_circle_data]:
                     for c in all_circle_data:
@@ -293,12 +288,7 @@ with gr.Blocks(title="Solar CME Detection") as demo:
     fetch_button.click(
         fn=handle_fetch,
         inputs=[start_date, end_date, ident, size, tool],
-        outputs=[report, preview],
-        js="() => {return {fetched_frames_state: []}}"
-    ).then(
-        fn=lambda report, preview_frames: (preview_frames, report),
-        inputs=[report, preview],
-        outputs=[fetched_frames_state, report]
+        outputs=[report, preview, fetched_frames_state]
     )
 
     # Analyze button action