Update app.py

app.py

@@ -805,7 +805,7 @@ def update_controls(bg_prompt):
     ]
 
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
-    position = gr.State(value="bottom-center")  # 여기로 이동
+    position = gr.State(value="bottom-center")
 
     gr.HTML("""
         <div class="main-title">
@@ -818,63 +818,78 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         # 왼쪽 패널 (입력)
         with gr.Column(scale=1):
             with gr.Group(elem_classes="input-panel"):
-                input_image = gr.Image(
-                    type="pil",
-                    label="Upload Image",
-                    interactive=True,
-                    height=400
-                )
-                text_prompt = gr.Textbox(
-                    label="Object to Extract",
-                    placeholder="Enter what you want to extract...",
-                    interactive=True
-                )
-                with gr.Row():
-                    bg_prompt = gr.Textbox(
-                        label="Background Prompt (optional)",
-                        placeholder="Describe the background...",
-                        interactive=True,
-                        scale=3
-                    )
-                    aspect_ratio = gr.Dropdown(
-                        choices=["1:1", "16:9", "9:16", "4:3"],
-                        value="1:1",
-                        label="Aspect Ratio",
-                        interactive=True,
-                        visible=True,
-                        scale=1
-                    )
-
-                with gr.Group(elem_classes="controls-panel", visible=False) as object_controls:
-                    with gr.Column(scale=1):
-                        position = gr.State(value="bottom-center")  # 초기값 설정
-                        with gr.Row():
-                            btn_top_left = gr.Button("↖", elem_classes="position-btn")
-                            btn_top_center = gr.Button("↑", elem_classes="position-btn")
-                            btn_top_right = gr.Button("↗", elem_classes="position-btn")
-                        with gr.Row():
-                            btn_middle_left = gr.Button("←", elem_classes="position-btn")
-                            btn_middle_center = gr.Button("•", elem_classes="position-btn")
-                            btn_middle_right = gr.Button("→", elem_classes="position-btn")
-                        with gr.Row():
-                            btn_bottom_left = gr.Button("↙", elem_classes="position-btn")
-                            btn_bottom_center = gr.Button("↓", elem_classes="position-btn", value="selected")
-                            btn_bottom_right = gr.Button("↘", elem_classes="position-btn")
-                    with gr.Column(scale=1):
-                        scale_slider = gr.Slider(
-                            minimum=10,
-                            maximum=200,
-                            value=50,
-                            step=5,
-                            label="Object Size (%)"
+                with gr.Tabs():
+                    # 첫 번째 탭: 이미지 업로드 및 인페인팅
+                    with gr.Tab("Image Upload & Inpainting"):
+                        input_image = gr.Image(
+                            type="pil",
+                            label="Upload Image",
+                            interactive=True,
+                            height=400
                         )
+                        with gr.Group():
+                            inpaint_prompt = gr.Textbox(
+                                label="Inpainting Prompt",
+                                placeholder="Describe what you want to add in the masked area..."
+                            )
+                            mask_input = image_annotator(
+                                label="Draw mask for inpainting",
+                                height=400
+                            )
+                            inpaint_btn = gr.Button("Apply Inpainting", variant="primary")
+
+                    # 두 번째 탭: 배경 제거 및 생성
+                    with gr.Tab("Background Removal"):
+                        text_prompt = gr.Textbox(
+                            label="Object to Extract",
+                            placeholder="Enter what you want to extract...",
+                            interactive=True
+                        )
+                        with gr.Row():
+                            bg_prompt = gr.Textbox(
+                                label="Background Prompt (optional)",
+                                placeholder="Describe the background...",
+                                interactive=True,
+                                scale=3
+                            )
+                            aspect_ratio = gr.Dropdown(
+                                choices=["1:1", "16:9", "9:16", "4:3"],
+                                value="1:1",
+                                label="Aspect Ratio",
+                                interactive=True,
+                                visible=True,
+                                scale=1
+                            )
+
+                        with gr.Group(elem_classes="controls-panel", visible=False) as object_controls:
+                            with gr.Column(scale=1):
+                                with gr.Row():
+                                    btn_top_left = gr.Button("↖", elem_classes="position-btn")
+                                    btn_top_center = gr.Button("↑", elem_classes="position-btn")
+                                    btn_top_right = gr.Button("↗", elem_classes="position-btn")
+                                with gr.Row():
+                                    btn_middle_left = gr.Button("←", elem_classes="position-btn")
+                                    btn_middle_center = gr.Button("•", elem_classes="position-btn")
+                                    btn_middle_right = gr.Button("→", elem_classes="position-btn")
+                                with gr.Row():
+                                    btn_bottom_left = gr.Button("↙", elem_classes="position-btn")
+                                    btn_bottom_center = gr.Button("↓", elem_classes="position-btn", value="selected")
+                                    btn_bottom_right = gr.Button("↘", elem_classes="position-btn")
+                            with gr.Column(scale=1):
+                                scale_slider = gr.Slider(
+                                    minimum=10,
+                                    maximum=200,
+                                    value=50,
+                                    step=5,
+                                    label="Object Size (%)"
+                                )
 
-                process_btn = gr.Button(
-                    "Process",
-                    variant="primary",
-                    interactive=False,
-                    size="lg"
-                )
+                        process_btn = gr.Button(
+                            "Process",
+                            variant="primary",
+                            interactive=False,
+                            size="lg"
+                        )
 
         # 오른쪽 패널 (출력)
         with gr.Column(scale=1):
@@ -887,7 +902,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                         height=400
                     )
                     
-                    # 텍스트 삽입 옵션을 Accordion으로 변경
                     with gr.Accordion("Text Insertion Options", open=False):
                         with gr.Group():
                             with gr.Row():
@@ -904,14 +918,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                             with gr.Row():
                                 with gr.Column(scale=1):
                                     font_choice = gr.Dropdown(
-                                        choices=["Default", "Korean Regular"],  # "Korean Son" 제거
-
+                                        choices=["Default", "Korean Regular"],
                                         value="Default",
                                         label="Font Selection",
                                         interactive=True
                                     )
-
-                                    
                                     font_size = gr.Slider(
                                         minimum=10,
                                         maximum=200,
@@ -962,7 +973,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                         height=200
                     )
 
-    # CSS 클래스를 위한 스타일 추가
+    # CSS 스타일
     gr.HTML("""
         <style>
         .position-btn.selected {
@@ -972,7 +983,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         </style>
     """)
 
-# 버튼 클릭 이벤트 바인딩
+    # 이벤트 바인딩
     position_mapping = {
         btn_top_left: "top-left",
         btn_top_center: "top-center",
@@ -987,12 +998,16 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
 
     for btn, pos in position_mapping.items():
         btn.click(
-            fn=lambda pos=pos: update_position(pos),  # 클로저 문제 해결을 위해 수정
+            fn=lambda pos=pos: update_position(pos),
             outputs=position
         )
- 
 
-    # 이벤트 바인딩
+    inpaint_btn.click(
+        fn=process_inpainting,
+        inputs=[input_image, mask_input, inpaint_prompt],
+        outputs=input_image
+    )
+
     bg_prompt.change(
         fn=update_controls,
         inputs=bg_prompt,
@@ -1028,12 +1043,11 @@ 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,      # 두 번째 인자로 텍스트
+            combined_image,
+            text_input,
             font_size,
             color_dropdown,
             opacity_slider,
@@ -1044,10 +1058,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             font_choice
         ],
         outputs=combined_image,
-        api_name="add_text"  # API 이름 추가
+        api_name="add_text"
     )
-    
-
 
 demo.queue(max_size=5)
 demo.launch(