Update app.py

app.py

@@ -17,7 +17,7 @@ load_dotenv()
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
-# ------------------- 기본 이미지 생성 관련 함수 (기본코드) -------------------
+# ------------------- 기본 이미지 생성 관련 함수 -------------------
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
@@ -144,15 +144,8 @@ def process_images_with_prompt(image1, prompt, variation_index=0, max_retries=3)
             valid_images = [img for img in images if img is not None]
             if not valid_images:
                 return None, "이미지를 업로드해주세요.", ""
-            if prompt and prompt.strip():
-                processed_prompt = preprocess_prompt(prompt, image1)
-                if re.search("[가-힣]", processed_prompt):
-                    final_prompt = translate_prompt_to_english(processed_prompt)
-                else:
-                    final_prompt = processed_prompt
-            else:
-                final_prompt = "Please creatively transform this image into a more vivid and artistic version. Do not include any text or watermarks in the generated image."
-                logger.info("Default prompt generated for single image")
+            # 이미 Gemini API로 생성된 프롬프트는 영어이므로 추가 번역은 생략
+            final_prompt = prompt.strip()
             result_img, status = generate_with_images(final_prompt, valid_images, variation_index)
             if result_img is not None:
                 return result_img, status, final_prompt
@@ -168,34 +161,7 @@ def process_images_with_prompt(image1, prompt, variation_index=0, max_retries=3)
             time.sleep(1)
     return None, f"최대 재시도 횟수({max_retries}회) 초과 후 실패: {last_error}", prompt
 
-def generate_multiple_images(image1, prompt, progress=gr.Progress()):
-    results = []
-    statuses = []
-    prompts = []
-    num_images = 4
-    max_retries = 3
-    progress(0, desc="이미지 생성 준비 중...")
-    for i in range(num_images):
-        progress((i / num_images), desc=f"{i+1}/{num_images} 이미지 생성 중...")
-        result_img, status, final_prompt = process_images_with_prompt(image1, prompt, i, max_retries)
-        if result_img is not None:
-            results.append(result_img)
-            statuses.append(f"이미지 #{i+1}: {status}")
-            prompts.append(f"이미지 #{i+1}: {final_prompt}")
-        else:
-            results.append(None)
-            statuses.append(f"이미지 #{i+1} 생성 실패: {status}")
-            prompts.append(f"이미지 #{i+1}: {final_prompt}")
-        time.sleep(1)
-    progress(1.0, desc="이미지 생성 완료!")
-    while len(results) < 4:
-        results.append(None)
-    combined_status = "\n".join(statuses)
-    combined_prompts = "\n".join(prompts)
-    return results[0], results[1], results[2], results[3], combined_status, combined_prompts
-
-# ------------------- 상품 프롬프트 생성 관련 함수 (참조코드) -------------------
-
+# ------------------- 상품 프롬프트 생성 관련 함수 -------------------
 # 배경 JSON 파일 경로 설정
 BACKGROUNDS_DIR = "./background"
 if not os.path.exists(BACKGROUNDS_DIR):
