fix some bugs

app.py

@@ -246,7 +246,16 @@ if __name__ == "__main__":
                         label="Generated images", show_label=False, elem_id="gallery"
                     ).style(grid=[2], height="auto")
 
-                    output_txt2img_copy_to_input_btn = gr.Button("Go to img2img")
+                    output_txt2img_copy_to_input_btn = gr.Button("发送图片到图生图(Sent the image to img2img)").style(
+                                margin=False,
+                                rounded=(True, True, True, True),
+                            )
+
+                    go_to_img2img_btn = gr.Button("转至图生图(Go to img2img)").style(
+                                margin=False,
+                                rounded=(True, True, True, True),
+                            )
+
 
                     gr.Examples(examples=examples, fn=request_images, inputs=text, outputs=gallery, examples_per_page=100)
                     text.submit(request_images, inputs=[text, class_draw, style_draw, sample_size, sr_option], outputs=gallery)
@@ -407,7 +416,7 @@ if __name__ == "__main__":
                     img2img_image_mask
                 )
 
-                # 把上面这个注释掉下面就不管用了，很神奇无法理解...
+                # 把上面这个注释掉下面就不管用了，很神奇无法理解... 后来又好了
                 # output_txt2img_copy_to_input_btn.click(
                 #     uifn.copy_img_to_input,
                 #     [gallery],
@@ -421,6 +430,11 @@ if __name__ == "__main__":
                     [gallery],
                     [tabs, img2img_image_editor, img2img_image_mask],
                 )
+                go_to_img2img_btn.click(
+                    uifn.go_to_img2img,
+                    [],
+                    [tabs],
+                )
                 
                 # 下面这几个函数现在都没什么用
                 output_img2img_copy_to_input_btn.click(

custom.css

@@ -1,20 +0,0 @@
-/* Mask background */
-.dark .bg-gray-200{
-    background-color:rgba(55, 55, 55, 0.105)!important;
-}
-.cropper-wrap-box, .cropper-canvas{
-    background-color:rgba(55, 55, 55, 0.105)!important;
-}
-.cropper-bg {
-    background: none!important;
-}
-
-select {
-    background:#000;
-    color:#fff;
-}
-
-select * {
-    background:#373737ff;
-    color:#9c85fb;
-}

ui_functions.py

@@ -79,15 +79,18 @@ def copy_img_params_to_lab(params):
 def copy_img_to_input(img):
     try:
         # print(img)
-        print("=============")
-        print("The img type is:{}".format(type(img[0])))
+        # print("=============")
+        # print("The img type is:{}".format(type(img[0])))
         image_data = re.sub('^data:image/.+;base64,', '', img[0])
         processed_image = Image.open(BytesIO(base64.b64decode(image_data)))
         tab_update = gr.update(selected='img2img_tab')
         img_update = gr.update(value=processed_image)
-        return tab_update, processed_image, processed_image
+        return processed_image, processed_image
     except IndexError:
         return [None, None]
+def go_to_img2img():
+    tab_update = gr.update(selected='img2img_tab')
+    return tab_update
 
 def copy_img_to_edit(img):
     try: