From 438c582a7d34ee5e63e2404d44c4a39b3db5722c Mon Sep 17 00:00:00 2001
From: ReiHashimoto <42664619+ReiHashimoto@users.noreply.github.com>
Date: Wed, 13 Dec 2023 10:56:00 +0900
Subject: [PATCH 1/2] fix suite2p_spike_deconv

---
 studio/app/optinist/wrappers/suite2p/spike_deconv.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/studio/app/optinist/wrappers/suite2p/spike_deconv.py b/studio/app/optinist/wrappers/suite2p/spike_deconv.py
index ad17592ee..549ca05b4 100644
--- a/studio/app/optinist/wrappers/suite2p/spike_deconv.py
+++ b/studio/app/optinist/wrappers/suite2p/spike_deconv.py
@@ -43,7 +43,7 @@ def suite2p_spike_deconv(
         ).T
         roi_list.append(kargs)
 
-    nwbfile[NWBDATASET.ROI] = {function_id: {"roi_list": roi_list}}
+    nwbfile[NWBDATASET.ROI] = {function_id: roi_list}
 
     # Fluorenceを追加
     nwbfile[NWBDATASET.FLUORESCENCE] = {

From 4deb5509cff5feca8821c45301006bb9cfa4b137 Mon Sep 17 00:00:00 2001
From: ReiHashimoto <42664619+ReiHashimoto@users.noreply.github.com>
Date: Wed, 13 Dec 2023 11:39:45 +0900
Subject: [PATCH 2/2] output movie can be loaded with index

---
 studio/app/common/dataclass/image.py | 29 ++++++++++++++++++++--------
 studio/app/common/dataclass/utils.py | 13 ++++---------
 studio/app/common/routers/outputs.py |  6 ++++--
 3 files changed, 29 insertions(+), 19 deletions(-)

diff --git a/studio/app/common/dataclass/image.py b/studio/app/common/dataclass/image.py
index 54e529fee..80fd7443f 100644
--- a/studio/app/common/dataclass/image.py
+++ b/studio/app/common/dataclass/image.py
@@ -55,14 +55,27 @@ def data(self):
             return np.array(imageio.volread(self.path))
 
     def save_json(self, json_dir):
-        self.json_path = join_filepath([json_dir, f"{self.file_name}.json"])
-        JsonWriter.write_as_split(self.json_path, create_images_list(self.data))
-        JsonWriter.write_plot_meta(json_dir, self.file_name, self.meta)
+        if self.data.ndim < 3:
+            self.json_path = join_filepath([json_dir, f"{self.file_name}.json"])
+            JsonWriter.write_as_split(self.json_path, create_images_list(self.data))
+            JsonWriter.write_plot_meta(json_dir, self.file_name, self.meta)
 
     @property
     def output_path(self) -> OutputPath:
-        return OutputPath(
-            path=self.json_path,
-            type=OutputType.IMAGE,
-            max_index=len(self.data) if self.data.ndim == 3 else 1,
-        )
+        if self.data.ndim == 3:
+            # self.path will be a list if self.data got into else statement on __init__
+            if isinstance(self.path, list) and isinstance(self.path[0], str):
+                _path = self.path[0]
+            else:
+                _path = self.path
+            return OutputPath(
+                path=_path,
+                type=OutputType.IMAGE,
+                max_index=len(self.data),
+            )
+        else:
+            return OutputPath(
+                path=self.json_path,
+                type=OutputType.IMAGE,
+                max_index=1,
+            )
diff --git a/studio/app/common/dataclass/utils.py b/studio/app/common/dataclass/utils.py
index 0b82c5b94..ca81139da 100644
--- a/studio/app/common/dataclass/utils.py
+++ b/studio/app/common/dataclass/utils.py
@@ -4,16 +4,11 @@
 
 
 def create_images_list(data):
-    if len(data.shape) == 2:
-        save_data = copy.deepcopy(data)
-    elif len(data.shape) == 3:
-        save_data = copy.deepcopy(data[:10])
-    else:
-        assert False, "data is error"
+    assert len(data.shape) == 2, "data is error"
 
-    if len(data.shape) == 2:
-        data = data[np.newaxis, :, :]
-        save_data = save_data[np.newaxis, :, :]
+    save_data = copy.deepcopy(data)
+    data = data[np.newaxis, :, :]
+    save_data = save_data[np.newaxis, :, :]
 
     images = []
     for _img in save_data:
diff --git a/studio/app/common/routers/outputs.py b/studio/app/common/routers/outputs.py
index 15ea454d5..850ed4d4c 100644
--- a/studio/app/common/routers/outputs.py
+++ b/studio/app/common/routers/outputs.py
@@ -118,11 +118,13 @@ async def get_image(
     filepath: str,
     workspace_id: str,
     start_index: Optional[int] = 0,
-    end_index: Optional[int] = 1,
+    end_index: Optional[int] = 10,
 ):
     filename, ext = os.path.splitext(os.path.basename(filepath))
     if ext in ACCEPT_TIFF_EXT:
-        filepath = join_filepath([DIRPATH.INPUT_DIR, workspace_id, filepath])
+        if not filepath.startswith(join_filepath([DIRPATH.OUTPUT_DIR, workspace_id])):
+            filepath = join_filepath([DIRPATH.INPUT_DIR, workspace_id, filepath])
+
         save_dirpath = join_filepath(
             [
                 os.path.dirname(filepath),