gen_image_var??

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 
 from util.instantmesh import generate_mvs, make3d, preprocess, check_input_image
-from util.text_img import generate_txttoimg, check_prompt, generate_imgtoimg
+from util.text_img import generate_txttoimg, check_prompt, generate_imgtoimg, update_image
 
 _CITE_ = r"""
 ```bibtex
@@ -24,6 +24,7 @@ theme = gr.themes.Soft(
 
 
 with gr.Blocks(theme=theme) as GenDemo:
+    gen_image_var = gr.State()
     with gr.Tab("Image to Image Generator"):
         with gr.Row(variant="panel"):
             with gr.Column():
@@ -42,7 +43,7 @@ with gr.Blocks(theme=theme) as GenDemo:
                 button_img = gr.Button("Generate Image", elem_id="generateIm", variant="primary")
                 gen_image = gr.Image(label="Generated Image", image_mode="RGBA", type='pil', show_download_button=True, show_label=False)
         
-        button_img.click(check_prompt, inputs=[prompt]).success(generate_imgtoimg, inputs=[prompt, image, strength], outputs=[gen_image])
+        button_img.click(check_prompt, inputs=[prompt]).success(generate_imgtoimg, inputs=[prompt, image, strength], outputs=[gen_image_var]).success(update_image, inputs=[gen_image_var], outputs=[gen_image])
 
     with gr.Tab("Text to Image Generator"):
         with gr.Row(variant="panel"):
@@ -62,12 +63,18 @@ with gr.Blocks(theme=theme) as GenDemo:
                 button_txt = gr.Button("Generate Image", elem_id="generateIm", variant="primary")
                 gen_image = gr.Image(label="Generated Image", image_mode="RGBA", type='pil', show_download_button=True, show_label=False)
         
-        button_txt.click(check_prompt, inputs=[prompt]).success(generate_txttoimg, inputs=[prompt, controlNet_image, select], outputs=[gen_image])
+        button_txt.click(check_prompt, inputs=[prompt]).success(generate_txttoimg, inputs=[prompt, controlNet_image, select], outputs=[gen_image_var]).success(update_image, inputs=[gen_image_var], outputs=[gen_image])
 
     with gr.Tab("Image to 3D Model Generator"):
         with gr.Row(variant="panel"):
             with gr.Column():
                 with gr.Row():
+                    input_image = gr.Image(
+                        label="Input Image",
+                        image_mode="RGBA",
+                        type="pil",
+                        interactive=True
+                    )
                     processed_image = gr.Image(
                         label="Processed Image", 
                         image_mode="RGBA", 
@@ -125,7 +132,7 @@ with gr.Blocks(theme=theme) as GenDemo:
 
     mv_images = gr.State()
 
-    submit.click(fn=check_input_image, inputs=[gen_image]).success(
+    submit.click(fn=check_input_image, inputs=[gen_image_var]).success(
         fn=preprocess,
         inputs=[gen_image, do_remove_background],
         outputs=[processed_image],

util/text_img.py

@@ -119,4 +119,8 @@ def get_depth(image):
 #     image = image[:,:,None]
 #     image = np.concatenate([image, image, image], axis=2)
 #     canny_image = Image.fromarray(image)
-#     return canny_image
+#     return canny_image
+
+def update_image(image):
+    return image
+