Update app.py

app.py

@@ -606,229 +606,223 @@ def update_process_button(img, prompt):
         variant="primary" if bool(img and prompt) else "secondary"
     )
 
-demo = gr.Blocks(
-    theme=gr.themes.Soft(),
-    css=css,
-    analytics_enabled=False,  # 분석 비활성화
-    cache_examples=False  # 예제 캐싱 비활성화
-)
+if __name__ == "__main__":
+    # 모델 초기화
+    initialize_models()
     
-
-    gr.HTML("""
-        <div class="main-title">
-            <h1>🎨GiniGen Canvas</h1>
-            <p>AI Integrated Image Creator: Extract objects, generate backgrounds, and adjust ratios and positions to create complete images with AI.</p>
-        </div>
-    """)
-
-    with gr.Row():
-        # 입력 컬럼
-        with gr.Column(scale=1):
-            input_image = gr.Image(
-                type="pil",
-                label="Upload Image",
-                interactive=True
-            )
-            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
+    # Gradio UI 정의
+    with gr.Blocks(
+        theme=gr.themes.Soft(),
+        css=css,
+        analytics_enabled=False,
+        cache_examples=False
+    ) as demo:
+        # HTML 헤더
+        gr.HTML("""
+            <div class="main-title">
+                <h1>🎨GiniGen Canvas</h1>
+                <p>AI Integrated Image Creator: Extract objects, generate backgrounds, and adjust ratios and positions to create complete images with AI.</p>
+            </div>
+        """)
+
+        with gr.Row():
+            # 입력 컬럼
+            with gr.Column(scale=1):
+                input_image = gr.Image(
+                    type="pil",
+                    label="Upload Image",
+                    interactive=True
                 )
-                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
+                text_prompt = gr.Textbox(
+                    label="Object to Extract",
+                    placeholder="Enter what you want to extract...",
+                    interactive=True
                 )
-
-            # 오브젝트 컨트롤
-            with gr.Row(visible=False) as object_controls:
-                # 위치 컨트롤
-                with gr.Column(scale=1):
-                    position = gr.State(value="bottom-center")
-                    with gr.Row():
-                        btn_top_left = gr.Button("↖")
-                        btn_top_center = gr.Button("↑")
-                        btn_top_right = gr.Button("↗")
-                    with gr.Row():
-                        btn_middle_left = gr.Button("←")
-                        btn_middle_center = gr.Button("•")
-                        btn_middle_right = gr.Button("→")
-                    with gr.Row():
-                        btn_bottom_left = gr.Button("↙")
-                        btn_bottom_center = gr.Button("↓")
-                        btn_bottom_right = gr.Button("↘")
                 
