From cfbd32363f27ad6a2d461fdeb6a1f8dc7d3315cc Mon Sep 17 00:00:00 2001
From: vahid khroasani <62482657+v8hid@users.noreply.github.com>
Date: Wed, 6 Mar 2024 07:13:08 +0300
Subject: [PATCH] Apply suggestions from code review

Co-authored-by: Loki's Wager <32408858+LokiWager@users.noreply.github.com>
Signed-off-by: vahid khroasani <62482657+v8hid@users.noreply.github.com>
---
 iz_helpers/ui.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/iz_helpers/ui.py b/iz_helpers/ui.py
index 5fcbbbb..7cb0430 100644
--- a/iz_helpers/ui.py
+++ b/iz_helpers/ui.py
@@ -253,6 +253,14 @@ Our best experience and trade-off is the R-ERSGAn4x upscaler.
                     output_video = gr.Video(label="Output").style(width=512, height=512)
                     output_panel = create_output_panel("infinite-zoom", shared.opts.outdir_img2img_samples)
                     
+                if isinstance(output_panel, tuple):
+                    (
+                        out_image,
+                        generation_info,
+                        html_info,
+                        html_log,
+                    ) = output_panel
+                else:
                     out_image = output_panel.gallery
                     generation_info = output_panel.generation_info
                     html_info = output_panel.infotext