Update app.py

app.py

@@ -20,13 +20,76 @@ def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
 
-def preprocess_prompt(prompt):
+def preprocess_prompt(prompt, image1, image2, image3):
     """
-    프롬프트에 있는 #1, #2, #3 참조를 적절한 텍스트로 변환
+    프롬프트를 처리하고 기능 명령을 해석
     """
-    # #1, #2, #3 참조를 텍스트로 변환
-    processed_prompt = prompt.replace("#1", "첫 번째 이미지").replace("#2", "두 번째 이미지").replace("#3", "세 번째 이미지")
-    return processed_prompt
+    # 이미지 없는 참조 확인 및 처리
+    has_img1 = image1 is not None
+    has_img2 = image2 is not None
+    has_img3 = image3 is not None
+    
+    # #1, #2, #3 참조를 설명으로 변환 (이미지가 없는 경우 무시)
+    if "#1" in prompt and not has_img1:
+        prompt = prompt.replace("#1", "첫 번째 이미지(없음)")
+    else:
+        prompt = prompt.replace("#1", "첫 번째 이미지")
+        
+    if "#2" in prompt and not has_img2:
+        prompt = prompt.replace("#2", "두 번째 이미지(없음)")
+    else:
+        prompt = prompt.replace("#2", "두 번째 이미지")
+        
+    if "#3" in prompt and not has_img3:
+        prompt = prompt.replace("#3", "세 번째 이미지(없음)")
+    else:
+        prompt = prompt.replace("#3", "세 번째 이미지")
+    
+    # 기능 명령 해석
+    if "1. 이미지 변경" in prompt:
+        # 설명 추출을 시도하지만 실패해도 기본 프롬프트 제공
+        desc_match = re.search(r'#1을 "(.*?)"으로 바꿔라', prompt)
+        if desc_match:
+            description = desc_match.group(1)
+            prompt = f"첫 번째 이미지를 {description}으로 변경해주세요. 원본 이미지의 주요 내용은 유지하되 새로운 스타일과 분위기로 재해석해주세요."
+        else:
+            prompt = "첫 번째 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
+    
+    elif "2. 글자지우기" in prompt:
+        # 지울 텍스트 추출을 시도하지만 실패해도 기본 프롬프트 제공
+        text_match = re.search(r'#1에서 "(.*?)"를 지워라', prompt)
+        if text_match:
+            text_to_remove = text_match.group(1)
+            prompt = f"첫 번째 이미지에서 '{text_to_remove}' 텍스트를 찾아 자연스럽게 제거해주세요. 텍스트가 있던 부분을 배경과 조화롭게 채워주세요."
+        else:
+            prompt = "첫 번째 이미지에서 모든 텍스트를 찾아 자연스럽게 제거해주세요. 깔끔한 이미지로 만들어주세요."
+    
+    elif "3. 얼굴바꾸기" in prompt:
+        prompt = "첫 번째 이미지의 인물 얼굴을 두 번째 이미지의 얼굴로 자연스럽게 교체해주세요. 얼굴의 표정과 특징은 두 번째 이미지를 따르되, 나머지 부분은 첫 번째 이미지를 유지해주세요."
+    
+    elif "4. 옷바꾸기" in prompt:
+        # 여러 이미지 참조 처리
+        if "#3" in prompt or "또는 #3" in prompt:
+            prompt = "첫 번째 이미지의 인물 의상을 두 번째 또는 세 번째 이미지의 의상으로 자연스럽게 교체해주세요. 의상의 스타일과 색상은 참조 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
+        else:
+            prompt = "첫 번째 이미지의 인물 의상을 두 번째 이미지의 의상으로 자연스럽게 교체해주세요. 의상의 스타일과 색상은 두 번째 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
+    
+    elif "5. 배경바꾸기" in prompt:
+        prompt = "첫 번째 이미지의 배경을 두 번째 이미지의 배경으로 자연스럽게 교체해주세요. 첫 번째 이미지의 주요 피사체는 유지하고, 두 번째 이미지의 배경과 조화롭게 합성해주세요."
+    
+    elif "6. 이미지 합성(상품포함)" in prompt:
+        # 여러 이미지 참조 처리
+        if "#3" in prompt or "또는 #3" in prompt:
+            prompt = "첫 번째 이미지와 두 번째, 세 번째 이미지를 자연스럽게 합성해주세요. 모든 이미지의 주요 요소를 포함하고, 특히 상품이 잘 보이도록 조화롭게 통합해주세요."
+        else:
+            prompt = "첫 번째 이미지와 두 번째 이미지를 자연스럽게 합성해주세요. 두 이미지의 주요 요소를 포함하고, 특히 상품이 잘 보이도록 조화롭게 통합해주세요."
+    
+    elif "7. 이미지 합성(스타일적용)" in prompt:
+        prompt = "첫 번째 이미지의 내용을 두 번째 이미지의 스타일로 변환해주세요. 첫 번째 이미지의 주요 피사체와 구도는 유지하되, 두 번째 이미지의 예술적 스타일, 색상, 질감을 적용해주세요."
+    
+    # 입력이 위 기능들에 해당하지 않는 경우 원본 프롬프트 유지
+    
+    return prompt
 
 def process_images_with_prompt(image1, image2, image3, prompt):
     """
@@ -61,8 +124,8 @@ def process_images_with_prompt(image1, image2, image3, prompt):
                 prompt = "이 세 이미지를 창의적으로 합성해주세요. 모든 이미지의 주요 요소를 포함하되 자연스럽고 일관된 하나의 장면으로 만들어주세요."
                 logger.info("세 이미지 합성 프롬프트 자동 생성")
         else:
-            # #1, #2, #3 참조 처리
-            prompt = preprocess_prompt(prompt)
+            # 프롬프트 전처리 및 기능 명령 해석
+            prompt = preprocess_prompt(prompt, image1, image2, image3)
         
         # 컨텐츠 리스트 생성 (이미지와 프롬프트 결합)
         parts = []
@@ -130,7 +193,7 @@ with gr.Blocks() as demo:
         """
         <div style="text-align: center; margin-bottom: 1rem;">
             <h1>간단한 이미지 생성기</h1>
