Update app.py

app.py

@@ -118,9 +118,11 @@ def detect_circles(frame_diff, image_center, center_tolerance, param1, param2, m
         return filtered_circles if filtered_circles else None
     return None
 
-def create_gif(frames, output_path, duration=0.5):
-    """Create a GIF from a list of frames."""
+def create_gif(frames, output_path, duration=0.5, scale_to_512=False):
+    """Create a GIF from a list of frames, optionally scaling to 512x512."""
     pil_frames = [Image.fromarray(frame) for frame in frames]
+    if scale_to_512:
+        pil_frames = [frame.resize((512, 512), Image.Resampling.LANCZOS) for frame in pil_frames]
     pil_frames[0].save(
         output_path,
         save_all=True,
@@ -138,7 +140,7 @@ def handle_fetch(max_images, size, tool):
     preview_frames = [Image.fromarray(frame) for frame in frames]
     return message, preview_frames, frames, total_images
 
-def analyze_images(frames, lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode):
+def analyze_images(frames, lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode, scale_to_512):
     """Analyze frames for concentric circles, highlighting growing series."""
     try:
         if not frames or len(frames) < 2:
@@ -233,13 +235,13 @@ def analyze_images(frames, lower_bound, upper_bound, param1, param2, center_tole
         if results:
             gif_frames = [np.array(img) for img in results]
             gif_path = "output.gif"
-            create_gif(gif_frames, gif_path)
+            create_gif(gif_frames, gif_path, scale_to_512=scale_to_512)
 
         return report, results, gif_path
     except Exception as e:
         return f"Error during analysis: {str(e)}", [], None
 
-def process_input(gif_file, max_images, size, tool, lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode, fetched_frames_state):
+def process_input(gif_file, max_images, size, tool, lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode, fetched_frames_state, scale_to_512):
     """Process either uploaded GIF or fetched SDO images."""
     if gif_file:
         frames, error = extract_frames(gif_file.name)
@@ -255,10 +257,11 @@ def process_input(gif_file, max_images, size, tool, lower_bound, upper_bound, pa
     
     # Analyze frames
     report, results, gif_path = analyze_images(
-        frames, lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode
+        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:
@@ -296,6 +299,7 @@ def load_demo():
     return message, preview, frames, total_images
 
  
+# Gradio Blocks interface
 # Gradio Blocks interface
 with gr.Blocks(title="Solar CME Detection") as demo:
     gr.Markdown("""
@@ -307,7 +311,6 @@ with gr.Blocks(title="Solar CME Detection") as demo:
     # State to store fetched frames
     fetched_frames_state = gr.State(value=[])
     with gr.Sidebar(open=False):
-
         gr.Markdown("### Analysis Parameters")
         size = gr.Textbox(label="Image Size", value="1024by960")
         tool = gr.Textbox(label="Instrument", value="ccor1")
@@ -323,20 +326,22 @@ with gr.Blocks(title="Solar CME Detection") as demo:
             value="Both (Detected Replaces Original)",
             label="Display Mode"
         )
+        scale_to_512 = gr.Checkbox(label="Scale Output GIF to 512x512", value=False)
+
     with gr.Row():
         with gr.Column():
             gr.Markdown("### Input Options")
             demo_btn = gr.Button("Load Demo")
             gif_input = gr.File(label="Upload Solar GIF (optional)", file_types=[".gif"])
             max_images = gr.Slider(minimum=1, maximum=100, value=10, step=1, label="Max Images to Fetch")
-
             fetch_button = gr.Button("Fetch Images from URL")
             analyze_button = gr.Button("Analyze")
 
         with gr.Column():
             gr.Markdown("### Outputs")
             report = gr.Textbox(label="Analysis Report", lines=10)
- 
+            gif_viewer = gr.Image(label="Output GIF Preview", type="filepath")
+
     with gr.Row():
         with gr.Column():
             gif_output = gr.File(label="Download Resulting GIF")
@@ -345,6 +350,7 @@ with gr.Blocks(title="Solar CME Detection") as demo:
         with gr.Column():
             total_images = gr.Textbox(label="Total Images Available in Directory", value="0")
             preview = gr.Gallery(label="Input Preview (All Frames)")
+
     # Fetch button action
     fetch_button.click(
         fn=handle_fetch,
@@ -357,14 +363,15 @@ with gr.Blocks(title="Solar CME Detection") as demo:
         inputs=[],
         outputs=[report, preview, fetched_frames_state, total_images]
     )
+
     # Analyze button action
     analyze_button.click(
         fn=process_input,
         inputs=[
             gif_input, max_images, size, tool,
-            lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode, fetched_frames_state
+            lower_bound, upper_bound, param1, param2, center_tolerance, morph_iterations, min_rad, display_mode, fetched_frames_state, scale_to_512
         ],
-        outputs=[report, gallery, gif_output, preview, total_images]
+        outputs=[report, gallery, gif_output, preview, total_images, gif_viewer]
     )
 
 if __name__ == "__main__":