Update app.py

app.py

@@ -27,8 +27,8 @@ def save_binary_file(file_name, data):
     logger.debug(f"파일 저장 완료: {file_name}")
 
 
-def merge_product_and_person(person_img_path, product_img_path, prompt, model="gemini-2.0-flash-exp-image-generation"):
-    logger.debug(f"merge_product_and_person 함수 시작 - 프롬프트: '{prompt}'")
+def merge_images(person_img_path, product_img_path, background_img_path, prompt, model="gemini-2.0-flash-exp-image-generation"):
+    logger.debug(f"merge_images 함수 시작 - 프롬프트: '{prompt}'")
     
     try:
         # API 키는 환경변수에서 불러옴
@@ -46,9 +46,17 @@ def merge_product_and_person(person_img_path, product_img_path, prompt, model="g
         person_img = Image.open(person_img_path)
         product_img = Image.open(product_img_path)
         
-        # 컨텐츠 리스트 생성 (공식 문서 방식대로)
-        # 순서: 사람 이미지, 상품 이미지, 프롬프트 텍스트
-        contents = [person_img, product_img, prompt]
+        # 컨텐츠 리스트 생성
+        contents = [person_img, product_img]
+        
+        # 배경 이미지가 있으면 추가
+        if background_img_path:
+            background_img = Image.open(background_img_path)
+            contents.append(background_img)
+            logger.debug("배경 이미지 추가됨")
+        
+        # 마지막에 프롬프트 추가
+        contents.append(prompt)
         logger.debug(f"컨텐츠 객체 생성 완료: {len(contents)} 아이템")
 
         # 생성 설정
@@ -99,16 +107,22 @@ def merge_product_and_person(person_img_path, product_img_path, prompt, model="g
         return None, str(e)  # 오류 발생 시 None과 오류 메시지 반환
 
 
-def process_images_and_prompt(person_pil, product_pil, prompt):
+def process_images_and_prompt(person_pil, product_pil, background_pil, prompt):
     logger.debug(f"process_images_and_prompt 함수 시작 - 프롬프트: '{prompt}'")
     try:
         # 기본 프롬프트 설정 (비어있는 경우)
         if not prompt or not prompt.strip():
-            prompt = "이 사람이 이 상품을 사용하는 모습을 자연스럽게 보여주세요. 상품을 잘 보이게 해주세요. Make a natural composite image showing this person using this product. Make sure the product is clearly visible."
+            if background_pil:
+                prompt = "이 배경에 이 사람이 이 상품을 사용하는 모습을 자연스럽게 보여주세요. 상품을 잘 보이게 해주세요. Create a natural composite image showing this person using this product in this background setting. Make sure the product is clearly visible."
+            else:
+                prompt = "이 사람이 이 상품을 사용하는 모습을 자연스럽게 보여주세요. 상품을 잘 보이게 해주세요. Create a natural composite image showing this person using this product. Make sure the product is clearly visible."
         
         # 프롬프트에 영어가 없으면 영어 프롬프트 추가 (더 나은 결과를 위해)
         if not any(ord(c) < 128 for c in prompt):
-            prompt += " Create a realistic composite image of this person with this product."
+            if background_pil:
+                prompt += " Create a realistic composite image of this person with this product in this background."
+            else:
+                prompt += " Create a realistic composite image of this person with this product."
         
         # 이미지 저장
         with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_person:
@@ -120,11 +134,20 @@ def process_images_and_prompt(person_pil, product_pil, prompt):
             product_path = tmp_product.name
             product_pil.save(product_path)
             logger.debug(f"상품 이미지 저장 완료: {product_path}")
+        
+        # 배경 이미지 저장 (있는 경우)
+        background_path = None
+        if background_pil is not None:
+            with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_bg:
+                background_path = tmp_bg.name
+                background_pil.save(background_path)
+                logger.debug(f"배경 이미지 저장 완료: {background_path}")
 
         # 이미지 합성 실행
-        result_path, response_text = merge_product_and_person(
+        result_path, response_text = merge_images(
             person_img_path=person_path,
             product_img_path=product_path,
+            background_img_path=background_path,
             prompt=prompt
         )
 
@@ -139,12 +162,14 @@ def process_images_and_prompt(person_pil, product_pil, prompt):
             try:
                 os.unlink(person_path)
                 os.unlink(product_path)
+                if background_path:
+                    os.unlink(background_path)
             except Exception as e:
                 logger.warning(f"임시 파일 삭제 중 오류: {str(e)}")
             
             return [result_img], response_text
         else:
-            logger.error("merge_product_and_person 함���에서 None 반환됨.")
+            logger.error("merge_images 함수에서 None 반환됨.")
             return [], response_text  # 오류 시 빈 리스트 반환
 
     except Exception as e:
@@ -161,21 +186,22 @@ with gr.Blocks() as demo:
                 <img src="https://www.gstatic.com/lamda/images/gemini_favicon_f069958c85030456e93de685481c559f160ea06b.png" style="width: 100px; height: 100px;">
             </div>
             <div>
-                <h1>Gemini를 이용한 상품-사람 이미지 합성</h1>
-                <p>Gemini API 키는 환경변수(GEMINI_API_KEY)로 설정되어 있습니다.</p>
+                <h1>Gemini를 이용한 이미지 합성</h1>
+                <p>사람, 상품, 배경 이미지를 합성하는 애플리케이션입니다.</p>
             </div>
         </div>
         """
     )
-    gr.Markdown("사람 이미지와 상품 이미지를 업로드하고, 어떻게 합성할지 설명해주세요.")
+    gr.Markdown("사람 이미지, 상품 이미지, 배경 이미지를 업로드하고, 어떻게 합성할지 설명해주세요.")
 
     with gr.Row():
         with gr.Column():
-            person_input = gr.Image(type="pil", label="사람 이미지", image_mode="RGB")
-            product_input = gr.Image(type="pil", label="상품 이미지", image_mode="RGB")
+            person_input = gr.Image(type="pil", label="사람 이미지 (필수)", image_mode="RGB")
+            product_input = gr.Image(type="pil", label="상품 이미지 (필수)", image_mode="RGB")
+            background_input = gr.Image(type="pil", label="배경 이미지 (선택 사항)", image_mode="RGB")
             prompt_input = gr.Textbox(
                 lines=2,
-                placeholder="합성 방법을 설명해주세요. (예: '이 사람이 상품을 들고 있는 모습' 또는 '이 사람이 이 상품을 사용하는 모습')",
+                placeholder="합성 방법을 설명해주세요. (예: '이 배경에서 이 사람이 상품을 사용하는 모습' 또는 '이 사람이 이 상품을 들고 이 배경에 서 있는 모습')",
                 label="합성 방법 설명"
             )
             submit_btn = gr.Button("이미지 합성 실행")
@@ -185,7 +211,7 @@ with gr.Blocks() as demo:
 
     submit_btn.click(
         fn=process_images_and_prompt,
-        inputs=[person_input, product_input, prompt_input],
+        inputs=[person_input, product_input, background_input, prompt_input],
         outputs=[output_gallery, output_text],
     )
 
@@ -193,11 +219,12 @@ with gr.Blocks() as demo:
     <div style="margin-top: 20px; padding: 10px; background-color: #f8f9fa; border-radius: 8px;">
         <h3>사용 팁:</h3>
         <ul>
-            <li><strong>자연스러운 합성:</strong> "이 사람이 이 상품을 사용하는 자연스러운 모습을 만들어주세요."</li>
-            <li><strong>특정 위치 지정:</strong> "이 사람이 이 상품을 손에 들고 있는 모습을 보여주세요."</li>
-            <li><strong>상품 강조:</strong> "이 사람이 이 상품을 사용하는 모습을 보여주되, 상품이 잘 보이도록 해주세요."</li>
+            <li><strong>배경 활용:</strong> "이 배경에서 이 사람이 이 상품을 사용하는 자연스러운 모습을 만들어주세요."</li>
+            <li><strong>특정 위치 지정:</strong> "이 사람이 이 상품을 손에 들고 이 배경 앞에 서 있는 모습을 보여주세요."</li>
+            <li><strong>상품 강조:</strong> "이 사람이 이 배경에서 이 상품을 사용하는 모습을 보여주되, 상품이 잘 보이도록 해주세요."</li>
+            <li><strong>장면 설정:</strong> "이 사람이 이 상품으로 요리하는 모습을 이 주방 배경에서 보여주세요."</li>
             <li><strong>영어 프롬프트:</strong> 더 나은 결과를 위해 영어와 한국어를 함께 사용해 보세요.</li>
-            <li><strong>구체적인 지시:</strong> 합성 방법을 구체적으로 설명할수록 더 좋은 결과를 얻을 수 있습니다.</li>
+            <li><strong>배경 선택사항:</strong> 배경 이미지는 선택사항입니다. 배경 없이 사람과 상품만 합성할 수도 있습니다.</li>
         </ul>
     </div>
     """)