Update app.py

app.py

@@ -33,7 +33,6 @@ COLORFUL_PATTERN_BACKGROUNDS = {}
 ABSTRACT_BACKGROUNDS = {}
 JEWELRY_BACKGROUNDS = {}  # 쥬얼리 배경 전역 변수 추가
 
-
 # ------------------- 배경 JSON 파일 로드 함수 -------------------
 def load_background_json(filename):
     """배경 JSON 파일 로드 함수"""
@@ -56,22 +55,18 @@ def initialize_backgrounds():
     global SIMPLE_BACKGROUNDS, STUDIO_BACKGROUNDS, NATURE_BACKGROUNDS, INDOOR_BACKGROUNDS
     global TECHNOLOGY_BACKGROUNDS, COLORFUL_PATTERN_BACKGROUNDS, ABSTRACT_BACKGROUNDS
     global JEWELRY_BACKGROUNDS  # 쥬얼리 배경 추가
-
     
-    # 디렉토리 내 모든 파일 로깅
     logger.info(f"Backgrounds 디렉토리 경로: {BACKGROUNDS_DIR}")
     logger.info(f"디렉토리 내 파일 목록: {os.listdir(BACKGROUNDS_DIR)}")
     
-    # 각 배경 파일 로드
     SIMPLE_BACKGROUNDS = load_background_json("simple_backgrounds.json")
     STUDIO_BACKGROUNDS = load_background_json("studio_backgrounds.json")
     NATURE_BACKGROUNDS = load_background_json("nature_backgrounds.json")
     INDOOR_BACKGROUNDS = load_background_json("indoor_backgrounds.json")
-    TECHNOLOGY_BACKGROUNDS = load_background_json("tech-backgrounds-final.json")  # 파일 이름 수정
-    COLORFUL_PATTERN_BACKGROUNDS = load_background_json("colorful-pattern-backgrounds.json")  # 파일 이름 수정
+    TECHNOLOGY_BACKGROUNDS = load_background_json("tech-backgrounds-final.json")
+    COLORFUL_PATTERN_BACKGROUNDS = load_background_json("colorful-pattern-backgrounds.json")
     ABSTRACT_BACKGROUNDS = load_background_json("abstract_backgrounds.json")
     JEWELRY_BACKGROUNDS = load_background_json("jewelry_backgrounds.json")
-
     
     # 기본값 설정 (파일이 없거나 비어있는 경우)
     if not SIMPLE_BACKGROUNDS:
@@ -93,10 +88,9 @@ def initialize_backgrounds():
     
     logger.info("모든 배경 옵션 초기화 완료")
 
-# 배경 드롭다운 초기화를 위한 함수 추가
+# 배경 드롭다운 초기화를 위한 함수
 def initialize_dropdowns():
     """드롭다운 메뉴 초기화 함수"""
-    # 각 배경 유형별 드롭다운 선택 목록 생성
     simple_choices = list(SIMPLE_BACKGROUNDS.keys())
     studio_choices = list(STUDIO_BACKGROUNDS.keys())
     nature_choices = list(NATURE_BACKGROUNDS.keys())
@@ -105,8 +99,6 @@ def initialize_dropdowns():
     colorful_choices = list(COLORFUL_PATTERN_BACKGROUNDS.keys())
     abstract_choices = list(ABSTRACT_BACKGROUNDS.keys())
     jewelry_choices = list(JEWELRY_BACKGROUNDS.keys())
-
-
     
     return {
         "simple": simple_choices,
@@ -116,10 +108,9 @@ def initialize_dropdowns():
         "tech": tech_choices,
         "colorful": colorful_choices,
         "abstract": abstract_choices,
-        "jewelry": jewelry_choices,  # 새로 추가
+        "jewelry": jewelry_choices,
     }
 
