Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 import gradio as gr
 import logging
 import re
+import io
 
 from google import genai
 from google.genai import types
@@ -87,6 +88,10 @@ def preprocess_prompt(prompt, image1, image2, image3):
     elif "7. 이미지 합성(스타일적용)" in prompt:
         prompt = "첫 번째 이미지의 내용을 두 번째 이미지의 스타일로 변환해주세요. 첫 번째 이미지의 주요 피사체와 구도는 유지하되, 두 번째 이미지의 예술적 스타일, 색상, 질감을 적용해주세요."
     
+    # 간단한 색상 변경 요청 처리
+    elif "을 붉은색으로 바꿔라" in prompt or "를 붉은색으로 바꿔라" in prompt:
+        prompt = "첫 번째 이미지를 붉은색 톤으로 변경해주세요. 전체적인 색상을 붉은 계열로 조정하고 자연스러운 느낌을 유지해주세요."
+    
     # 입력이 위 기능들에 해당하지 않는 경우 원본 프롬프트 유지
     
     return prompt
@@ -131,27 +136,32 @@ def process_images_with_prompt(image1, image2, image3, prompt):
         parts = []
         
         # 텍스트 프롬프트 추가
-        parts.append(types.Part.from_text(text=prompt))
+        parts.append({
+            "text": prompt
+        })
         
         # 사용 가능한 이미지 추가
         for idx, img in enumerate(images, 1):
             if img is not None:
                 # PIL 이미지를 바이트로 변환
-                with tempfile.NamedTemporaryFile(suffix=".png") as tmp:
-                    img_path = tmp.name
-                    img.save(img_path, format="PNG")
-                    with open(img_path, "rb") as f:
-                        image_bytes = f.read()
+                img_byte_arr = io.BytesIO()
+                img.save(img_byte_arr, format='PNG')
+                img_bytes = img_byte_arr.getvalue()
                 
-                # 이미지를 파트로 추가
-                parts.append(types.Part.from_data(data=image_bytes, mime_type="image/png"))
+                # 이미지를 파트로 추가 (genai.types.Part 대신 직접 딕셔너리 사용)
+                parts.append({
+                    "inline_data": {
+                        "mime_type": "image/png",
+                        "data": img_bytes
+                    }
+                })
                 logger.info(f"이미지 #{idx} 추가됨")
         
         # 생성 설정
-        generate_content_config = types.GenerateContentConfig(
-            temperature=1,
-            response_modalities=["image"],
-        )
+        generate_content_config = {
+            "temperature": 1.0,
+            "response_modalities": ["image"]
+        }
         
         # 임시 파일 생성
         with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
@@ -159,14 +169,21 @@ def process_images_with_prompt(image1, image2, image3, prompt):
             
             # Gemini 모델로 요청 전송
             logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}")
+            
+            # API 요청 형식 수정
             response = client.models.generate_content(
                 model="gemini-2.0-flash-exp-image-generation",
-                contents=[types.Content(role="user", parts=parts)],
-                config=generate_content_config,
+                contents=[{
+                    "role": "user",
+                    "parts": parts
+                }],
+                generation_config=generate_content_config
             )
             
             # 응답에서 이미지 추출
             image_found = False
+            
+            # 응답 구조 처리
             for part in response.candidates[0].content.parts:
                 if hasattr(part, 'inline_data') and part.inline_data:
                     save_binary_file(temp_path, part.inline_data.data)
@@ -187,13 +204,27 @@ def process_images_with_prompt(image1, image2, image3, prompt):
         logger.exception("이미지 생성 중 오류 발생:")
         return None, f"오류 발생: {str(e)}"
 
+# 기능 선택 콜백
+def update_prompt_from_function(function_choice, custom_text=""):
+    function_templates = {
+        "1. 이미지 변경": f'#1을 "{custom_text if custom_text else "원하는 설명"}"으로 바꿔라',
+        "2. 글자지우기": f'#1에서 "{custom_text if custom_text else "지울 텍스트"}"를 지워라',
+        "3. 얼굴바꾸기": "#1의 인물을 #2의 얼굴로 바꿔라",
+        "4. 옷바꾸기": "#1의 인물에 #2 또는 #3의 옷으로 변경하라",
+        "5. 배경바꾸기": "#1의 이미지에 #2의 배경으로 자연스럽게 바꿔라",
+        "6. 이미지 합성(상품포함)": "#1와 #2 또는 #3의 를 합성하라",
+        "7. 이미지 합성(스타일적용)": "#1와 #2를 스타일로 변환하라"
+    }
+    
+    return function_templates.get(function_choice, "")
+
 # Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 1rem;">
             <h1>간단한 이미지 생성기</h1>
