Update app.py

app.py

@@ -1,3 +1,4 @@
+import psutil
 import gradio as gr
 import numpy as np
 import os
@@ -264,26 +265,28 @@ def process_satellite_images(red_file, green_file, blue_file, nir_file, batch_si
     
     return rgb_display, visualization, stats
 
+def update_cpu():
+    return f"CPU Usage: {psutil.cpu_percent()}%"
 
-# Create Gradio interface with default examples
-demo = gr.Interface(
-    fn=process_satellite_images,
-    inputs=[
-        gr.File(label="Red Channel (JP2)"),
-        gr.File(label="Green Channel (JP2)"),
-        gr.File(label="Blue Channel (JP2)"),
-        gr.File(label="NIR Channel (JP2)"),
-        gr.Slider(minimum=1, maximum=32, value=1, step=1, label="Batch Size", info="Higher values use more memory but process faster"),
-        gr.Slider(minimum=500, maximum=2000, value=1000, step=100, label="Patch Size", info="Size of image patches for processing"),
-        gr.Slider(minimum=100, maximum=500, value=300, step=50, label="Patch Overlap", info="Overlap between patches to avoid edge artifacts")
-    ],
-    outputs=[
-        gr.Image(label="Original RGB Image"),
-        gr.Image(label="Cloud Detection Visualization"),
-        gr.Textbox(label="Statistics")
-    ],
-    title="Satellite Cloud Detection",
-    description="""
+with gr.Blocks() as demo:
+    cpu_text = gr.Textbox(label="CPU Usage")
+    check_cpu_btn = gr.Button("Check CPU")
+    
+    # Attach the event handler using the click method
+    check_cpu_btn.click(fn=update_cpu, inputs=None, outputs=cpu_text)
+    
+
+# Define the CPU check function
+def check_cpu_usage():
+    """Check and return the current CPU usage."""
+    return f"CPU Usage: {psutil.cpu_percent()}%"
+
+# Create the Gradio application with Blocks
+with gr.Blocks(title="Satellite Cloud Detection") as demo:
+    # Add the description
+    gr.Markdown("""
+    # Satellite Cloud Detection
+    
     Upload separate JP2 files for Red, Green, Blue, and NIR channels to detect clouds in satellite imagery.
     
     This application uses the OmniCloudMask model to classify each pixel as:
@@ -293,11 +296,60 @@ demo = gr.Interface(
     - Cloud Shadow (3)
     
     The model works best with imagery at 10-50m resolution. For higher resolution imagery, downsampling is recommended.
-    """,
-    examples=[
-        ["jp2s/B04.jp2", "jp2s/B03.jp2", "jp2s/B02.jp2", "jp2s/B8A.jp2", 1, 1000, 300]
-    ]
-)
+    """)
+    
+    # Main cloud detection interface
+    with gr.Row():
+        with gr.Column():
+            # Input components
+            red_input = gr.Image(type="filepath", label="Red Channel (JP2)")
+            green_input = gr.Image(type="filepath", label="Green Channel (JP2)")
+            blue_input = gr.Image(type="filepath", label="Blue Channel (JP2)")
+            nir_input = gr.Image(type="filepath", label="NIR Channel (JP2)")
+            
+            batch_size = gr.Slider(minimum=1, maximum=32, value=1, step=1, 
+                                  label="Batch Size", 
+                                  info="Higher values use more memory but process faster")
+            patch_size = gr.Slider(minimum=500, maximum=2000, value=1000, step=100, 
+                                  label="Patch Size", 
+                                  info="Size of image patches for processing")
+            patch_overlap = gr.Slider(minimum=100, maximum=500, value=300, step=50, 
+                                     label="Patch Overlap", 
+                                     info="Overlap between patches to avoid edge artifacts")
+            
+            process_btn = gr.Button("Process Cloud Detection")
+            
+        with gr.Column():
+            # Output components
+            rgb_output = gr.Image(label="Original RGB Image")
+            cloud_output = gr.Image(label="Cloud Detection Visualization")
+            stats_output = gr.Textbox(label="Statistics")
+    
+    # CPU usage monitoring section
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("## System Monitoring")
+            cpu_button = gr.Button("Check CPU Usage")
+            cpu_output = gr.Textbox(label="CPU Usage")
+    
+    # Set up event handlers
+    process_btn.click(
+        fn=process_satellite_images,
+        inputs=[red_input, green_input, blue_input, nir_input, batch_size, patch_size, patch_overlap],
+        outputs=[rgb_output, cloud_output, stats_output]
+    )
+    
+    cpu_button.click(
+        fn=check_cpu_usage,
+        inputs=None,
+        outputs=cpu_output
+    )
+    
+    # Add examples
+    gr.Examples(
+        examples=[["jp2s/B04.jp2", "jp2s/B03.jp2", "jp2s/B02.jp2", "jp2s/B8A.jp2", 1, 1000, 300]],
+        inputs=[red_input, green_input, blue_input, nir_input, batch_size, patch_size, patch_overlap]
+    )
 
 # Launch the app
-demo.launch(debug=True)
+demo.launch()