Better error handling in tab

Author: semjon00

src/common_ui.py

@@ -474,7 +474,24 @@ def run_generate(*inputs):
 
     show_images = []
     inpainted_mesh_fi = mesh_simple_fi = None
-    for input_i, type, result in gen_obj:
+    msg = ""  # Empty string is never returned
+    while True:
+        try:
+            input_i, type, result = next(gen_obj)
+        except StopIteration:
+            # TODO: return more info
+            msg = '<h3>Successfully generated.</h3>'
+            break
+        except Exception as e:
+            traceback.print_exc()
+            msg = '<h3>' + 'ERROR: ' + str(e) + '</h3>' + '\n'
+            if 'out of GPU memory' not in msg:
+                msg +=\
+                    'Please report this issue ' \
+                    f'<a href="https://github.com/thygate/{REPOSITORY_NAME}/issues">here</a>. ' \
+                    'Make sure to provide the full stacktrace: \n'
+                msg += '<code style="white-space: pre;">' + traceback.format_exc() + '</code>'
+            break
         if type == 'simple_mesh':
             mesh_simple_fi = result
             continue
@@ -507,5 +524,4 @@ def run_generate(*inputs):
         if backbone.get_opt('depthmap_script_show_3d_inpaint', True):
             if inpainted_mesh_fi is not None and len(inpainted_mesh_fi) > 0:
                 display_mesh_fi = inpainted_mesh_fi
-    # TODO: return more info
-    return show_images, inpainted_mesh_fi, display_mesh_fi, 'Generated!'
+    return show_images, inpainted_mesh_fi, display_mesh_fi, msg.replace('\n', '<br>')

src/core.py

@@ -267,23 +267,27 @@ def core_generation_funnel(outpath, inputimages, inputdepthmaps, inputnames, inp
                 yield count, 'simple_mesh', meshsimple_fi
 
         print("Computing output(s) done.")
-    except RuntimeError as e:
-        # TODO: display in UI
-        if 'out of memory' in str(e):
-            suggestion = "ERROR: out of GPU memory, could not generate depthmap! " \
+    except Exception as e:
+        import traceback
+        if 'out of memory' in str(e).lower():
+            print(str(e))
+            suggestion = "out of GPU memory, could not generate depthmap! " \
                          "Here are some suggestions to work around this issue:\n"
             if inp[go.BOOST]:
                 suggestion += " * Disable BOOST (generation will be faster, but the depthmap will be less detailed)\n"
-            if backbone.USED_BACKBONE == backbone.BackboneType.WEBUI:
+            if backbone.USED_BACKBONE != backbone.BackboneType.STANDALONE:
                 suggestion += " * Run DepthMap in the standalone mode - without launching the SD WebUI\n"
             if device != torch.device("cpu"):
                 suggestion += " * Select CPU as the processing device (this will be slower)\n"
             if inp[go.MODEL_TYPE] != 6:
-                suggestion += " * Use a different model (this could reduce quality)\n"
+                suggestion +=\
+                    " * Use a different model (generally, more memory-consuming models produce better depthmaps)\n"
             if not inp[go.BOOST]:
                 suggestion += " * Reduce net size (this could reduce quality)\n"
-            print(f"{suggestion}")
+            print('Fail.\n')
+            raise Exception(suggestion)
         else:
+            print('Fail.\n')
             raise e
     finally:
         if backbone.get_opt('depthmap_script_keepmodels', True):