Add folder button, cmd_opts bugfix

Author: semjonsona

src/backbone.py

@@ -12,13 +12,19 @@
 
     def get_opt(name, default):
         from modules.shared import opts
-
         if hasattr(opts, name):
             return opts.__getattr__(name)
         return default
 
+    def get_cmd_opt(name, default):
+        """Get command line argument"""
+        from modules.shared import cmd_opts
+        if hasattr(cmd_opts, name):
+            return cmd_opts.__getattribute__(name)
+        return default
 
     def gather_ops():
+        """Parameters for depthmap generation"""
         from modules.shared import cmd_opts
         ops = {}
         if get_opt('depthmap_script_boost_rmax', None) is not None:
@@ -29,6 +35,7 @@ def gather_ops():
 
 
     def get_outpath():
+        """Get path where results are saved by default"""
         path = get_opt('outdir_samples', None)
         if path is None or len(path) == 0:
             path = get_opt('outdir_extras_samples', None)
@@ -87,6 +94,9 @@ def listfiles(dirname):
 
     def get_opt(name, default): return default  # Configuring is not supported
 
+
+    def get_cmd_opt(name, default): return default  # Configuring is not supported
+
     def gather_ops(): return {}  # Configuring is not supported
 
     def get_outpath(): return '.'

src/common_ui.py

@@ -217,6 +217,25 @@ def update_delault_net_size(model_type):
 
     return inp
 
+def open_folder_action():
+    # Adapted from stable-diffusion-webui
+    f = backbone.get_outpath()
+    if backbone.get_cmd_opt('hide_ui_dir_config', False):
+        return
+    if not os.path.exists(f) or not os.path.isdir(f):
+        raise "Couldn't open output folder"  # .isdir is security-related, do not remove!
+    import platform
+    import subprocess as sp
+    path = os.path.normpath(f)
+    if platform.system() == "Windows":
+        os.startfile(path)
+    elif platform.system() == "Darwin":
+        sp.Popen(["open", path])
+    elif "microsoft-standard-WSL2" in platform.uname().release:
+        sp.Popen(["wsl-open", path])
+    else:
+        sp.Popen(["xdg-open", path])
+
 def on_ui_tabs():
     inp = GradioComponentBundle()
     with gr.Blocks(analytics_enabled=False, title="DepthMap") as depthmap_interface:
@@ -260,6 +279,10 @@ def on_ui_tabs():
                                                        elem_id=f"depthmap_gallery").style(grid=4)
                         with gr.Column():
                             html_info = gr.HTML()
+                        folder_symbol = '\U0001f4c2'  # 📂
+                        gr.Button(folder_symbol, visible=not backbone.get_cmd_opt('hide_ui_dir_config', False)).click(
+                            fn=lambda: open_folder_action(), inputs=[], outputs=[],
+                        )
 
                     with gr.TabItem('3D Mesh'):
                         with gr.Group():
@@ -301,6 +324,7 @@ def on_ui_tabs():
                                 submit_vid = gr.Button('Generate Video', elem_id="depthmap_generatevideo",
                                                        variant='primary')
 
+
         inp += inp.enkey_tail()
 
         depthmap_mode_0.select(lambda: '0', None, inp['depthmap_mode'])
@@ -401,7 +425,7 @@ def run_generate(*inputs):
             inputimages.append(image)
             inputnames.append(os.path.splitext(img.orig_name)[0])
     elif depthmap_mode == '2':  # Batch from Directory
-        assert not backbone.get_opt('hide_ui_dir_config', False), '--hide-ui-dir-config option must be disabled'
+        assert not backbone.get_cmd_opt('hide_ui_dir_config', False), '--hide-ui-dir-config option must be disabled'
         if depthmap_batch_input_dir == '':
             return [], None, None, "Please select an input directory."
         if depthmap_batch_input_dir == depthmap_batch_output_dir:

src/depthmap_generation.py

@@ -253,7 +253,7 @@ def reload(self):
         """Undoes offload"""
         if self.offloaded:
             self.move_models_to(self.device)
-            self.offloaded = True
+            self.offloaded = False
 
     def move_models_to(self, device):
         if self.depth_model is not None: