Update app.py

app.py

@@ -81,16 +81,20 @@ def process_image(image):
     
     # Cell size distribution
     df = pd.DataFrame(features)
-    df['area'].hist(ax=axes[0], bins=20)
-    axes[0].set_title('Cell Size Distribution')
-    axes[0].set_xlabel('Area')
-    axes[0].set_ylabel('Count')
-    
-    # Circularity vs Intensity
-    axes[1].scatter(df['circularity'], df['mean_intensity'])
-    axes[1].set_title('Circularity vs Intensity')
-    axes[1].set_xlabel('Circularity')
-    axes[1].set_ylabel('Mean Intensity')
+    if not df.empty:
+        df['area'].hist(ax=axes[0], bins=20)
+        axes[0].set_title('Cell Size Distribution')
+        axes[0].set_xlabel('Area')
+        axes[0].set_ylabel('Count')
+        
+        # Circularity vs Intensity
+        axes[1].scatter(df['circularity'], df['mean_intensity'])
+        axes[1].set_title('Circularity vs Intensity')
+        axes[1].set_xlabel('Circularity')
+        axes[1].set_ylabel('Mean Intensity')
+    else:
+        axes[0].text(0.5, 0.5, 'No cells detected', ha='center')
+        axes[1].text(0.5, 0.5, 'No cells detected', ha='center')
     
     plt.tight_layout()
     
@@ -121,10 +125,10 @@ with gr.Blocks(title="Cell Analysis Tool") as demo:
     
     with gr.Row():
         with gr.Column():
+            # Fixed the Image component configuration
             input_image = gr.Image(
                 label="Upload Image",
-                type="numpy",
-                tool="upload"
+                type="numpy"
             )
             analyze_btn = gr.Button(
                 "Analyze Image",