-
 # ------------------- 기본 유틸리티 함수 -------------------
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
@@ -191,7 +182,6 @@ def generate_with_images(prompt, images, variation_index=0):
         client = genai.Client(api_key=api_key)
         logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}, 변형 인덱스: {variation_index}")
         
-        # 향상된 변형 접미사
         variation_suffixes = [
             " Create this as a professional studio product shot with precise focus on the product details. Do not add any text, watermarks, or labels to the image.",
             " Create this as a high-contrast artistic studio shot with dramatic lighting and shadows. Do not add any text, watermarks, or labels to the image.",
@@ -210,16 +200,15 @@ def generate_with_images(prompt, images, variation_index=0):
                 contents.append(img)
                 logger.info(f"이미지 #{idx} 추가됨")
         
-        # 향상된 생성 설정 - 더 높은 temperature로 창의성 증가, 더 높은 max_output_tokens
         response = client.models.generate_content(
             model="gemini-2.0-flash-exp-image-generation",
             contents=contents,
             config=types.GenerateContentConfig(
                 response_modalities=['Text', 'Image'],
-                temperature=1.05,  # 약간 높은 온도로 더 다양한 결과
-                top_p=0.97,       # 약간 높임
-                top_k=50,         # 다양성 증가
-                max_output_tokens=10240  # 더 높은 토큰 한도
+                temperature=1.05,
+                top_p=0.97,
+                top_k=50,
+                max_output_tokens=10240
             )
         )
         
@@ -274,21 +263,16 @@ def process_images_with_prompt(image1, prompt, variation_index=0, max_retries=3)
 # ------------------- 프롬프트 관련 함수 -------------------
 def filter_prompt_only(prompt):
     """Gemini의 설명 및 불필요한 메시지를 제거하고 실제 프롬프트만 추출하는 함수"""
-    
-    # 코드 블록 내부의 프롬프트 찾기
     code_block_pattern = r"```\s*(.*?)```"
     code_match = re.search(code_block_pattern, prompt, re.DOTALL)
     if code_match:
         return code_match.group(1).strip()
     
-    # Midjourney 파라미터를 포함하는 프롬프트 부분 찾기
     if "--ar 1:1" in prompt:
         lines = prompt.split('\n')
         prompt_lines = []
         in_prompt = False
-        
         for line in lines:
-            # 프롬프트 시작 부분 인식 (일반적인 제품 설명이나 'Magazine-worthy' 같은 키워드로 시작)
             if (not in_prompt and 
                 ("product" in line.lower() or 
                  "magazine" in line.lower() or 
@@ -296,18 +280,13 @@ def filter_prompt_only(prompt):
                  "photography" in line.lower())):
                 in_prompt = True
                 prompt_lines.append(line)
-            # 이미 프롬프트 영역에 있는 경우 계속 추가
             elif in_prompt:
-                # 설명이나 메타 텍스트가 시작되면 중단
                 if "explanation" in line.lower() or "let me know" in line.lower():
                     break
                 prompt_lines.append(line)
-        
-        # 프롬프트 라인 합치기
         if prompt_lines:
             return '\n'.join(prompt_lines).strip()
     
-    # 위 방법으로 찾지 못한 경우 원본 반환
     return prompt.strip()
 
 def get_selected_background_info(bg_type, simple, studio, nature, indoor, tech, colorful, abstract, jewelry):
@@ -354,14 +333,12 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, tech,
             "name": abstract,
             "english": ABSTRACT_BACKGROUNDS.get(abstract, "abstract background")
         }
-
     elif bg_type == "쥬얼리 배경":
         return {
             "category": "쥬얼리 배경",
             "name": jewelry,
             "english": JEWELRY_BACKGROUNDS.get(jewelry, "jewelry backdrop")
         }
-
     else:
         return {
             "category": "기본 배경",
@@ -369,14 +346,11 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, tech,
             "english": "white background"
         }
 
