Update app.py

app.py

@@ -601,7 +601,6 @@ def add_text_to_image(
     # Convert PIL image back to numpy array for Gradio
     return np.array(output_image)
     
-# UI 구성
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
         <div class="main-title">
@@ -668,35 +667,88 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                 interactive=False
             )
 
-            # 각 버튼에 대한 클릭 이벤트 처리
-            def update_position(new_position):
-                return new_position
-
-            btn_top_left.click(fn=lambda: update_position("top-left"), outputs=position)
-            btn_top_center.click(fn=lambda: update_position("top-center"), outputs=position)
-            btn_top_right.click(fn=lambda: update_position("top-right"), outputs=position)
-            btn_middle_left.click(fn=lambda: update_position("middle-left"), outputs=position)
-            btn_middle_center.click(fn=lambda: update_position("middle-center"), outputs=position)
-            btn_middle_right.click(fn=lambda: update_position("middle-right"), outputs=position)
-            btn_bottom_left.click(fn=lambda: update_position("bottom-left"), outputs=position)
-            btn_bottom_center.click(fn=lambda: update_position("bottom-center"), outputs=position)
-            btn_bottom_right.click(fn=lambda: update_position("bottom-right"), outputs=position)
-
         with gr.Column(scale=1):
-            with gr.Row():
+            with gr.Tab("Result"):
                 combined_image = gr.Image(
                     label="Combined Result",
                     show_download_button=True,
                     type="pil",
                     height=512
                 )
-            with gr.Row():
-                extracted_image = gr.Image(
-                    label="Extracted Object",
-                    show_download_button=True,
-                    type="pil",
-                    height=256
-                )
+                
+                # 텍스트 삽입 컨트롤을 더 명확하게 구분
+                with gr.Box():
+                    gr.Markdown("### Add Text to Image")
+                    text_input = gr.Textbox(
+                        label="Text Content",
+                        placeholder="Enter text to add to image..."
+                    )
+                    with gr.Row():
+                        with gr.Column(scale=1):
+                            font_size = gr.Slider(
+                                minimum=10,
+                                maximum=200,
+                                value=40,
+                                step=5,
+                                label="Font Size"
+                            )
+                            color_dropdown = gr.Dropdown(
+                                choices=["White", "Black", "Red", "Green", "Blue", "Yellow", "Purple"],
+                                value="White",
+                                label="Text Color"
+                            )
+                            thickness = gr.Slider(
+                                minimum=0,
+                                maximum=10,
+                                value=1,
+                                step=1,
+                                label="Text Thickness"
+                            )
+                        with gr.Column(scale=1):
+                            opacity_slider = gr.Slider(
+                                minimum=0,
+                                maximum=255,
+                                value=255,
+                                step=1,
+                                label="Opacity"
+                            )
+                            x_position = gr.Slider(
+                                minimum=0,
+                                maximum=100,
+                                value=50,
+                                step=1,
+                                label="X Position (%)"
+                            )
+                            y_position = gr.Slider(
+                                minimum=0,
+                                maximum=100,
+                                value=50,
+                                step=1,
+                                label="Y Position (%)"
+                            )
+                    add_text_btn = gr.Button("Apply Text", variant="primary")
+
+                with gr.Row():
+                    extracted_image = gr.Image(
+                        label="Extracted Object",
+                        show_download_button=True,
+                        type="pil",
+                        height=256
+                    )
+
+    # 각 버튼에 대한 클릭 이벤트 처리
+    def update_position(new_position):
+        return new_position
+
+    btn_top_left.click(fn=lambda: update_position("top-left"), outputs=position)
+    btn_top_center.click(fn=lambda: update_position("top-center"), outputs=position)
+    btn_top_right.click(fn=lambda: update_position("top-right"), outputs=position)
+    btn_middle_left.click(fn=lambda: update_position("middle-left"), outputs=position)
+    btn_middle_center.click(fn=lambda: update_position("middle-center"), outputs=position)
+    btn_middle_right.click(fn=lambda: update_position("middle-right"), outputs=position)
+    btn_bottom_left.click(fn=lambda: update_position("bottom-left"), outputs=position)
+    btn_bottom_center.click(fn=lambda: update_position("bottom-center"), outputs=position)
+    btn_bottom_right.click(fn=lambda: update_position("bottom-right"), outputs=position)
 
     # Event bindings
     input_image.change(
@@ -742,6 +794,21 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         queue=True
     )
 
+    # 텍스트 추가 버튼 이벤트 연결
+    add_text_btn.click(
+        fn=add_text_to_image,
+        inputs=[
+            combined_image,
+            text_input,
+            font_size,
+            color_dropdown,
+            opacity_slider,
+            x_position,
+            y_position,
+            thickness
+        ],
+        outputs=combined_image
+    )
 
 demo.queue(max_size=5)  # 큐 크기 제한
 demo.launch(
@@ -749,4 +816,4 @@ demo.launch(
     server_port=7860,
     share=False,
     max_threads=2  # 스레드 수 제한
-)
+)