Update app.py

app.py

@@ -739,6 +739,10 @@ def txt_season_visibility(a):
 def gen_edit_fn(b):
     return b["background"], b["background"]
 
+
+def send_f2e(a):
+    return a["background"]
+
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     with gr.Row():
         gr.Markdown(
@@ -840,6 +844,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 gen_edit = gr.Button(value="Send to front image", interactive=True)
             output_edit = gr.Image(label="Edited Image")
 
+    img.upload(send_f2e, img, edit_img)
     #back_tab.select(trans_f2b, [obj_color, obj_color_picker, obj_color_img, txt_color, txt_color_picker, dropdown_name, txt_name, radio_txt_group_name, txt_group_name, chk, obj_num, choose_z_a, choose_etc, obj_count, ai_num], [obj_color2, obj_color_picker2, obj_color_img2, txt_color2, txt_color_picker2, dropdown_name2, txt_name2, radio_txt_group_name2, txt_group_name2, chk3, obj_num2, choose_z_a2, choose_etc2, obj_count2, ai_num2])
     gen_edit.click(gen_edit_fn, edit_img, [output_edit, img])
     obj_color.change(update_obj_color_picker_visibility, obj_color, [obj_color_picker, ai_num, change_ai])
@@ -854,7 +859,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                      [obj_color, obj_color_picker, obj_color_img, txt_color, txt_color_picker, dropdown_name, txt_name,
                       radio_txt_group_name, txt_group_name, chk, obj_num, choose_z_a, choose_etc, obj_count, chk_num])
     change_ai.click(change_ran, obj_color, ai_num)
-    btn.click(generate, [img, obj_color, obj_color_picker, obj_color_img, txt_color, txt_color_picker, dropdown_name, txt_name, radio_txt_group_name, txt_group_name, chk, obj_num, choose_z_a, choose_etc, obj_count, ai_num], [outputs, edit_img])
+    btn.click(generate, [img, obj_color, obj_color_picker, obj_color_img, txt_color, txt_color_picker, dropdown_name, txt_name, radio_txt_group_name, txt_group_name, chk, obj_num, choose_z_a, choose_etc, obj_count, ai_num], outputs)
     btn_back.click(generate_back, [img, obj_color, obj_color_picker, obj_color_img, txt_color, txt_color_picker, dropdown_name, txt_name, radio_txt_group_name, txt_group_name, chk, obj_num, choose_z_a, choose_etc, obj_count, radio_class, txt_class, obj_color_outline, obj_color_outline_picker, radio_season, txt_season, cr, btn_logo, qr_url, ai_num], outputs2)
     back_tab.select(generate_back, [img, obj_color, obj_color_picker, obj_color_img, txt_color, txt_color_picker, dropdown_name, txt_name, radio_txt_group_name, txt_group_name, chk, obj_num, choose_z_a, choose_etc, obj_count, radio_class, txt_class, obj_color_outline, obj_color_outline_picker, radio_season, txt_season, cr, btn_logo, qr_url, ai_num], preview)
     radio_class.change(txt_class_visibility, radio_class, txt_class)