Update app.py

app.py

@@ -8,7 +8,7 @@ import tempfile
 from PIL import Image
 import io
 from huggingface_hub import hf_hub_download
-import replicate
+
 
 HF_REPO_ID = "myang4218/cellposemodel"
 MODEL_OPTIONS = {
@@ -342,10 +342,10 @@ def run_segmentation_manual(image, model_choice, crop_coords):
         return 0, None, f"Error during segmentation: {str(e)}", gr.update(visible=False), None, None, 0.0
 
 def update_viability_realtime(blue_threshold, stored_masks, stored_image_np):
-    """
-    Updates viability assessment in real-time based on blue threshold.
-    Takes stored masks and image_np from state.
-    """
+    
+    stored_masks = unpack_array(stored_masks)
+    stored_image_np = unpack_array(stored_image_np)
+
     if stored_masks is None or stored_image_np is None:
         return None, 0, 0, 0.0, "Please run segmentation first."
     
@@ -389,8 +389,16 @@ if ('serviceWorker' in navigator) {
 }
 </script>
 """
+def pack_array(arr):
+    pil = Image.fromarray(arr.astype(np.uint8))
+    buf = io.BytesIO()
+    pil.save(buf, format="PNG")
+    return buf.getvalue()
+
+def unpack_array(data):
+    return np.array(Image.open(io.BytesIO(data)))
 
-# Create the Gradio interface
+# Gradio interface
 with gr.Blocks(
     title="CellposeCellCounter", 
     theme=gr.themes.Soft(),
@@ -403,6 +411,9 @@ with gr.Blocks(
     masks_state = gr.State(value=None)
     image_state = gr.State(value=None)
 
+    masks_state = pack_array(masks)
+    image_state = pack_array(processed_image_np)
+
     with gr.Tab("Image Editor (Draw Selection)"):
         gr.Markdown("### Draw selection and run segmentation")