-# ------------------- 프롬프트 생성 함수 -------------------
 def generate_enhanced_system_instruction():
     """향상된 시스템 인스트럭션 생성 함수"""
     return """당신은 상품 이미지의 배경을 변경하기 위한 최고 품질의 프롬프트를 생성하는 전문가입니다.
 사용자가 제공하는 상품명, 배경 유형, 추가 요청사항을 바탕으로 미드저니(Midjourney)에 사용할 수 있는 상세하고 전문적인 프롬프트를 영어로 생성해주세요.
-
 다음 가이드라인을 반드시 준수해야 합니다:
-
 1. 상품을 "#1"로 지정하여 참조합니다. (예: "skincare tube (#1)")
 2. *** 매우 중요: 상품의 원래 특성(디자인, 색상, 형태, 로고, 패키지 등)은 어떤 상황에서도 절대 변경하지 않습니다. ***
 3. *** 상품의 본질적 특성을 유지하되, 상품에 포커스를 맞춰 모든 세부 사항이 선명하게 드러나도록 하며, 
@@ -396,8 +370,8 @@ def generate_enhanced_system_instruction():
    - "ultra high definition"
    - "crystal clear details"
 10. 사용자가 제공한 구체적인 배경과 추가 요청사항을 프롬프트에 정확히 반영하고 확장합니다.
-11. 프롬프트 끝에 "--ar 1:1 --s 750 --q 2 --v 5.2" 파라미터를 추가하여 미드저니에서 고품질 정사각형 비율을 강제합니다.
-12. 매우 중요: 프롬프트 외에 다른 설명이나 메타 텍스트를 포함하지 마세요. 예를 들어 "Here's a prompt for you" 또는 "Let me know if you need adjustments" 같은 메시지나 설명을 포함하지 마세요. 오직 프롬프트 자체만 제공하세요.
+11. 프롬프트 끝에 미드저니 파라미터 "--ar 1:1 --s 750 --q 2 --v 5.2" 파라미터를 추가하여 미드저니에서 고품질 정사각형 비율을 강제합니다.
+12. 매우 중요: 프롬프트 외에 다른 설명이나 메타 텍스트를 포함하지 마세요. 오직 프롬프트 자체만 제공하세요.
 """
 
 def generate_prompt_with_gemini(product_name, background_info, additional_info=""):
