add frame ui

Author: damp11113

main.py

@@ -87,6 +87,8 @@ def capture(self, _, __):
             frame_file = self.ProMan.frames_folder / f"{len(list(self.ProMan.frames_folder.glob('*.png'))):06d}.png"
             cv2.imwrite(str(frame_file), frame)  # Fix: Convert Path to string
 
+            self.refetch_frames_list()
+
     def render_capture(self):
         if not (self.isInitCap and self.isopenProject):
             return
@@ -102,6 +104,28 @@ def render_capture(self):
             texture_data = np.true_divide(data, 255.0)  # Normalize to [0, 1]
             dpg.set_value("texture_preview", texture_data)
 
+    def refetch_frames_list(self):
+        if not self.isopenProject:
+            return
+
+        frames = self.ProMan.list_frames()
+
+        for frame in frames:
+            frame_name = frame.name
+            frame_path = frame
+
+            with dpg.texture_registry():
+                width, height, channels, data = dpg.load_image(str(frame_path))
+
+                try:
+                    dpg.add_raw_texture(width, height, data, format=dpg.mvFormat_Float_rgb, tag=f"frame_texture_{frame_name}")
+                except Exception as e:
+                    continue
+
+            with dpg.group(horizontal=True, parent="frames_window_group"):
+                dpg.add_image(f"frame_texture_{frame_name}")
+                dpg.add_button(label="Delete", callback=lambda _, __: frame_path.unlink(), tag=f"frame_delete_{frame_name}")
+
     def create_project(self, _, __):
         project_name = dpg.get_value("new_project_name")
         project_fps = dpg.get_value("new_project_fps")
@@ -121,16 +145,19 @@ def create_project(self, _, __):
         try:
             self.ProMan.create_project()
             self.isopenProject = True
+            self.refetch_frames_list()
             Thread(target=self.start_camera).start()
         except Exception as e:
             dpg.show_item("dialog_window")
             dpg.set_value("dialog_window_title", "can't create project")
             dpg.set_value("dialog_window_text", str(traceback.format_exc()))
 
+
     def open_project(self, _, data):
         try:
             self.ProMan = ProjectManager.load_project(data["file_path_name"])
             self.isopenProject = True
+            self.refetch_frames_list()
             Thread(target=self.start_camera).start()
         except Exception as e:
             dpg.show_item("dialog_window")
@@ -195,7 +222,7 @@ def init(self):
 
         raw_data = array.array("f", texture_data)
 
-        with dpg.texture_registry():
+        with dpg.texture_registry(show=True):
             dpg.add_raw_texture(self.preview_size[0], self.preview_size[1], raw_data, format=dpg.mvFormat_Float_rgb, tag="texture_preview")
             dpg.add_raw_texture(self.preview_size[0], self.preview_size[1], raw_data, format=dpg.mvFormat_Float_rgb, tag="texture_playback")
 
@@ -244,6 +271,10 @@ def render(self):
 
             dpg.configure_item("effect_window", width=window_width, height=effect_height, pos=(0, effect_y))
 
+            # Frames window height at the full right side of the viewport
+            frames_width = viewport_width - window_width
+            dpg.configure_item("frames_window", width=frames_width, height=viewport_height-18, pos=(window_width, 18))
+
     def exit(self):
         self.close_project(None, None)
         dpg.destroy_context()

manager.py

@@ -144,3 +144,6 @@ def load_project(project_location):
             project_height=project_settings["height"],
             project_location=project_location
         )
+
+    def list_frames(self):
+        return sorted(self.frames_folder.glob("*.png"))

ui.py

@@ -31,6 +31,12 @@ def windows(self):
             dpg.add_text("Effect")
             dpg.add_group(horizontal=True, tag="effect_window_group")
 
+        with dpg.window(label="Frames", tag="frames_window", show=True, no_close=True, no_resize=True, no_title_bar=True, no_move=True):
+            dpg.add_text("Frames")
+            # add group for frames
+            dpg.add_group(horizontal=True, tag="frames_window_group")
+
+
         dpg.add_file_dialog(label="Open Project", tag="open_project_dialog", callback=self.app.open_project, directory_selector=True, show=False)
 
         with dpg.window(label="Preferences", tag="preferences_window", show=False):