Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import google.generativeai as genai
 from PIL import Image
@@ -439,25 +440,27 @@ def create_app():
         def update_bg_subcategory(bg_main):
             subcategories = get_subcategories_background(bg_main)
             first_sub = subcategories[0] if subcategories else None
-            return {"choices": subcategories, "value": first_sub}
+            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
         
         # 배경 서브 카테고리 변경 시 구체적 배경 드롭다운 업데이트
         def update_bg_specific(bg_main, bg_sub):
             specific_options = get_specific_backgrounds(bg_main, bg_sub)
             first_specific = specific_options[0] if specific_options else None
-            return {"choices": specific_options, "value": first_specific}
+            return specific_options, first_specific
         
         # 스타일 메인 카테고리 변경 시 구체적 스타일 드롭다운 업데이트
         def update_style_specific(style_main):
             if style_main == "필요시 선택하세요":
-                return {"choices": [default_style_specific], "value": default_style_specific}
+                return [default_style_specific], default_style_specific
             specific_styles = get_specific_styles(style_main)
             first_specific = specific_styles[0] if specific_styles else None
-            return {"choices": specific_styles, "value": first_specific}
+            return specific_styles, first_specific
         
         def update_preview(prompt):
             if not prompt or "API 키" in prompt:
-                return {"visible": False}
+                return False, ""
             preview = f"""
             <div style="padding:10px; border:1px solid #ddd; border-radius:8px; margin-top:10px;">
                 <h3>프롬프트 요약</h3>
@@ -466,31 +469,39 @@ def create_app():
                 <p><strong>미드저니 파라미터:</strong> {" ".join([param for param in ["--ar 1:1", "--s 750", "--q 2"] if param in prompt])}</p>
             </div>
             """
-            return {"value": preview, "visible": True}
+            return True, preview
 
         # 드롭다운 변경 이벤트 연결
         bg_main_category.change(
             fn=update_bg_subcategory,
             inputs=[bg_main_category],
-            outputs=[bg_subcategory]
-        ).then(
-            fn=update_bg_specific,
-            inputs=[bg_main_category, bg_subcategory],
-            outputs=[bg_specific]
+            outputs=[
+                bg_subcategory.choices,
+                bg_subcategory.value,
+                bg_specific.choices,
+                bg_specific.value
+            ]
         )
         
         bg_subcategory.change(
             fn=update_bg_specific,
             inputs=[bg_main_category, bg_subcategory],
-            outputs=[bg_specific]
+            outputs=[
+                bg_specific.choices,
+                bg_specific.value
+            ]
         )
         
         style_main.change(
             fn=update_style_specific,
             inputs=[style_main],
-            outputs=[style_specific]
+            outputs=[
+                style_specific.choices,
+                style_specific.value
+            ]
         )
         
+        
         submit_btn.click(
             fn=process_inputs,
             inputs=[image_input, bg_main_category, bg_subcategory, bg_specific, style_main, style_specific, product_name, additional_info],