Update app.py

app.py

@@ -269,7 +269,7 @@ def change_text_in_image_two_times(original_image, instruction):
 
 
 #######################################
-# 5. Main Process
+# 5. Main Process (Generation from Prompt)
 #######################################
 
 def run_process(
@@ -313,7 +313,27 @@ def run_process(
     return [final_imgs[0], final_imgs[1]]
 
 #######################################
-# 6. Gradio UI
+# 5-2. Process for Editing Uploaded Image
+#######################################
+
+def run_edit_process(input_image, final_text1, final_text2, final_text3):
+    """
+    Process for editing an uploaded image.
+    1) For each provided text, if it's purely English, use as-is;
+       otherwise, replace with random letters.
+    2) Build instructions and call the text modification function twice.
+    """
+    r1 = maybe_use_random_or_original(final_text1)
+    r2 = maybe_use_random_or_original(final_text2)
+    r3 = maybe_use_random_or_original(final_text3)
+    print(f"[DEBUG] Editing image with placeholders: r1='{r1}', r2='{r2}', r3='{r3}'")
+    instruction = build_multi_change_instruction(r1, final_text1, r2, final_text2, r3, final_text3)
+    print(f"[DEBUG] Editing instruction: {instruction}")
+    final_imgs = change_text_in_image_two_times(input_image, instruction)
+    return [final_imgs[0], final_imgs[1]]
+
+#######################################
+# 6. Gradio UI with Two Tabs
 #######################################
 
 with gr.Blocks(title="Eevery Text Imaginator: FLUX") as demo:
@@ -367,90 +387,113 @@ with gr.Blocks(title="Eevery Text Imaginator: FLUX") as demo:
         <hr style="margin: 15px 0;">
         """
     )
-
-    # 6 example prompts (slightly more colorful list)
-    examples = [
-        [
-            "Futuristic neon sign with <text1>, plus near the bottom",
-            "OPEN", "", ""
-        ],
-        [
-            "On a grand stage, <text1> in big letters and on the left side",
-            "환영합니다.", "", ""
-        ],
-        [
-            "A classical poster reading <text1> in bold,  as a subtitle",
-            "错觉", "", ""
-        ],
-        [
-            "In a cartoon style, a speech bubble with <text1> and another text ",
-            "안녕", "", ""
-        ],
-        [
-            "Large billboard featuring <text1>",
-            "아름다운 당신", "", ""
-        ],
-        [
-            "썬글라스 착용한 흰색 고양이의 배너 <text1>",
-            "안녕", "", ""
-        ],
-    ]
-
-    with gr.Row():
-        with gr.Column():
-            with gr.Group():
-                prompt_input = gr.Textbox(
-                    lines=3,
-                    label="Prompt (Korean or English)",
-                    placeholder="On a grand stage, <text1> in big letters..."
-                )
-                final_text1 = gr.Textbox(
-                    label="New Text #1 (Required)",
-                    placeholder="Example: HELLO or 안녕하세요"
-                )
-                final_text2 = gr.Textbox(
-                    label="New Text #2 (Optional)",
-                    placeholder="Example: WORLD or 반갑습니다"
-                )
-                final_text3 = gr.Textbox(
-                    label="New Text #3 (Optional)",
-                    placeholder="(Leave blank if not used)"
-                )
-
-            with gr.Accordion("Advanced Settings (optional)", open=False):
-                height = gr.Slider(label="Height", minimum=256, maximum=1152, step=64, value=512)
-                width = gr.Slider(label="Width", minimum=256, maximum=1152, step=64, value=512)
-                steps = gr.Slider(label="Inference Steps", minimum=6, maximum=25, step=1, value=8)
-                scale = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=10.0, step=0.5, value=3.5)
-                seed = gr.Number(label="Seed", value=1234, precision=0)
-
-            run_btn = gr.Button("Generate 2 Final Images", variant="primary")
-
-            # Show the examples
-            gr.Examples(
-                examples=examples,
-                inputs=[prompt_input, final_text1, final_text2, final_text3],
-                label="Example Prompts"
+    
+    with gr.Tabs():
+        with gr.TabItem("Generate from Prompt"):
+            with gr.Row():
+                with gr.Column():
+                    with gr.Group():
+                        prompt_input = gr.Textbox(
+                            lines=3,
+                            label="Prompt (Korean or English)",
+                            placeholder="On a grand stage, <text1> in big letters..."
+                        )
+                        final_text1 = gr.Textbox(
+                            label="New Text #1 (Required)",
+                            placeholder="Example: HELLO or 안녕하세요"
+                        )
+                        final_text2 = gr.Textbox(
+                            label="New Text #2 (Optional)",
+                            placeholder="Example: WORLD or 반갑습니다"
+                        )
+                        final_text3 = gr.Textbox(
+                            label="New Text #3 (Optional)",
+                            placeholder="(Leave blank if not used)"
+                        )
+                    with gr.Accordion("Advanced Settings (optional)", open=False):
+                        height = gr.Slider(label="Height", minimum=256, maximum=1152, step=64, value=512)
+                        width = gr.Slider(label="Width", minimum=256, maximum=1152, step=64, value=512)
+                        steps = gr.Slider(label="Inference Steps", minimum=6, maximum=25, step=1, value=8)
+                        scale = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=10.0, step=0.5, value=3.5)
+                        seed = gr.Number(label="Seed", value=1234, precision=0)
+                    run_btn = gr.Button("Generate 2 Final Images", variant="primary")
+                    gr.Examples(
+                        examples=[
+                            [
+                                "Futuristic neon sign with <text1>, plus near the bottom",
+                                "OPEN", "", ""
+                            ],
+                            [
+                                "On a grand stage, <text1> in big letters and on the left side",
+                                "환영합니다.", "", ""
+                            ],
+                            [
+                                "A classical poster reading <text1> in bold,  as a subtitle",
+                                "错觉", "", ""
+                            ],
+                            [
+                                "In a cartoon style, a speech bubble with <text1> and another text ",
+                                "안녕", "", ""
+                            ],
+                            [
+                                "Large billboard featuring <text1>",
+                                "아름다운 당신", "", ""
+                            ],
+                            [
+                                "썬글라스 착용한 흰색 고양이의 배너 <text1>",
+                                "안녕", "", ""
+                            ],
+                        ],
+                        inputs=[prompt_input, final_text1, final_text2, final_text3],
+                        label="Example Prompts"
+                    )
+                with gr.Column():
+                    final_image_output1 = gr.Image(label="Final Image #1", type="pil")
+                    final_image_output2 = gr.Image(label="Final Image #2", type="pil")
+            run_btn.click(
+                fn=run_process,
+                inputs=[
+                    prompt_input,
+                    final_text1,
+                    final_text2,
+                    final_text3,
+                    height,
+                    width,
+                    steps,
+                    scale,
+                    seed
+                ],
+                outputs=[final_image_output1, final_image_output2]
+            )
+        
+        with gr.TabItem("Edit Uploaded Image"):
+            with gr.Row():
+                with gr.Column():
+                    uploaded_image = gr.Image(
+                        label="Upload Image for Editing", 
+                        type="pil", 
+                        source="upload"
+                    )
+                    final_text1_edit = gr.Textbox(
+                        label="New Text #1 (Required)",
+                        placeholder="Example: HELLO or 안녕하세요"
+                    )
+                    final_text2_edit = gr.Textbox(
+                        label="New Text #2 (Optional)",
+                        placeholder="Example: WORLD or 반갑습니다"
+                    )
+                    final_text3_edit = gr.Textbox(
+                        label="New Text #3 (Optional)",
+                        placeholder="(Leave blank if not used)"
+                    )
+                    run_edit_btn = gr.Button("Edit Image", variant="primary")
+                with gr.Column():
+                    edited_image_output1 = gr.Image(label="Edited Image #1", type="pil")
+                    edited_image_output2 = gr.Image(label="Edited Image #2", type="pil")
+            run_edit_btn.click(
+                fn=run_edit_process,
+                inputs=[uploaded_image, final_text1_edit, final_text2_edit, final_text3_edit],
+                outputs=[edited_image_output1, edited_image_output2]
             )
-
-        with gr.Column():
-            final_image_output1 = gr.Image(label="Final Image #1", type="pil")
-            final_image_output2 = gr.Image(label="Final Image #2", type="pil")
-
-    run_btn.click(
-        fn=run_process,
-        inputs=[
-            prompt_input,
-            final_text1,
-            final_text2,
-            final_text3,
-            height,
-            width,
-            steps,
-            scale,
-            seed
-        ],
-        outputs=[final_image_output1, final_image_output2]
-    )
 
 demo.launch(max_threads=20)