@@ -409,14 +383,12 @@ def generate_prompt_with_gemini(product_name, background_info, additional_info="
     try:
         genai_generative.configure(api_key=GEMINI_API_KEY)
         
-        # 더 상세한 프롬프트 요청 템플릿
         prompt_request = f"""
 상품명: {product_name}
 배경 유형: {background_info.get('english', 'studio')}
 배경 카테고리: {background_info.get('category', '')}
 배경 이름: {background_info.get('name', '')}
 추가 요청사항: {additional_info}
-
 중요 요구사항:
 1. 상품(#1)이 이미지 구도에서 중심적인 위치를 차지하며 적절한 크기(이미지의 60-70%)로 표현되도록 프롬프트를 생성해주세요.
 2. 이미지는 정확히 1:1 비율(정사각형)이어야 합니다.
@@ -431,29 +403,25 @@ def generate_prompt_with_gemini(product_name, background_info, additional_info="
 7. 색상 구성(color palette, color harmonies)을 명확히 해주세요.
 8. 고급스러운 상업 광고 품질의 이미지가 되도록 프롬프트를 작성해주세요.
 9. 프롬프트 끝에 미드저니 파라미터 "--ar 1:1 --s 750 --q 2 --v 5.2"를 추가해주세요.
-
 한국어 입력 내용을 전문적인 영어로 번역하여 반영해주세요.
 """
-        # 향상된 시스템 인스트럭션 사용
         model = genai_generative.GenerativeModel(
             'gemini-2.0-flash',
             system_instruction=generate_enhanced_system_instruction()
         )
         
-        # 더 창의적인 결과를 위한 생성 설정 조정
         response = model.generate_content(
             prompt_request,
             generation_config=genai_generative.types.GenerationConfig(
-                temperature=0.8,       # 더 창의적인 결과를 위해 온도 상향
+                temperature=0.8,
                 top_p=0.97,
                 top_k=64,
-                max_output_tokens=1600,  # 더 상세한 프롬프트 허용
+                max_output_tokens=1600,
             )
         )
         
         response_text = response.text.strip()
         
-        # 미드저니 파라미터가 없을 경우 추가
         if "--ar 1:1" not in response_text:
             response_text = response_text.rstrip(".") + ". --ar 1:1 --s 750 --q 2 --v 5.2"
             
@@ -504,51 +472,34 @@ def generate_product_images(image, bg_type, simple, studio, nature, indoor, tech
         result_img, status, _ = process_images_with_prompt(image, final_prompt, variation_index=i)
         images_list.append(result_img)
         statuses.append(f"이미지 #{i+1}: {status}")
-        time.sleep(1)  # 순차적 실행을 위한 딜레이
+        time.sleep(1)
     combined_status = "\n".join(statuses)
     return images_list[0], images_list[1], images_list[2], images_list[3], combined_status, final_prompt
 
 # ------------------- Gradio 인터페이스 구성 -------------------
 def create_app():
-    # 드롭다운 옵션 초기화
     dropdown_options = initialize_dropdowns()
     
     with gr.Blocks(title="고급 상품 이미지 배경 프롬프트 및 이미지 생성") as demo:
         gr.Markdown("# 고급 상품 이미지 배경 프롬프트 및 이미지 생성")
         gr.Markdown(
-            "상품 이미지를 업로드하고, 제품명, 배경 옵션, 추가 요청사항을 입력하면 Gemini API를 통해 영어 프롬프트를 생성하고, 해당 프롬프트로 이미지가 생성됩니다.\n\n"
+            "상품 이미지를 업로드하고, 상품명, 배경 옵션, 추가 요청사항 및 Gemini API 키를 입력하면 Gemini API를 통해 영어 프롬프트를 생성하고, 해당 프롬프트로 이미지가 생성됩니다.\n\n"
             "단일 이미지 생성과 4장 이미지 생성 모두 가능합니다."
         )
         
         with gr.Row():
-            # API 키 입력 필드 추가
+            # API 키 입력 필드만 표시 (API 키 상태, 설정 버튼 제거)
             gemini_api_key = gr.Textbox(
                 label="Gemini API 키", 
                 type="password", 
                 placeholder="API 키를 입력하세요",
                 interactive=True
             )
-
-            # API 키 설정 버튼 추가
-            api_key_btn = gr.Button("API 키 설정")
-            api_key_status = gr.Textbox(label="API 키 상태", interactive=False)
-
-        def set_api_key(api_key):
-            os.environ["GEMINI_API_KEY"] = api_key
-            return "API 키가 성공적으로 설정되었습니다."
-
-        api_key_btn.click(
-            fn=set_api_key, 
-            inputs=[gemini_api_key], 
-            outputs=[api_key_status]
-        )
-
+        
         with gr.Row():
             with gr.Column(scale=1):
                 product_name = gr.Textbox(label="상품명 (한국어 입력)", placeholder="예: 스킨케어 튜브, 스마트워치, 향수, 운동화 등", interactive=True)
-                # 상품 이미지 업로드를 상품명 아래로 이동
                 image_input = gr.Image(label="상품 이미지 업로드", type="pil")
-                
                 background_type = gr.Radio(
                     choices=["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "테크놀로지 배경", "컬러풀 패턴 배경", "추상/특수 배경", "쥬얼리 배경"],
                     label="배경 유형",
@@ -603,7 +554,6 @@ def create_app():
                     visible=False,
                     interactive=True
                 )
-                
                 jewelry_dropdown = gr.Dropdown(
                     choices=dropdown_options["jewelry"],
                     value=dropdown_options["jewelry"][0] if dropdown_options["jewelry"] else None,
@@ -611,13 +561,13 @@ def create_app():
                     visible=False,
                     interactive=True
                 )
-        
                 additional_info = gr.Textbox(
                     label="추가 요청사항 (선택사항)",
                     placeholder="예: 고급스러운 느낌, 밝은 조명, 자연스러운 보조 객체 등",
                     lines=3,
                     interactive=True
                 )
+                
                 def update_dropdowns(bg_type):
                     return {
                         simple_dropdown: gr.update(visible=(bg_type == "심플 배경")),
@@ -640,47 +590,45 @@ def create_app():
                     multi_btn = gr.Button("프롬프트 및 4장 이미지 생성", variant="primary")
             
             with gr.Column(scale=1):
-                # 공통 프롬프트 출력 영역
-                prompt_output = gr.Textbox(label="생성된 프롬프트 (영어)", lines=6)
-                
-                # 이미지 레이아웃 변경: 2x2 그리드로 배치
+                # 프롬프트 출력 UI 제거, 이미지 4장만 출력
                 with gr.Row():
                     image_output1 = gr.Image(label="이미지 #1", type="filepath")
                     image_output2 = gr.Image(label="이미지 #2", type="filepath")
-                
                 with gr.Row():
                     image_output3 = gr.Image(label="이미지 #3", type="filepath")
                     image_output4 = gr.Image(label="이미지 #4", type="filepath")
-                
                 status_output = gr.Textbox(label="결과 정보", lines=3)
         
-        # 단일 이미지 생성 함수 수정
-        def modified_single_image_gen(*args):
-            # 기존 함수 호출
-            result_img, status, prompt = generate_product_image(*args)
-            # 단일 이미지를 첫 번째 이미지 슬롯에 배치하고 나머지는 None으로 설정
-            return prompt, result_img, None, None, None, status
+        # 단일 이미지 생성을 위한 함수 (API 키 입력값 활용)
+        def modified_single_image_gen(api_key, image, bg_type, simple, studio, nature, indoor, tech, colorful, abstract, jewelry, product_name, additional_info):
+            os.environ["GEMINI_API_KEY"] = api_key.strip()
+            result_img, status, _ = generate_product_image(image, bg_type, simple, studio, nature, indoor, tech, colorful, abstract, jewelry, product_name, additional_info)
+            # 단일 이미지는 첫 번째 슬롯에 배치, 나머지는 None 반환
+            return result_img, None, None, None, status
         
         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, jewelry_dropdown, product_name, additional_info],
-            outputs=[prompt_output, image_output1, image_output2, image_output3, image_output4, status_output]
+            inputs=[gemini_api_key, 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]
         )
         
+        # 4장 이미지 생성을 위한 함수 (API 키 입력값 활용)
+        def modified_multi_image_gen(api_key, image, bg_type, simple, studio, nature, indoor, tech, colorful, abstract, jewelry, product_name, additional_info):
+            os.environ["GEMINI_API_KEY"] = api_key.strip()
+            img1, img2, img3, img4, status, _ = generate_product_images(image, bg_type, simple, studio, nature, indoor, tech, colorful, abstract, jewelry, product_name, additional_info)
+            return img1, img2, img3, img4, status
+        
         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, jewelry_dropdown, product_name, additional_info],
-            outputs=[image_output1, image_output2, image_output3, image_output4, status_output, prompt_output]
+            fn=modified_multi_image_gen,
+            inputs=[gemini_api_key, 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]
         )
     
     return demo
 
 # ------------------- 메인 실행 함수 -------------------
 if __name__ == "__main__":
-    # 배경 옵션 초기화 - JSON 파일에서 로드
     initialize_backgrounds()
-    
-    # 앱 생성 및 실행
     app = create_app()
     app.queue()
-    app.launch()
+    app.launch()