@@ -250,7 +216,7 @@ def generate_system_instruction():
 6. 상품을 이미지의 주요 초점으로 부각시키고, 상품의 비율이 전체 이미지에서 크게 차지하도록 합니다.
 7. 상품 이미지 컷아웃(#1)의 기본 형태와 색상은 유지하면서, 선택한 환경에 자연스럽게 통합되도록 합니다.
 8. 고급스러운 상업적 이미지를 위한 다음 환경 요소들을 포함하세요:
-   - 상품과 어울리는 주변 환경/배경 요소를 추가합니다. 예를 들어, 화장품 주변에 꽃이나 허브, 음료 제품 옆에 과일, 전자제품 근처에 현대적 소품 등.
+   - 상품과 어울리는 주변 환경/배경 요소를 추가합니다.
    - 환경의 조명 효과(림 라이트, 백라이트, 소프트박스 등)를 설명합니다.
    - 상품이 환경에 자연스럽게 존재하는 것처럼 보이도록 적절한 그림자와 빛 표현을 포함합니다.
    - 상품의 용도나 장점을 간접적으로 암시하는 배경 요소를 포함합니다.
@@ -261,16 +227,10 @@ def generate_system_instruction():
    - "professional advertising imagery"
    - "studio quality"
    - "magazine advertisement quality"
-10. 배경 환경 요소를 상품 카테고리에 맞게 선택합니다:
-    - 스킨케어 제품: 깨끗한 욕실 선반, 우아한 화장대, 스파 같은 환경 등
-    - 음료 제품: 세련된 테이블, 파티 환경, 야외 피크닉 장면 등
-    - 전자 제품: 세련된 작업 공간, 현대적인 거실, 미니멀한 책상 등
-    - 패션/의류: 세련된 쇼룸, 도시 거리, 엘레강스한 라이프스타일 환경 등
-    - 식품 제품: 깔끔한 주방, 식탁, 요리 환경 등
+10. 배경 환경 요소를 상품 카테고���에 맞게 선택합니다.
 11. 사용자가 제공한 구체적인 배경과 추가 요청사항을 정확히 반영합니다.
 12. 프롬프트는 미드저니 AI에 최적화되어야 합니다.
 13. 프롬프트 끝에 "--ar 1:1 --s 750 --q 2" 파라미터를 추가하여 미드저니에서 고품질 정사각형 비율을 강제합니다.
-출력 형식은 영어로 된 단일 단락의 상세한 프롬프트여야 하며, 끝에 미드저니 파라미터가 포함되어야 합니다.
 """
 
 def generate_prompt_with_gemini(product_name, background_info, additional_info=""):
@@ -354,91 +314,40 @@ def get_selected_background_info(bg_type, simple, studio, nature, indoor, abstra
             "english": "white background"
         }
 
-def generate_product_prompt_output(image, bg_type, simple, studio, nature, indoor, abstract, product_name, additional_info):
+# ------------------- 통합 기능: 프롬프트 생성 후 이미지 생성 -------------------
+def generate_product_image(image, bg_type, simple, studio, nature, indoor, abstract, product_name, additional_info):
     if image is None:
-        gr.Warning("이미지를 업로드해주세요.")
-        return "이미지를 업로드해주세요.", None, "이미지를 업로드 후 프롬프트를 생성해주세요."
+        return None, "이미지를 업로드해주세요.", "이미지를 업로드 후 프롬프트를 생성해주세요."
     product_name = product_name.strip() or "제품"
     background_info = get_selected_background_info(bg_type, simple, studio, nature, indoor, abstract)
-    try:
-        prompt = generate_prompt_with_gemini(product_name, background_info, additional_info)
-        if "Gemini API 키가 설정되지 않았습니다" in prompt:
-            gr.Warning("Gemini API 키가 설정되지 않았습니다. 키를 발급받아 설정해주세요.")
-            prompt = """
-[Gemini API 키 누락]
-API 키 설정 방법:
-1. 환경 변수: export GEMINI_API_KEY="your-api-key"
-2. 코드 내 직접 입력: GEMINI_API_KEY = "your-api-key"
-키 발급: https://makersuite.google.com/
-"""
-            return prompt, image, "API 키를 설정해야 합니다."
-        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> {background_info['category']} &gt; {background_info['name']}</p>
-    <p><strong>미드저니 파라미터:</strong> {" ".join([param for param in ["--ar 1:1", "--s 750", "--q 2"] if param in prompt])}</p>
-</div>
-"""
-        return prompt, image, preview
-    except Exception as e:
-        error_msg = f"프롬프트 생성 중 오류 발생: {str(e)}"
-        gr.Error(error_msg)
-        return error_msg, image, "오류가 발생했습니다."
-
-# ------------------- Gradio 인터페이스 구성 -------------------
-
-def create_basic_app():
-    with gr.Blocks() as basic_app:
-        with gr.Column():
-            with gr.Group():
-                with gr.Column():
-                    with gr.Row():
-                        image1_input = gr.Image(type="pil", label="이미지 업로드", image_mode="RGB")
-                    prompt_input = gr.Textbox(
-                        lines=3,
-                        placeholder="프롬프트를 입력하세요. '#1'으로 업로드한 이미지를 참조할 수 있습니다.",
-                        label="프롬프트 입력"
-                    )
-                    with gr.Row():
-                        submit_single_btn = gr.Button('이미지 생성 (1장)')
-                        submit_btn = gr.Button('이미지 생성 (4장)')
-                with gr.Column():
-                    with gr.Row():
-                        with gr.Column():
-                            output_image1 = gr.Image(label="이미지 #1", type="filepath")
-                            output_image3 = gr.Image(label="이미지 #3", type="filepath")
-                        with gr.Column():
-                            output_image2 = gr.Image(label="이미지 #2", type="filepath")
-                            output_image4 = gr.Image(label="이미지 #4", type="filepath")
-                    output_text = gr.Textbox(label="결과 정보", lines=2)
-                    prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", lines=2)
-        submit_single_btn.click(
-            fn=lambda image1, prompt: process_images_with_prompt(image1, prompt, 0),
-            inputs=[image1_input, prompt_input],
-            outputs=[output_image1, output_text, prompt_display],
+    # Gemini API로 프롬프트 생성
+    generated_prompt = generate_prompt_with_gemini(product_name, background_info, additional_info)
+    if "Gemini API 키가 설정되지 않았습니다" in generated_prompt:
+        warning_msg = (
+            "[Gemini API 키 누락]\n"
+            "API 키 설정 방법:\n"
+            "1. 환경 변수: export GEMINI_API_KEY=\"your-api-key\"\n"
+            "2. 코드 내 직접 입력: GEMINI_API_KEY = \"your-api-key\"\n"
+            "키 발급: https://makersuite.google.com/"
         )
-        submit_btn.click(
-            fn=generate_multiple_images,
-            inputs=[image1_input, prompt_input],
-            outputs=[output_image1, output_image2, output_image3, output_image4, output_text, prompt_display],
-        )
-    return basic_app
+        return None, warning_msg, warning_msg
+    # 생성된 프롬프트를 프롬프트 입력창에 반영하고, 그 프롬프트로 이미지 생성
+    result_image, status, final_prompt = process_images_with_prompt(image, generated_prompt, 0)
+    return result_image, status, final_prompt
 
-def create_product_prompt_app():
-    with gr.Blocks(title="고급 상품 이미지 배경 프롬프트 생성기") as product_app:
-        gr.Markdown("# 고급 상품 이미지 배경 프롬프트 생성기")
-        gr.Markdown("상품 이미지를 업로드하고 옵션을 선택하면 고품질 상업용 프롬프트가 생성됩니다.")
-        with gr.Row():
-            product_name = gr.Textbox(label="상품명 (한국어 입력)", placeholder="예: 스킨케어 튜브, 텀블러 등", interactive=True)
-        background_type = gr.Radio(
-            choices=["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "추상/특수 배경"],
-            label="배경 유형",
-            value="심플 배경"
-        )
+# ------------------- Gradio 인터페이스 구성 -------------------
+def create_app():
+    with gr.Blocks(title="고급 상품 이미지 배경 프롬프트 생성 및 이미지 생성") as demo:
+        gr.Markdown("# 고급 상품 이미지 배경 프롬프트 생성 및 이미지 생성")
+        gr.Markdown("상품 이미지를 업로드하고, 제품명 및 배경 옵션과 추가 요청사항을 입력하면 Gemini API를 이용하여 영어 프롬프트를 생성하고, 그 프롬프트로 이미지를 생성합니다.")
         with gr.Row():
             with gr.Column(scale=1):
-                image_input = gr.Image(label="상품 이미지 업로드", type="pil")
+                product_name = gr.Textbox(label="상품명 (한국어 입력)", placeholder="예: 스킨케어 튜브, 텀블러 등", interactive=True)
+                background_type = gr.Radio(
+                    choices=["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "추상/특수 배경"],
+                    label="배경 유형",
+                    value="심플 배경"
+                )
                 simple_dropdown = gr.Dropdown(
                     choices=list(SIMPLE_BACKGROUNDS.keys()),
                     value=list(SIMPLE_BACKGROUNDS.keys())[0] if SIMPLE_BACKGROUNDS else None,
@@ -476,63 +385,38 @@ def create_product_prompt_app():
                 )
                 additional_info = gr.Textbox(
                     label="추가 요청사항 (선택사항)",
-                    placeholder="예: 고급스러운 느낌, 밝은 조명, 자연스러운 보조적인 객체를 추가해주세요 등",
+                    placeholder="예: 고급스러운 느낌, 밝은 조명, 자연스러운 보조 객체 등",
                     lines=3,
                     interactive=True
                 )
-                submit_btn = gr.Button("프롬프트 생성", variant="primary")
+                # 배경 유형 변경 시 해당 드롭다운만 보이도록 업데이트
+                def update_dropdowns(bg_type):
+                    return {
+                        simple_dropdown: gr.update(visible=(bg_type == "심플 배경")),
+                        studio_dropdown: gr.update(visible=(bg_type == "스튜디오 배경")),
+                        nature_dropdown: gr.update(visible=(bg_type == "자연 환경")),
+                        indoor_dropdown: gr.update(visible=(bg_type == "실내 환경")),
+                        abstract_dropdown: gr.update(visible=(bg_type == "추상/특수 배경"))
+                    }
+                background_type.change(
+                    fn=update_dropdowns,
+                    inputs=[background_type],
+                    outputs=[simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, abstract_dropdown]
+                )
+                image_input = gr.Image(label="상품 이미지 업로드", type="pil")
+                generate_btn = gr.Button("프롬프트 및 이미지 생성", variant="primary")
             with gr.Column(scale=1):
-                prompt_output = gr.Textbox(label="생성된 프롬프트", lines=10)
-                image_preview = gr.Image(label="업로드된 이미지 (#1)", type="pil")
-                preview_html = gr.HTML("프롬프트 생성 시 여기에 미리보기가 표시됩니다.")
-        def update_dropdowns(bg_type):
-            return {
-                simple_dropdown: gr.update(visible=(bg_type == "심플 배경")),
-                studio_dropdown: gr.update(visible=(bg_type == "스튜디오 배경")),
-                nature_dropdown: gr.update(visible=(bg_type == "자연 환경")),
-                indoor_dropdown: gr.update(visible=(bg_type == "실내 환경")),
-                abstract_dropdown: gr.update(visible=(bg_type == "추상/특수 배경"))
-            }
-        background_type.change(
-            fn=update_dropdowns,
-            inputs=[background_type],
-            outputs=[simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, abstract_dropdown]
+                prompt_output = gr.Textbox(label="생성된 프롬프트 (영어)", lines=6)
+                image_output = gr.Image(label="생성된 이미지", type="filepath")
+                status_output = gr.Textbox(label="결과 정보", lines=3)
+        generate_btn.click(
+            fn=generate_product_image,
+            inputs=[image_input, background_type, simple_dropdown, studio_dropdown, nature_dropdown, indoor_dropdown, abstract_dropdown, product_name, additional_info],
+            outputs=[image_output, status_output, prompt_output]
         )
-        def generate_output(image, bg_type, simple, studio, nature, indoor, abstract, product_name, additional_text):
-            return generate_product_prompt_output(image, bg_type, simple, studio, nature, indoor, abstract, product_name, additional_text)
-        submit_btn.click(
-            fn=generate_output,
-            inputs=[
-                image_input,
-                background_type,
-                simple_dropdown,
-                studio_dropdown,
-                nature_dropdown,
-                indoor_dropdown,
-                abstract_dropdown,
-                product_name,
-                additional_info
-            ],
-            outputs=[
-                prompt_output,
-                image_preview,
-                preview_html
-            ]
-        )
-    return product_app
-
-def main():
-    basic_app = create_basic_app()
-    product_app = create_product_prompt_app()
-    with gr.Blocks() as demo:
-        gr.Markdown("# 통합 이미지 생성 및 프롬프트 생성 앱")
-        with gr.Tabs():
-            with gr.TabItem("일반 이미지 생성"):
-                basic_app.render()
-            with gr.TabItem("상품 프롬프트 생성 및 이미지 생성"):
-                product_app.render()
-    demo.queue()
-    demo.launch()
+    return demo
 
 if __name__ == "__main__":
-    main()
+    app = create_app()
+    app.queue()
+    app.launch()