Update app.py

app.py

@@ -460,7 +460,7 @@ def update_background_style_options(background_type):
     else:
         return gr.update(choices=["선택하세요"], value="선택하세요")
 
-# Gradio 인터페이스
+# 인터페이스 부분 수정
 with gr.Blocks() as demo:
     gr.HTML(
         """
@@ -487,11 +487,11 @@ with gr.Blocks() as demo:
             )
             
             with gr.Row():
-                image_change_btn = gr.Button("이미지 변경")
-                text_remove_btn = gr.Button("글자지우기")
-                clothes_change_btn = gr.Button("옷바꾸기")
-                background_change_btn = gr.Button("배경바꾸기", variant="primary")
-                composite_product_btn = gr.Button("상품합성")
+                image_change_btn = gr.Button("이미지 변경", variant="secondary")
+                text_remove_btn = gr.Button("글자지우기", variant="secondary")
+                clothes_change_btn = gr.Button("옷바꾸기", variant="secondary")
+                background_change_btn = gr.Button("배경바꾸기", variant="secondary")
+                composite_product_btn = gr.Button("상품합성", variant="secondary")
             
             # 배경 변경 관련 섹션 (기본적으로 숨김 상태)
             with gr.Row(visible=False) as background_header:
@@ -571,12 +571,11 @@ with gr.Blocks() as demo:
                 visible=False
             )
             
-            # 버튼들을 그룹화
-            with gr.Row(visible=False) as buttons_row:
-                # 프롬프트 생성 버튼 추가
-                prompt_generate_btn = gr.Button("프롬프트 생성", variant="secondary")
-                # 이미지 생성 버튼
-                submit_btn = gr.Button("이미지 생성", variant="primary")
+            # 프롬프트 생성 버튼 (배경바꾸기 선택시에만 나타남)
+            prompt_generate_btn = gr.Button("프롬프트 생성", variant="secondary", visible=False)
+            
+            # 이미지 생성 버튼 (항상 표시)
+            submit_btn = gr.Button("이미지 생성", variant="primary")
             
         with gr.Column(scale=1):
             output_image = gr.Image(label="생성된 이미지")
@@ -591,14 +590,14 @@ with gr.Blocks() as demo:
                 gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), 
                 gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), 
                 gr.update(visible=True), gr.update(visible=True), gr.update(visible=True),
-                gr.update(visible=True), "", gr.update(visible=True))  # 버튼 그룹 표시 추가
+                gr.update(visible=True), "", gr.update(visible=True))  # 프롬프트 생성 버튼 표시
     
     background_change_btn.click(
         fn=show_background_options,
         inputs=[],
         outputs=[prompt_input, current_selection, background_header, product_name, product_category, 
                  background_type, background_style, additional_header, lighting, texture, 
-                 color_tone, mood, selection_preview, buttons_row]
+                 color_tone, mood, selection_preview, prompt_generate_btn]
     )
     
     # 다른 기능 버튼들도 간단한 방식으로 수정
@@ -607,14 +606,14 @@ with gr.Blocks() as demo:
                 gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), 
                 gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), 
                 gr.update(visible=False), gr.update(visible=False), gr.update(visible=False),
-                gr.update(visible=False), "", gr.update(visible=False))  # 버튼 그룹 숨김 추가
+                gr.update(visible=False), "", gr.update(visible=False))  # 프롬프트 생성 버튼 숨김
     
     image_change_btn.click(
         fn=lambda: hide_background_options("#1 이미지의 [다른 모습]으로 변경하라."),
         inputs=[],
         outputs=[prompt_input, current_selection, background_header, product_name, product_category, 
                  background_type, background_style, additional_header, lighting, texture, 
-                 color_tone, mood, selection_preview, buttons_row]
+                 color_tone, mood, selection_preview, prompt_generate_btn]
     )
     
     text_remove_btn.click(
@@ -622,7 +621,7 @@ with gr.Blocks() as demo:
         inputs=[],
         outputs=[prompt_input, current_selection, background_header, product_name, product_category, 
                  background_type, background_style, additional_header, lighting, texture, 
-                 color_tone, mood, selection_preview, buttons_row]
+                 color_tone, mood, selection_preview, prompt_generate_btn]
     )
     
     clothes_change_btn.click(
@@ -630,7 +629,7 @@ with gr.Blocks() as demo:
         inputs=[],
         outputs=[prompt_input, current_selection, background_header, product_name, product_category, 
                  background_type, background_style, additional_header, lighting, texture, 
-                 color_tone, mood, selection_preview, buttons_row]
+                 color_tone, mood, selection_preview, prompt_generate_btn]
     )
     
     composite_product_btn.click(
@@ -638,7 +637,7 @@ with gr.Blocks() as demo:
         inputs=[],
         outputs=[prompt_input, current_selection, background_header, product_name, product_category, 
                  background_type, background_style, additional_header, lighting, texture, 
-                 color_tone, mood, selection_preview, buttons_row]
+                 color_tone, mood, selection_preview, prompt_generate_btn]
     )
     
     # 선택 옵션 변경 시 미리보기와 LLM 프롬프트 업데이트
@@ -746,7 +745,4 @@ with gr.Blocks() as demo:
         
         > **팁**: 선택 내용 미리보기에서 생성될 프롬프트를 확인할 수 있습니다.
         """
-    )
-
-if __name__ == "__main__":
-    demo.launch(share=True)
+    )