Update app.py

app.py

@@ -41,9 +41,25 @@ def process_images_with_prompt(image1, image2, image3, prompt):
         # Gemini 클라이언트 초기화
         client = genai.Client(api_key=api_key)
         
+        # 이미지 개수 확인
+        images = [image1, image2, image3]
+        valid_images = [img for img in images if img is not None]
+        
+        if not valid_images:
+            return None, "적어도 하나의 이미지를 업로드해주세요."
+        
         # 프롬프트 처리
         if not prompt or not prompt.strip():
-            prompt = "이 이미지들을 활용하여 새로운 이미지를 생성해주세요."
+            # 프롬프트가 없으면 업로드된 이미지 수에 따라 자동 합성 프롬프트 생성
+            if len(valid_images) == 1:
+                prompt = "이 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
+                logger.info("단일 이미지 프롬프트 자동 생성")
+            elif len(valid_images) == 2:
+                prompt = "이 두 이미지를 자연스럽게 합성해주세요. 두 이미지의 요소를 조화롭게 통합하여 하나의 이미지로 만들어주세요."
+                logger.info("두 이미지 합성 프롬프트 자동 생성")
+            else:
+                prompt = "이 세 이미지를 창의적으로 합성해주세요. 모든 이미지의 주요 요소를 포함하되 자연스럽고 일관된 하나의 장면으로 만들어주세요."
+                logger.info("세 이미지 합성 프롬프트 자동 생성")
         else:
             # #1, #2, #3 참조 처리
             prompt = preprocess_prompt(prompt)
@@ -55,8 +71,7 @@ def process_images_with_prompt(image1, image2, image3, prompt):
         parts.append(types.Part.from_text(text=prompt))
         
         # 사용 가능한 이미지 추가
-        images_added = 0
-        for idx, img in enumerate([image1, image2, image3], 1):
+        for idx, img in enumerate(images, 1):
             if img is not None:
                 # PIL 이미지를 바이트로 변환
                 with tempfile.NamedTemporaryFile(suffix=".png") as tmp:
@@ -67,12 +82,8 @@ def process_images_with_prompt(image1, image2, image3, prompt):
                 
                 # 이미지를 파트로 추가
                 parts.append(types.Part.from_data(data=image_bytes, mime_type="image/png"))
-                images_added += 1
                 logger.info(f"이미지 #{idx} 추가됨")
         
-        if images_added == 0:
-            return None, "적어도 하나의 이미지를 업로드해주세요."
-        
         # 생성 설정
         generate_content_config = types.GenerateContentConfig(
             temperature=1,
@@ -119,7 +130,7 @@ with gr.Blocks() as demo:
         """
         <div style="text-align: center; margin-bottom: 1rem;">
             <h1>간단한 이미지 생성기</h1>
-            <p>이미지와 프롬프트를 입력하세요. #1, #2, #3으로 각 이미지를 참조할 수 있습니다.</p>
+            <p>이미지를 업로드하고 바로 실행하면 자동으로 합성합니다. 또는 #1, #2, #3으로 각 이미지를 참조하는 프롬프트를 입력할 수 있습니다.</p>
         </div>
         """
     )
@@ -132,11 +143,11 @@ with gr.Blocks() as demo:
                 image2_input = gr.Image(type="pil", label="#2", image_mode="RGB")
                 image3_input = gr.Image(type="pil", label="#3", image_mode="RGB")
             
-            # 프롬프트 입력
+            # 프롬프트 입력 (선택 사항)
             prompt_input = gr.Textbox(
                 lines=3,
-                placeholder="프롬프트를 입력하세요. 예: '#1과 #2를 합성해 주세요' 또는 '#1의 인물을 #3의 배경에 넣어주세요'",
-                label="프롬프트"
+                placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다. 예: '#1과 #2를 합성해 주세요'",
+                label="프롬프트 (선택 사항)"
             )
             
             # 예시 프롬프트 버튼들
@@ -152,6 +163,9 @@ with gr.Blocks() as demo:
             # 결과 출력
             output_image = gr.Image(label="생성된 이미지")
             output_text = gr.Textbox(label="상태 메시지")
+            
+            # 자동 생성된 프롬프트 표시
+            prompt_display = gr.Textbox(label="사용된 프롬프트", visible=True)
     
     # 예시 프롬프트 버튼 클릭 이벤트
     prompt1_btn.click(lambda: "#1과 #2를 자연스럽게 합성해 주세요", outputs=prompt_input)
@@ -159,10 +173,30 @@ with gr.Blocks() as demo:
     prompt3_btn.click(lambda: "#3의 스타일로 #1을 변환해 주세요", outputs=prompt_input)
     
     # 이미지 생성 버튼 클릭 이벤트
+    def process_and_show_prompt(image1, image2, image3, prompt):
+        # 이미지 개수 확인
+        images = [image1, image2, image3]
+        valid_images = [img for img in images if img is not None]
+        
+        # 자동 프롬프트 생성 (표시용)
+        auto_prompt = prompt
+        if not prompt or not prompt.strip():
+            if len(valid_images) == 1:
+                auto_prompt = "이 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
+            elif len(valid_images) == 2:
+                auto_prompt = "이 두 이미지를 자연스럽게 합성해주세요. 두 이미지의 요소를 조화롭게 통합하여 하나의 이미지로 만들어주세요."
+            else:
+                auto_prompt = "이 세 이미지를 창의적으로 합성해주세요. 모든 이미지의 주요 요소를 포함하되 자연스럽고 일관된 하나의 장면으로 만들어주세요."
+        
+        # 이미지 생성 함수 호출
+        result_img, status = process_images_with_prompt(image1, image2, image3, prompt)
+        
+        return result_img, status, auto_prompt
+    
     submit_btn.click(
-        fn=process_images_with_prompt,
+        fn=process_and_show_prompt,
         inputs=[image1_input, image2_input, image3_input, prompt_input],
-        outputs=[output_image, output_text],
+        outputs=[output_image, output_text, prompt_display],
     )
 
     gr.HTML(
@@ -170,10 +204,10 @@ with gr.Blocks() as demo:
         <div style="margin-top: 1rem; padding: 1rem; background-color: #f8f9fa; border-radius: 0.5rem;">
             <h3>사용 방법:</h3>
             <ul>
-                <li>이미지를 #1, #2, #3 슬롯에 업로드하세요</li>
-                <li>프롬프트에서 #1, #2, #3으로 각 이미지를 참조할 수 있습니다</li>
-                <li>예시 버튼을 클릭하면 자주 사용하는 프롬프트 템플릿을 적용할 수 있습니다</li>
-                <li>모든 이미지를 업로드할 필요는 없으며, 필요한 이미지만 업로드하세요</li>
+                <li><strong>자동 합성:</strong> 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다</li>
+                <li><strong>이미지 참조:</strong> 프롬프트에서 #1, #2, #3으로 각 이미지를 참조할 수 있습니다</li>
+                <li><strong>예시 버튼:</strong> 자주 사용하는 프롬프트 템플릿을 클릭하여 적용할 수 있습니다</li>
+                <li><strong>필요한 이미지만:</strong> 모든 이미지를 업로드할 필요는 없으며, 필요한 이미지만 업로드하세요</li>
             </ul>
         </div>
         """