update

UFISH-Team · Apr 19, 2024 · f2350a3 · f2350a3
1 parent f6cab16
commit f2350a3
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 5 deletions.
diff --git a/public/plugins/chatbot-extension.imjoy.html b/public/plugins/chatbot-extension.imjoy.html
@@ -103,12 +103,21 @@
               type: "number",
               description: "Diameter of the cells",
               default: 60,
-            }
+            },
+            "model_type": {
+              type: "string",
+              description: "Model type, cyto or nuclei",
+              default: "cyto",
+            },
+            "flow_threshold": {
+              type: "number",
+              description: "Flow threshold",
+              default: 0.4,
+            },
           },
         };
       },
       execute: async (config) => {
-        debugger
         const client = await this.getUfishClient()
         const bioengine_runner = await this.getBioengineRunner()
         const img = await client.getInputImage()

diff --git a/public/plugins/ufish.imjoy.html b/public/plugins/ufish.imjoy.html
@@ -159,8 +159,9 @@
         await self.viewer.view_image(image, name=name)
         return image.shape
 
-    async def process_enhanced(self, enh_img):
-        await self.viewer.view_image(enh_img, name="enhanced")
+    async def process_enhanced(self, enh_img, view_enhanced=True):
+        if view_enhanced:
+            await self.viewer.view_image(enh_img, name="enhanced")
         df = call_spots_local_maxima(enh_img)
         coords = df.values
         self.enhanced_image = enh_img

diff --git a/src/views/PredictView.vue b/src/views/PredictView.vue
@@ -132,7 +132,8 @@ export default {
           _rshape: modelOut.dims,
           _rvalue: (modelOut.data as Float32Array).buffer
         }
-        const [enhBytes, coords, numSpots] = await plugin.value.process_enhanced(outImg)
+        const viewEnhanced = !isPluginMode()
+        const [enhBytes, coords, numSpots] = await plugin.value.process_enhanced(outImg, viewEnhanced)
         output.value = {
           enhanced: enhBytes,
           coords: coords