Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr
 import google.generativeai as genai
 from PIL import Image
@@ -436,84 +435,95 @@ def create_app():
                 image_preview = gr.Image(label="업로드된 이미지 (#1)", type="pil")
                 preview_html = gr.HTML("프롬프트 생성 시 여기에 미리보기가 표시됩니다.", visible=True)
         
-        # 배경 메인 카테고리 변경 시 서브 및 구체적 드롭다운 업데이트
-        def update_bg_subcategory(bg_main):
+        # 이벤트 핸들러 함수들
+        def on_bg_main_change(bg_main):
             subcategories = get_subcategories_background(bg_main)
             first_sub = subcategories[0] if subcategories else None
             specific_options = get_specific_backgrounds(bg_main, first_sub) if first_sub else []
             first_specific = specific_options[0] if specific_options else None
-            return subcategories, first_sub, specific_options, first_specific
+            
+            return {
+                "choices": subcategories,
+                "value": first_sub
+            }, {
+                "choices": specific_options,
+                "value": first_specific
+            }
         
-        # 배경 서브 카테고리 변경 시 구체적 배경 드롭다운 업데이트
-        def update_bg_specific(bg_main, bg_sub):
+        def on_bg_sub_change(bg_main, bg_sub):
             specific_options = get_specific_backgrounds(bg_main, bg_sub)
             first_specific = specific_options[0] if specific_options else None
-            return specific_options, first_specific
+            
+            return {
+                "choices": specific_options,
+                "value": first_specific
+            }
         
-        # 스타일 메인 카테고리 변경 시 구체적 스타일 드롭다운 업데이트
-        def update_style_specific(style_main):
+        def on_style_main_change(style_main):
             if style_main == "필요시 선택하세요":
-                return [default_style_specific], default_style_specific
+                return {
+                    "choices": [default_style_specific],
+                    "value": default_style_specific
+                }
+            
             specific_styles = get_specific_styles(style_main)
             first_specific = specific_styles[0] if specific_styles else None
-            return specific_styles, first_specific
+            
+            return {
+                "choices": specific_styles,
+                "value": first_specific
+            }
         
-        def update_preview(prompt):
-            if not prompt or "API 키" in prompt:
-                return False, ""
-            preview = f"""
-            <div style="padding:10px; border:1px solid #ddd; border-radius:8px; margin-top:10px;">
-                <h3>프롬프트 요약</h3>
-                <p><strong>총 길이:</strong> {len(prompt)} 글자</p>
-                <p><strong>주요 요소:</strong> {", ".join([kw for kw in ["commercial photography", "product", "square format", "centered", "detailed"] if kw.lower() in prompt.lower()])}</p>
-                <p><strong>미드저니 파라미터:</strong> {" ".join([param for param in ["--ar 1:1", "--s 750", "--q 2"] if param in prompt])}</p>
-            </div>
-            """
-            return True, preview
+        def on_submit(img, bg_main, bg_sub, bg_specific, style_main, style_specific, prod_name, add_info):
+            output, preview_img = process_inputs(img, bg_main, bg_sub, bg_specific, style_main, style_specific, prod_name, add_info)
+            
+            preview_visible = not (not output or "API 키" in output)
+            preview_content = ""
+            
+            if preview_visible:
+                preview_content = f"""
+                <div style="padding:10px; border:1px solid #ddd; border-radius:8px; margin-top:10px;">
+                    <h3>프롬프트 요약</h3>
+                    <p><strong>총 길이:</strong> {len(output)} 글자</p>
+                    <p><strong>주요 요소:</strong> {", ".join([kw for kw in ["commercial photography", "product", "square format", "centered", "detailed"] if kw.lower() in output.lower()])}</p>
+                    <p><strong>미드저니 파라미터:</strong> {" ".join([param for param in ["--ar 1:1", "--s 750", "--q 2"] if param in output])}</p>
+                </div>
+                """
+            
+            return output, preview_img, preview_content, preview_visible
 
-        # 드롭다운 변경 이벤트 연결
+        # 이벤트 연결
         bg_main_category.change(
-            fn=update_bg_subcategory,
-            inputs=[bg_main_category],
-            outputs=[
-                bg_subcategory.choices,
-                bg_subcategory.value,
-                bg_specific.choices,
-                bg_specific.value
-            ]
+            fn=on_bg_main_change,
+            inputs=bg_main_category,
+            outputs=[bg_subcategory, bg_specific]
         )
         
         bg_subcategory.change(
-            fn=update_bg_specific,
+            fn=on_bg_sub_change,
             inputs=[bg_main_category, bg_subcategory],
-            outputs=[
-                bg_specific.choices,
-                bg_specific.value
-            ]
+            outputs=bg_specific
         )
         
         style_main.change(
-            fn=update_style_specific,
-            inputs=[style_main],
-            outputs=[
-                style_specific.choices,
-                style_specific.value
-            ]
+            fn=on_style_main_change,
+            inputs=style_main,
+            outputs=style_specific
         )
         
-        
         submit_btn.click(
-            fn=process_inputs,
-            inputs=[image_input, bg_main_category, bg_subcategory, bg_specific, style_main, style_specific, product_name, additional_info],
-            outputs=[prompt_output, image_preview]
-        ).then(
-            fn=update_preview,
-            inputs=[prompt_output],
-            outputs=[preview_html]
+            fn=on_submit,
+            inputs=[
+                image_input, bg_main_category, bg_subcategory, bg_specific, 
+                style_main, style_specific, product_name, additional_info
+            ],
+            outputs=[
+                prompt_output, image_preview, preview_html, preview_html
+            ]
         )
     
     return demo
 
 if __name__ == "__main__":
     app = create_app()
-    app.launch()
+    app.launch()