-            <p>이미지를 업로드하고 바로 실행하면 자동으로 합성합니다. 또는 #1, #2, #3으로 각 이미지를 참조하는 프롬프트를 입력할 수 있습니다.</p>
+            <p>이미지를 업로드하고 바로 실행하면 자동으로 합성합니다. 또는 기능 명령을 입력할 수 있습니다.</p>
         </div>
         """
     )
@@ -146,15 +209,27 @@ with gr.Blocks() as demo:
             # 프롬프트 입력 (선택 사항)
             prompt_input = gr.Textbox(
                 lines=3,
-                placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다. 예: '#1과 #2를 합성해 주세요'",
+                placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
                 label="프롬프트 (선택 사항)"
             )
             
-            # 예시 프롬프트 버튼들
-            with gr.Row():
-                prompt1_btn = gr.Button("예시: #1 + #2 합성")
-                prompt2_btn = gr.Button("예시: #1의 인물 + #2의 배경")
-                prompt3_btn = gr.Button("예시: #3 스타일로 #1 변환")
+            # 기능 명령 설명
+            gr.HTML(
+                """
+                <div style="margin: 0.5rem 0; padding: 0.5rem; background-color: #f0f0f0; border-radius: 0.5rem;">
+                    <h4>기능 명령 목록:</h4>
+                    <ol style="margin-left: 1.5rem;">
+                        <li>이미지 변경: <code>#1을 "설명해주세요"으로 바꿔라</code></li>
+                        <li>글자지우기: <code>#1에서 "중국어"를 지워라</code></li>
+                        <li>얼굴바꾸기: <code>#1의 인물을 #2의 얼굴로 바꿔라</code></li>
+                        <li>옷바꾸기: <code>#1의 인물에 #2 또는 #3의 옷으로 변경하라</code></li>
+                        <li>배경바꾸기: <code>#1의 이미지에 #2의 배경으로 자연스럽게 바꿔라</code></li>
+                        <li>이미지 합성(상품포함): <code>#1와 #2 또는 #3의 를 합성하라</code></li>
+                        <li>이미지 합성(스타일적용): <code>#1와 #2를 스타일로 변환하라</code></li>
+                    </ol>
+                </div>
+                """
+            )
             
             # 생성 버튼
             submit_btn = gr.Button("이미지 생성", variant="primary")
@@ -164,21 +239,16 @@ 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)
-    prompt2_btn.click(lambda: "#1의 인물을 #2의 배경에 넣어주세요", outputs=prompt_input)
-    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:
@@ -187,6 +257,8 @@ with gr.Blocks() as demo:
                 auto_prompt = "이 두 이미지를 자연스럽게 합성해주세요. 두 이미지의 요소를 조화롭게 통합하여 하나의 이미지로 만들어주세요."
             else:
                 auto_prompt = "이 세 이미지를 창의적으로 합성해주세요. 모든 이미지의 주요 요소를 포함하되 자연스럽고 일관�� 하나의 장면으로 만들어주세요."
+        else:
+            auto_prompt = preprocess_prompt(prompt, image1, image2, image3)
         
         # 이미지 생성 함수 호출
         result_img, status = process_images_with_prompt(image1, image2, image3, prompt)
@@ -205,9 +277,9 @@ with gr.Blocks() as demo:
             <h3>사용 방법:</h3>
             <ul>
                 <li><strong>자동 합성:</strong> 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다</li>
-                <li><strong>이미지 참조:</strong> 프롬프트에서 #1, #2, #3으로 각 이미지를 참조할 수 있습니다</li>
-                <li><strong>예시 버튼:</strong> 자주 사용하는 프롬프트 템플릿을 클릭하여 적용할 수 있습니다</li>
-                <li><strong>필요한 이미지만:</strong> 모든 이미지를 업로드할 필요는 없으며, 필요한 이미지만 업로드하세요</li>
+                <li><strong>기능 명령:</strong> 위 목록의 기능 명령을 복사하여 프롬프트 입력창에 붙여넣기 후 필요에 맞게 수정하세요</li>
+                <li><strong>이미지 참조:</strong> #1, #2, #3으로 각 이미지를 참조할 수 있습니다</li>
+                <li><strong>일부 이미지만:</strong> 필요한 이미지만 업로드해도 기능 실행이 가능합니다</li>
             </ul>
         </div>
         """