upd

gradio_seg2image.py

@@ -26,10 +26,13 @@ def encode(img_array):
     #print(encoded_string)
     return encoded_string
 
-def create_imgcomp(input_image, filename):
-    encoded_string = encode(input_image)
+def create_imgcomp(input_image, result_image): #(input_image, filename):
+    encoded_string_in = encode(input_image)
+    encoded_string_out = encode(result_image)
+    
     #dummyfun(result_gallery)
-    htmltag = '<img src= "data:image/jpeg;base64,' + encoded_string + '" alt="Original Image"/></div> <img src= "https://ysharma-controlnet-image-comparison.hf.space/file=' + filename + '" alt="Control Net Image"/>'
+    htmltag = '<img src= "data:image/jpeg;base64,' + encoded_string_in + '" alt="Original Image"/></div> <img src= "data:image/jpeg;base64,' + encoded_string_out + '" alt="Control Net Image"/>'
+    #htmltag = '<img src= "data:image/jpeg;base64,' + encoded_string + '" alt="Original Image"/></div> <img src= "https://ysharma-controlnet-image-comparison.hf.space/file=' + filename + '" alt="Control Net Image"/>'
     #https://ysharma-controlnetsegmentation.hf.space/file=/tmp/tmpqcz9yeta.png
     #https://ysharma-controlnet-image-comparison.hf.space/file=/tmp/tmpg4qx22xy.png
     print(f"htmltag is ^^ - {htmltag}")
@@ -155,23 +158,25 @@ def create_demo(process, max_images=12):
                         'longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality'
                     )
             with gr.Column():
-                result_gallery = gr.Gallery(label='Output', visible= False, 
-                                            show_label=False,
-                                            elem_id='gallery').style(
-                                                grid=2, height='auto')
-                b1 = gr.Button('Get filenames')
-                filename = gr.Textbox(label="image file names", visible=False)
-                b2 = gr.Button('Show Image-Comparison')
-                imagecomp = gr.HTML() 
+                #result_gallery = gr.Gallery(label='Output', visible= False, 
+                #                            show_label=False,
+                #                            elem_id='gallery').style(
+                #                                grid=2, height='auto')
+                result_image = gr.Image(visible=False).style(height='auto', type='numpy')
+                #b1 = gr.Button('Get filenames')
+                #filename = gr.Textbox(label="image file names", visible=False)
+                #b2 = gr.Button('Show Image-Comparison')
+                with gr.Box():
+                    imagecomp = gr.HTML() 
         ips = [
-            input_image, prompt, a_prompt, n_prompt, num_samples,
+            input_image, prompt, a_prompt, n_prompt, num_sampls,
             image_resolution, detect_resolution, ddim_steps, scale, seed, eta
         ]
         run_button.click(fn=process,
                          inputs=ips,
-                         outputs=[result_gallery, imagecomp],
+                         outputs=[result_image], #[result_gallery, imagecomp],
                          api_name='seg')
-        #b1.click(dummyfun, [result_gallery], [filename])
+        result_image.change(create_imgcomp, [input_image, result_image], [imagecomp])
         #b2.click(create_imgcomp, [input_image, filename], [imagecomp])
         
     return demo