Update app.py

app.py

@@ -172,6 +172,16 @@ def run_segmentation_editor(editor_data, model_choice):
         if region_np is None:
             return 0, None, f"No image provided.", gr.update(visible=False), None, None, 0.0
         
+        # Resize large images to prevent crashes
+        max_size = 1024
+        if region_np.shape[0] > max_size or region_np.shape[1] > max_size:
+            h, w = region_np.shape[:2]
+            if h > w:
+                new_h, new_w = max_size, int(w * max_size / h)
+            else:
+                new_h, new_w = int(h * max_size / w), max_size
+            region_np = cv2.resize(region_np, (new_w, new_h), interpolation=cv2.INTER_AREA)
+        
         # Process image format to RGB
         if len(region_np.shape) == 2:
             processed_image_np = cv2.cvtColor(region_np, cv2.COLOR_GRAY2RGB)
@@ -229,6 +239,16 @@ def run_segmentation_manual(image, model_choice, crop_coords):
             loaded_models[model_filename] = model
                 
         image_np = np.array(image)
+        
+        # Resize large images to prevent crashes
+        max_size = 1024
+        if image_np.shape[0] > max_size or image_np.shape[1] > max_size:
+            h, w = image_np.shape[:2]
+            if h > w:
+                new_h, new_w = max_size, int(w * max_size / h)
+            else:
+                new_h, new_w = int(h * max_size / w), max_size
+            image_np = cv2.resize(image_np, (new_w, new_h), interpolation=cv2.INTER_AREA)
                 
         # Apply crop if coordinates are provided
         if crop_coords and crop_coords.strip():
@@ -310,9 +330,35 @@ def update_viability_realtime(blue_threshold, stored_masks, stored_image_np):
     except Exception as e:
         return None, 0, 0, 0.0, f"Error updating viability: {str(e)}"
 
+# PWA Head HTML - Updated for nested zip structure
+pwa_head = """
+<link rel="manifest" href="/file=static.zip/static/manifest.json">
+<meta name="apple-mobile-web-app-capable" content="yes">
+<meta name="apple-mobile-web-app-status-bar-style" content="default">
+<meta name="apple-mobile-web-app-title" content="Cellpose Cell Counter">
+<meta name="viewport" content="width=device-width, initial-scale=1">
+<link rel="apple-touch-icon" href="/file=favicon.png">
+<link rel="icon" type="image/png" sizes="192x192" href="/file=favicon.png">
+<script>
+if ('serviceWorker' in navigator) {
+  window.addEventListener('load', function() {
+    navigator.serviceWorker.register('/file=static.zip/static/service-worker.js')
+      .then(function(registration) {
+        console.log('ServiceWorker registration successful');
+      }, function(err) {
+        console.log('ServiceWorker registration failed: ', err);
+      });
+  });
+}
+</script>
+"""
 
 # Create the Gradio interface
-with gr.Blocks(title="Cell Viability Counter with Trypan Blue Analysis", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(
+    title="Cell Viability Counter with Trypan Blue Analysis", 
+    theme=gr.themes.Soft(),
+    head=pwa_head  # Add PWA head elements
+) as demo:
     gr.Markdown("# Cell Viability Counter with Trypan Blue Analysis")
     gr.Markdown("Upload a trypan blue stained microscopy image and analyze cell viability. Dead cells appear blue, alive cells are unstained. For accurate cell confluency, crop the image to display only desired area.")