Update app.py

app.py

@@ -31,6 +31,8 @@ INDOOR_BACKGROUNDS = {}
 TECHNOLOGY_BACKGROUNDS = {}
 COLORFUL_PATTERN_BACKGROUNDS = {}
 ABSTRACT_BACKGROUNDS = {}
+JEWELRY_BACKGROUNDS = {}  # 쥬얼리 배경 전역 변수 추가
+
 
 # ------------------- 배경 JSON 파일 로드 함수 -------------------
 def load_background_json(filename):
@@ -462,11 +464,11 @@ def generate_prompt_with_gemini(product_name, background_info, additional_info="
         return f"프롬프트 생성 중 오류가 발생했습니다: {str(e)}"
 
 # ------------------- 단일 이미지 생성 함수 -------------------
-def generate_product_image(image, bg_type, simple, studio, nature, indoor, tech, colorful, abstract, product_name, additional_info):
+def generate_product_image(image, bg_type, simple, studio, nature, indoor, tech, colorful, abstract, jewelry, product_name, additional_info):
     if image is None:
         return None, "이미지를 업로드해주세요.", "이미지를 업로드 후 프롬프트를 생성해주세요."
     product_name = product_name.strip() or "제품"
-    background_info = get_selected_background_info(bg_type, simple, studio, nature, indoor, tech, colorful, abstract)
+    background_info = get_selected_background_info(bg_type, simple, studio, nature, indoor, tech, colorful, abstract, jewelry)
     generated_prompt = generate_prompt_with_gemini(product_name, background_info, additional_info)
     if "Gemini API 키가 설정되지 않았습니다" in generated_prompt:
         warning_msg = (
@@ -482,11 +484,11 @@ def generate_product_image(image, bg_type, simple, studio, nature, indoor, tech,
     return result_image, status, final_prompt
 
 # ------------------- 4장 이미지 생성 함수 -------------------
-def generate_product_images(image, bg_type, simple, studio, nature, indoor, tech, colorful, abstract, product_name, additional_info):
+def generate_product_images(image, bg_type, simple, studio, nature, indoor, tech, colorful, abstract, jewelry, product_name, additional_info):
     if image is None:
         return None, None, None, None, "이미지를 업로드해주세요.", "이미지를 업로드 후 프롬프트를 생성해주세요."
     product_name = product_name.strip() or "제품"
-    background_info = get_selected_background_info(bg_type, simple, studio, nature, indoor, tech, colorful, abstract)
+    background_info = get_selected_background_info(bg_type, simple, studio, nature, indoor, tech, colorful, abstract, jewelry)
     generated_prompt = generate_prompt_with_gemini(product_name, background_info, additional_info)
     if "Gemini API 키가 설정되지 않았습니다" in generated_prompt:
         warning_msg = (
@@ -640,13 +642,13 @@ def create_app():
         
         single_btn.click(
             fn=modified_single_image_gen,
-            inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, tech_dropdown, colorful_dropdown, abstract_dropdown, product_name, additional_info],
+            inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, tech_dropdown, colorful_dropdown, abstract_dropdown, jewelry_dropdown, product_name, additional_info],
             outputs=[prompt_output, image_output1, image_output2, image_output3, image_output4, status_output]
         )
         
         multi_btn.click(
             fn=generate_product_images,
-            inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, tech_dropdown, colorful_dropdown, abstract_dropdown, product_name, additional_info],
+            inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, tech_dropdown, colorful_dropdown, abstract_dropdown, jewelry_dropdown, product_name, additional_info],
             outputs=[image_output1, image_output2, image_output3, image_output4, status_output, prompt_output]
         )