-                # 크기 컨트롤
-                with gr.Column(scale=1):
-                    scale_slider = gr.Slider(
-                        minimum=10,
-                        maximum=200,
-                        value=50,
-                        step=5,
-                        label="Object Size (%)"
+                # 배경 및 비율 설정
+                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() as text_group:
-                text_input = gr.Textbox(
-                    label="Text to Add",
-                    placeholder="Enter text..."
-                )
-                with gr.Row():
+                # 오브젝트 컨트롤
+                with gr.Row(visible=False) as object_controls:
+                    # 위치 컨트롤
                     with gr.Column(scale=1):
-                        font_size = gr.Slider(
-                            minimum=10,
-                            maximum=800,
-                            value=400,
-                            step=10,
-                            label="Font Size"
-                        )
-                        thickness = gr.Slider(
-                            minimum=0,
-                            maximum=20,
-                            value=0,
-                            step=1,
-                            label="Text Thickness"
-                        )
-                        color_dropdown = gr.Dropdown(
-                            choices=["White", "Black", "Red", "Green", "Blue", "Yellow", "Purple"],
-                            value="White",
-                            label="Text Color"
-                        )
+                        position = gr.State(value="bottom-center")
+                        with gr.Row():
+                            btn_top_left = gr.Button("↖")
+                            btn_top_center = gr.Button("↑")
+                            btn_top_right = gr.Button("↗")
+                        with gr.Row():
+                            btn_middle_left = gr.Button("←")
+                            btn_middle_center = gr.Button("•")
+                            btn_middle_right = gr.Button("→")
+                        with gr.Row():
+                            btn_bottom_left = gr.Button("↙")
+                            btn_bottom_center = gr.Button("↓")
+                            btn_bottom_right = gr.Button("↘")
+                    
+                    # 크기 컨트롤
                     with gr.Column(scale=1):
-                        opacity_slider = gr.Slider(
-                            minimum=0,
-                            maximum=255,
-                            value=255,
-                            step=1,
-                            label="Opacity"
-                        )
-                        text_x_position = gr.Slider(
-                            minimum=0,
-                            maximum=100,
-                            value=50,
-                            step=1,
-                            label="Text X Position (%)"
-                        )
-                        text_y_position = gr.Slider(
-                            minimum=0,
-                            maximum=100,
+                        scale_slider = gr.Slider(
+                            minimum=10,
+                            maximum=200,
                             value=50,
-                            step=1,
-                            label="Text Y Position (%)"
+                            step=5,
+                            label="Object Size (%)"
                         )
 
-            # 처리 버튼
-            process_btn = gr.Button(
-                "Process",
-                variant="primary",
-                interactive=False
-            )
-
-        # 출력 컬럼
-        with gr.Column(scale=1):
-            combined_image = gr.Image(
-                label="Combined Result",
-                show_download_button=True,
-                type="pil",
-                height=512
-            )
-            extracted_image = gr.Image(
-                label="Extracted Object",
-                show_download_button=True,
-                type="pil",
-                height=256
-            )
-
-
-
-    def get_text_params():
-        return {
-            'text': text_input.value,
-            'font_size': font_size.value,
-            'thickness': thickness.value,
-            'color': color_dropdown.value,
-            'opacity': opacity_slider.value,
-            'x_position': text_x_position.value,
-            'y_position': text_y_position.value
-        }
-
-    # 이벤트 바인딩
-    input_image.change(
-        fn=update_process_button,
-        inputs=[input_image, text_prompt],
-        outputs=process_btn,
-        queue=False
-    )
-    
-    text_prompt.change(
-        fn=update_process_button,
-        inputs=[input_image, text_prompt],
-        outputs=process_btn,
-        queue=False
-    )
+                # 텍스트 입력 섹션
+                with gr.Group() as text_group:
+                    text_input = gr.Textbox(
+                        label="Text to Add",
+                        placeholder="Enter text..."
+                    )
+                    with gr.Row():
+                        with gr.Column(scale=1):
+                            font_size = gr.Slider(
+                                minimum=10,
+                                maximum=800,
+                                value=400,
+                                step=10,
+                                label="Font Size"
+                            )
+                            thickness = gr.Slider(
+                                minimum=0,
+                                maximum=20,
+                                value=0,
+                                step=1,
+                                label="Text Thickness"
+                            )
+                            color_dropdown = gr.Dropdown(
+                                choices=["White", "Black", "Red", "Green", "Blue", "Yellow", "Purple"],
+                                value="White",
+                                label="Text Color"
+                            )
+                        with gr.Column(scale=1):
+                            opacity_slider = gr.Slider(
+                                minimum=0,
+                                maximum=255,
+                                value=255,
+                                step=1,
+                                label="Opacity"
+                            )
+                            text_x_position = gr.Slider(
+                                minimum=0,
+                                maximum=100,
+                                value=50,
+                                step=1,
+                                label="Text X Position (%)"
+                            )
+                            text_y_position = gr.Slider(
+                                minimum=0,
+                                maximum=100,
+                                value=50,
+                                step=1,
+                                label="Text Y Position (%)"
+                            )
+
+                # 처리 버튼
+                process_btn = gr.Button(
+                    "Process",
+                    variant="primary",
+                    interactive=False
+                )
 
-    bg_prompt.change(
-        fn=update_controls,
-        inputs=[bg_prompt],
-        outputs=[aspect_ratio, object_controls],
-        queue=False
-    )
+            # 출력 컬럼
+            with gr.Column(scale=1):
+                combined_image = gr.Image(
+                    label="Combined Result",
+                    show_download_button=True,
+                    type="pil",
+                    height=512
+                )
+                extracted_image = gr.Image(
+                    label="Extracted Object",
+                    show_download_button=True,
+                    type="pil",
+                    height=256
+                )
 
-    # 위치 버튼 이벤트 (한 번만 정의)
-    for btn, pos in [
-        (btn_top_left, "top-left"), (btn_top_center, "top-center"), (btn_top_right, "top-right"),
-        (btn_middle_left, "middle-left"), (btn_middle_center, "middle-center"), (btn_middle_right, "middle-right"),
-        (btn_bottom_left, "bottom-left"), (btn_bottom_center, "bottom-center"), (btn_bottom_right, "bottom-right")
-    ]:
-        btn.click(fn=lambda p=pos: p, outputs=position)
-
-    # 메인 프로세스 이벤트
-    process_btn.click(
-        fn=process_prompt,
-        inputs=[
-            input_image,
-            text_prompt,
-            bg_prompt,
-            aspect_ratio,
-            position,
-            scale_slider,
-            gr.State(get_text_params)  # State 객체로 감싸서 전달
-        ],
-        outputs=[combined_image, extracted_image],
-        queue=True
-    )
+        # 텍스트 파라미터 가져오기 함수
+        def get_text_params():
+            return {
+                'text': text_input.value,
+                'font_size': font_size.value,
+                'thickness': thickness.value,
+                'color': color_dropdown.value,
+                'opacity': opacity_slider.value,
+                'x_position': text_x_position.value,
+                'y_position': text_y_position.value
+            }
+
+        # 이벤트 바인딩
+        input_image.change(
+            fn=update_process_button,
+            inputs=[input_image, text_prompt],
+            outputs=process_btn,
+            queue=False
+        )
+        
+        text_prompt.change(
+            fn=update_process_button,
+            inputs=[input_image, text_prompt],
+            outputs=process_btn,
+            queue=False
+        )
 
-# HF 토큰 및 파이프라인 초기화
-if HF_TOKEN:
-    login(token=HF_TOKEN, add_to_git_credential=False)
+        bg_prompt.change(
+            fn=update_controls,
+            inputs=[bg_prompt],
+            outputs=[aspect_ratio, object_controls],
+            queue=False
+        )
 
-pipe = FluxPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-dev",
-    torch_dtype=torch.float16,
-    token=HF_TOKEN
-)
+        # 위치 버튼 이벤트
+        for btn, pos in [
+            (btn_top_left, "top-left"), (btn_top_center, "top-center"), (btn_top_right, "top-right"),
+            (btn_middle_left, "middle-left"), (btn_middle_center, "middle-center"), (btn_middle_right, "middle-right"),
+            (btn_bottom_left, "bottom-left"), (btn_bottom_center, "bottom-center"), (btn_bottom_right, "bottom-right")
+        ]:
+            btn.click(fn=lambda p=pos: p, outputs=position)
+
+        # 메인 프로세스 이벤트
+        process_btn.click(
+            fn=process_prompt,
+            inputs=[
+                input_image,
+                text_prompt,
+                bg_prompt,
+                aspect_ratio,
+                position,
+                scale_slider,
+                gr.State(get_text_params)
+            ],
+            outputs=[combined_image, extracted_image],
+            queue=True
+        )
 
-demo.queue(max_size=3)  # 큐 크기 감소
-demo.launch(
-    server_name="0.0.0.0",
-    server_port=7860,
-    share=False,
-    max_threads=2,
-    enable_queue=True,
-    cache_examples=False,
-    show_error=True,
-    show_tips=False
-)
+    # Gradio 앱 실행
+    demo.queue(max_size=3)
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        max_threads=2,
+        enable_queue=True,
+        cache_examples=False,
+        show_error=True,
+        show_tips=False
+    )