Update app.py

app.py

@@ -245,12 +245,14 @@ def process_input(gif_file, max_images, size, tool, lower_bound, upper_bound, pa
     """Process either uploaded GIF or fetched SDO images."""
     if gif_file:
         frames, error = extract_frames(gif_file.name)
+        total_images = 0  # No total images when using uploaded GIF
         if error:
-            return error, [], None, [], 0
+            return error, [], None, [], total_images, None
     else:
         frames = fetched_frames_state
+        _, _, total_images = fetch_sdo_images(max_images, size, tool)  # Get total_images
         if not frames:
-            return "No fetched frames available. Please fetch images first.", [], None, [], 0
+            return "No fetched frames available. Please fetch images first.", [], None, [], total_images, None
     
     # Preview all frames
     preview = [Image.fromarray(frame) for frame in frames] if frames else []
@@ -260,46 +262,23 @@ def process_input(gif_file, max_images, size, tool, lower_bound, upper_bound, pa
         frames, lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode, scale_to_512
     )
     
-    return report, results, gif_path, preview, len(frames)
-
-def load_demo_2(gif_file, max_images, size, tool, lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode, fetched_frames_state):
-    gif_file="./demo_gif.gif"
-    if gif_file:
-        frames, error = extract_frames(gif_file)
-        if error:
-            return error, [], None, [], 0
-    else:
-        frames = fetched_frames_state
-        if not frames:
-            return "No fetched frames available. Please fetch images first.", [], None, [], 0
-    
-    # Preview all frames
-    preview = [Image.fromarray(frame) for frame in frames] if frames else []
-    
-    # Analyze frames
-    report, results, gif_path = analyze_images(
-        frames, lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode
-    )
-    
-    return report, results, gif_path, preview, len(frames)
+    return report, results, gif_path, preview, total_images, gif_path
 
 def load_demo():
-    gif_file="./demo_gif.gif"
-    if gif_file:
+    """Load demo GIF for analysis."""
+    gif_file = "./demo_gif.gif"
+    if os.path.exists(gif_file):
         frames, error = extract_frames(gif_file)
+        total_images = 0  # Demo GIF, so no directory total
         if error:
-            return error, [], None, [], 0
+            return error, [], None, total_images
     else:
-        frames = fetched_frames_state
-        if not frames:
-            return "No fetched frames available. Please fetch images first.", [], None, [], 0
+        return "Demo GIF not found.", [], None, 0
+    
     preview = [Image.fromarray(frame) for frame in frames] if frames else []
-
-    message="Demo Loaded"
+    message = "Demo Loaded"
     return message, preview, frames, total_images
 
- 
-# Gradio Blocks interface
 # Gradio Blocks interface
 with gr.Blocks(title="Solar CME Detection") as demo:
     gr.Markdown("""