-            <p>이미지를 업로드하고 바로 실행하면 자동으로 합성합니다. 또는 기능 명령을 입력할 수 있습니다.</p>
+            <p>이미지를 업로드하고 바로 실행하면 자동으로 합성합니다. 또는 아래 기능을 선택할 수 있습니다.</p>
         </div>
         """
     )
@@ -206,6 +237,28 @@ 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")
             
+            # 기능 선택 드롭다운과 커스텀 텍스트 입력
+            with gr.Row():
+                function_dropdown = gr.Dropdown(
+                    choices=[
+                        "1. 이미지 변경", 
+                        "2. 글자지우기", 
+                        "3. 얼굴바꾸기", 
+                        "4. 옷바꾸기",
+                        "5. 배경바꾸기", 
+                        "6. 이미지 합성(상품포함)", 
+                        "7. 이미지 합성(스타일적용)"
+                    ],
+                    label="기능 선택",
+                    value=None
+                )
+                custom_text_input = gr.Textbox(
+                    label="커스텀 텍스트 (1, 2번 기능용)", 
+                    placeholder="예: 붉은색, 수채화 스타일, 중국어..."
+                )
+            
+            apply_function_btn = gr.Button("기능 적용")
+            
             # 프롬프트 입력 (선택 사항)
             prompt_input = gr.Textbox(
                 lines=3,
@@ -213,24 +266,6 @@ with gr.Blocks() as demo:
                 label="프롬프트 (선택 사항)"
             )
             
-            # 기능 명령 설명
-            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")
         
@@ -242,28 +277,39 @@ with gr.Blocks() as demo:
             # 사용된 프롬프트 표시
             prompt_display = gr.Textbox(label="사용된 프롬프트", visible=True)
     
+    # 기능 적용 버튼 이벤트
+    apply_function_btn.click(
+        fn=update_prompt_from_function,
+        inputs=[function_dropdown, custom_text_input],
+        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 = "이 두 이미지를 자연스럽게 합성해주세요. 두 이미지의 요소를 조화롭게 통합하여 하나의 이미지로 만들어주세요."
+        try:
+            # 자동 프롬프트 생성 또는 프롬프트 전처리
+            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 = "이 세 이미지를 창의적으로 합성해주세요. 모든 이미지의 주요 ��소를 포함하되 자연스럽고 일관된 하나의 장면으로 만들어주세요."
             else:
-                auto_prompt = "이 세 이미지를 창의적으로 합성해주세요. 모든 이미지의 주요 요소를 포함하되 자연스럽고 일관된 하나의 장면으로 만들어주세요."
-        else:
-            auto_prompt = preprocess_prompt(prompt, image1, image2, image3)
-        
-        # 이미지 생성 함수 호출
-        result_img, status = process_images_with_prompt(image1, image2, image3, prompt)
-        
-        return result_img, status, auto_prompt
+                auto_prompt = preprocess_prompt(prompt, image1, image2, image3)
+            
+            # 이미지 생성 함수 호출
+            result_img, status = process_images_with_prompt(image1, image2, image3, prompt)
+            
+            return result_img, status, auto_prompt
+        except Exception as e:
+            logger.exception("처리 중 오류 발생:")
+            return None, f"오류 발생: {str(e)}", prompt
     
     submit_btn.click(
         fn=process_and_show_prompt,
@@ -271,17 +317,17 @@ with gr.Blocks() as demo:
         outputs=[output_image, output_text, prompt_display],
     )
 
-    gr.HTML(
+    gr.Markdown(
         """
-        <div style="margin-top: 1rem; padding: 1rem; background-color: #f8f9fa; border-radius: 0.5rem;">
-            <h3>사용 방법:</h3>
-            <ul>
-                <li><strong>자동 합성:</strong> 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다</li>
-                <li><strong>기능 명령:</strong> 위 목록의 기능 명령을 복사하여 프롬프트 입력창에 붙여넣기 후 필요에 맞게 수정하세요</li>
-                <li><strong>이미지 참조:</strong> #1, #2, #3으로 각 이미지를 참조할 수 있습니다</li>
-                <li><strong>일부 이미지만:</strong> 필요한 이미지만 업로드해도 기능 실행이 가능합니다</li>
-            </ul>
-        </div>
+        ### 사용 방법:
+        
+        1. **자동 합성**: 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다
+        2. **기능 사용**: 드롭다운에서 원하는 기능을 선택하고 '기능 적용' 버튼을 클릭하세요
+        3. **커스텀 텍스트**: 이미지 변경이나 글자지우기 기능을 선택할 때 추가 설명을 입력할 수 있습니다
+        4. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다
+        5. **일부 이미지만**: 필요한 이미지만 업로드해도 기능 실행이 가능합니다
+        
+        > **팁**: 기능을 선택한 후에도 프롬프트를 직접 수정할 수 있습니다
         """
     )