Update app.py

app.py

@@ -13,6 +13,7 @@ from transformers import pipeline  # 새로 추가된 import
 
 load_dotenv()
 
+
 # 기존 코드 유지 (로깅, 함수 등 모든 기능 코드)
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
@@ -42,6 +43,116 @@ def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
 
+
+# 이미지 필터 기능을 위한 추가 코드
+import cv2
+import numpy as np
+from PIL import Image, ImageEnhance, ImageFilter
+import tempfile
+from datetime import datetime, timedelta
+
+# 이미지 필터 처리 함수
+def adjust_brightness(image, value):
+    """이미지 밝기 조절"""
+    value = float(value - 1) * 100  # 0-2 범위를 -100에서 +100으로 변환
+    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
+    h, s, v = cv2.split(hsv)
+    v = cv2.add(v, value)
+    v = np.clip(v, 0, 255)
+    final_hsv = cv2.merge((h, s, v))
+    return cv2.cvtColor(final_hsv, cv2.COLOR_HSV2BGR)
+
+def adjust_contrast(image, value):
+    """이미지 대비 조절"""
+    value = float(value)
+    return np.clip(image * value, 0, 255).astype(np.uint8)
+
+def adjust_saturation(image, value):
+    """이미지 채도 조절"""
+    value = float(value - 1) * 100  # 0-2 범위를 -100에서 +100으로 변환
+    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
+    h, s, v = cv2.split(hsv)
+    s = cv2.add(s, value)
+    s = np.clip(s, 0, 255)
+    final_hsv = cv2.merge((h, s, v))
+    return cv2.cvtColor(final_hsv, cv2.COLOR_HSV2BGR)
+
+def adjust_temperature(image, value):
+    """이미지 색온도 조절 (색상 밸런스)"""
+    value = float(value) * 30  # 효과 스케일 조절
+    b, g, r = cv2.split(image)
+    if value > 0:  # 따뜻하게
+        r = cv2.add(r, value)
+        b = cv2.subtract(b, value)
+    else:  # 차갑게
+        r = cv2.add(r, value)
+        b = cv2.subtract(b, value)
+    
+    r = np.clip(r, 0, 255)
+    b = np.clip(b, 0, 255)
+    return cv2.merge([b, g, r])
+
+def adjust_tint(image, value):
+    """이미지 색조 조절"""
+    hsv_image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
+    h, s, v = cv2.split(hsv_image)
+    h = cv2.add(h, int(value))
+    h = np.clip(h, 0, 179)  # Hue 값은 0-179 범위
+    final_hsv = cv2.merge((h, s, v))
+    return cv2.cvtColor(final_hsv, cv2.COLOR_HSV2BGR)
+
+def adjust_exposure(image, value):
+    """이미지 노출 조절"""
+    enhancer = ImageEnhance.Brightness(Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB)))
+    img_enhanced = enhancer.enhance(1 + float(value) / 5.0)
+    return cv2.cvtColor(np.array(img_enhanced), cv2.COLOR_RGB2BGR)
+
+def adjust_vibrance(image, value):
+    """이미지 활기 조절"""
+    img = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
+    converter = ImageEnhance.Color(img)
+    factor = 1 + (float(value) / 100.0)
+    img = converter.enhance(factor)
+    return cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
+
+def adjust_color_mixer_blues(image, value):
+    """이미지 컬러 믹서 (블루) 조절"""
+    b, g, r = cv2.split(image)
+    b = cv2.add(b, float(value))
+    b = np.clip(b, 0, 255)
+    return cv2.merge([b, g, r])
+
+def adjust_shadows(image, value):
+    """이미지 그림자 조절"""
+    pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
+    enhancer = ImageEnhance.Brightness(pil_image)
+    factor = 1 + (float(value) / 100.0)
+    pil_image = enhancer.enhance(factor)
+    return cv2.cvtColor(np.array(pil_image), cv2.COLOR_BGR2RGB)
+
+def process_image(image, brightness, contrast, saturation, temperature, tint, exposure, vibrance, color_mixer_blues, shadows):
+    """모든 조정 사항을 이미지에 적용"""
+    if image is None:
+        return None
+        
+    # PIL 이미지를 OpenCV 형식으로 변환
+    image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
+    
+    # 조정 사항 순차 적용
+    image = adjust_brightness(image, brightness)
+    image = adjust_contrast(image, contrast)
+    image = adjust_saturation(image, saturation)
+    image = adjust_temperature(image, temperature)
+    image = adjust_tint(image, tint)
+    image = adjust_exposure(image, exposure)
+    image = adjust_vibrance(image, vibrance)
+    image = adjust_color_mixer_blues(image, color_mixer_blues)
+    image = adjust_shadows(image, shadows)
+    
+    # PIL 이미지로 다시 변환
+    return Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
+    
+
 def translate_prompt_to_english(prompt):
     # 기존 함수 유지
     if not re.search("[가-힣]", prompt):
@@ -196,7 +307,8 @@ def generate_with_images(prompt, images, variation_index=0):
             )
         )
 
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
+        # 임시 파일은 항상 JPG 확장자로 생성
+        with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp:
             temp_path = tmp.name
             result_text = ""
             image_found = False
@@ -210,16 +322,22 @@ def generate_with_images(prompt, images, variation_index=0):
                     logger.info("응답에서 이미지 추출 성공")
             if not image_found:
                 return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"
+            
             result_img = Image.open(temp_path)
             if result_img.mode == "RGBA":
-                result_img = result_img.convert("RGB")
-            return result_img, f"이미지가 성공적으로 생성되었습니다. {result_text}"
+                result_img = result_img.convert("RGB")  # JPG는 투명도를 지원하지 않으므로 RGB로 변환
+            
+            # 변환된 이미지를 JPG로 저장
+            result_img.save(temp_path, format="JPEG", quality=95)
+            
+            # 파일 경로 반환
+            return temp_path, f"이미지가 성공적으로 생성되었습니다. {result_text}"
     except Exception as e:
         logger.exception("이미지 생성 중 오류 발생:")
         return None, f"오류 발생: {str(e)}"
 
 def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0, max_retries=3):
-    # 기존 함수 유지
+    # 기존 함수 내용 유지
     retry_count = 0
     last_error = None
     
@@ -249,6 +367,7 @@ def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0
 
             result_img, status = generate_with_images(final_prompt, valid_images, variation_index)
             if result_img is not None:
+                # 이미 파일 경로가 반환되므로 추가 처리 없이 그대로 반환
                 return result_img, status, final_prompt
             else:
                 last_error = status
@@ -264,7 +383,7 @@ def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0
     return None, f"최대 재시도 횟수({max_retries}회) 초과 후 실패: {last_error}", prompt
 
 def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progress()):
-    # 기존 함수 유지
+    # 결과 이미지들이 JPG로 변환되도록 수정
     results = []
     statuses = []
     prompts = []
@@ -279,6 +398,7 @@ def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progres
         result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, 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}")
@@ -299,6 +419,7 @@ def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progres
     
     return results[0], results[1], results[2], results[3], combined_status, combined_prompts
 
+    
 # GFPGAN 관련 코드 유지
 import sys
 from torchvision.transforms import functional
@@ -842,6 +963,7 @@ with gr.Blocks(css=custom_css) as demo:
                                 <p style="font-size: 1.1rem; font-weight: 600; color: #ff6b6b;"><i class="fas fa-expand-alt"></i> 이미지 확장</p>
                                 원본 이미지 주변에 자연스럽게 추가 영역을 생성하여 이미지를 확장합니다.
                             </div>
+                            
                         </div>
                     </div>
                 </div>
@@ -882,22 +1004,55 @@ with gr.Blocks(css=custom_css) as demo:
                     </p>
                 </div>
                 
+
                 <div class="manual-section">
-                    <div class="manual-section-title"><i class="fas fa-question-circle"></i> 자주 묻는 질문</div>
-                    <p class="manual-text">
-                        <strong style="font-size: 1.1rem; color: #5561e9;">Q: 어떤 형식의 이미지를 업로드할 수 있나요?</strong><br>
-                        A: JPG, PNG, WEBP 등 대부분의 일반적인 이미지 형식을 지원합니다.<br><br>
-                        
-                        <strong style="font-size: 1.1rem; color: #5561e9;">Q: 이미지 생성 및 처리에 얼마나 시간이 걸리나요?</strong><br>
-                        A: 일반적으로 10-30초 정도 소요되며, 이미지 크기와 복잡도에 따라 달라질 수 있습니다.<br><br>
-                        
-                        <strong style="font-size: 1.1rem; color: #5561e9;">Q: 생성된 이미지를 어떻게 저장할 수 있나요?</strong><br>
-                        A: 생성된 이미지에 마우스 오른쪽 버튼을 클릭하여 '이미지 저장' 옵션을 선택하면 됩니다.
-                    </p>
+                    <div class="manual-section-title"><i class="fas fa-tools"></i> 주요 기능</div>
+                    <div class="manual-text">
+                        <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(250px, 1fr)); gap: 1rem;">
+                            <div>
+                                <p style="font-size: 1.1rem; font-weight: 600; color: #ff6b6b;"><i class="fas fa-exchange-alt"></i> 이미지 변경</p>
+                                포즈나 스타일을 변경하면서 원본의 주요 요소는 유지합니다.
+                            </div>
+                            
+                            <div>
+                                <p style="font-size: 1.1rem; font-weight: 600; color: #ff6b6b;"><i class="fas fa-eraser"></i> 글자 지우기/변경</p>
+                                이미지의 텍스트를 자연스럽게 제거하거나 다른 텍스트로 변경합니다.
+                            </div>
+                            
+                            <div>
+                                <p style="font-size: 1.1rem; font-weight: 600; color: #ff6b6b;"><i class="fas fa-tshirt"></i> 가상 상품착용</p>
+                                모델에게 다른 이미지의 상품을 자연스럽게 착용시킵니다.
+                            </div>
+                            
+                            <div>
+                                <p style="font-size: 1.1rem; font-weight: 600; color: #ff6b6b;"><i class="fas fa-image"></i> 배경 바꾸기</p>
+                                주요 피사체는 유지하며 배경만 교체합니다.
+                            </div>
+                            
+                            <div>
+                                <p style="font-size: 1.1rem; font-weight: 600; color: #ff6b6b;"><i class="fas fa-cut"></i> 부분 지우기</p>
+                                이미지의 특정 부분을 지우고 배경과 자연스럽게 조화되도록 채웁니다.
+                            </div>
+                            
+                            <div>
+                                <p style="font-size: 1.1rem; font-weight: 600; color: #ff6b6b;"><i class="fas fa-wine-glass"></i> 상품들고 있기</p>
+                                모델이 상품을 들고 있는 자연스러운 포즈의 이미지를 생성합니다.
+                            </div>
+                            
+                            <div>
+                                <p style="font-size: 1.1rem; font-weight: 600; color: #ff6b6b;"><i class="fas fa-expand-alt"></i> 이미지 확장</p>
+                                원본 이미지 주변에 자연스럽게 추가 영역을 생성하여 이미지를 확장합니다.
+                            </div>
+                            
+                            <div>
+                                <p style="font-size: 1.1rem; font-weight: 600; color: #ff6b6b;"><i class="fas fa-utensils"></i> 음식 플레이팅</p>
+                                음식을 특정 용기에 담아 상업적인 각도로 아름답게 플레이팅합니다.
+                            </div>
+                        </div>
+                    </div>
                 </div>
-            </div>
             """)
-            
+        
         with gr.TabItem("✨ 이커머스 이미지 생성기", elem_classes="tab-content"):
             with gr.Row(equal_height=True):
                 with gr.Column(scale=1):
@@ -928,6 +1083,8 @@ with gr.Blocks(css=custom_css) as demo:
                             background_change_btn = gr.Button('🖼️ 배경 바꾸기', elem_classes="custom-button")
                             composite_product_btn = gr.Button('✂️ 부분 지우기', elem_classes="custom-button")
                             outpainting_btn = gr.Button('🔍 이미지 확장', elem_classes="custom-button")
+                            food_plating_btn = gr.Button('🍽️ 음식플레이팅', elem_classes="custom-button")  # 새로 추가된 버튼
+
                         
                         submit_single_btn = gr.Button('✨ 이미지 생성 (1장)', elem_classes="custom-button primary")
                         submit_btn = gr.Button('✨ 이미지 생성 (4장)', elem_classes="custom-button primary")
@@ -937,11 +1094,11 @@ with gr.Blocks(css=custom_css) as demo:
                         gr.HTML('<div class="section-title"><i class="fas fa-images"></i> 생성된 이미지</div>')
                         with gr.Row():
                             with gr.Column():
-                                output_image1 = gr.Image(label="이미지 #1", elem_classes="image-container", height=400)
-                                output_image3 = gr.Image(label="이미지 #3", elem_classes="image-container", height=400)
+                                output_image1 = gr.Image(label="이미지 #1", elem_classes="image-container", height=400, type="filepath")
+                                output_image3 = gr.Image(label="이미지 #3", elem_classes="image-container", height=400, type="filepath")
                             with gr.Column():
-                                output_image2 = gr.Image(label="이미지 #2", elem_classes="image-container", height=400)
-                                output_image4 = gr.Image(label="이미지 #4", elem_classes="image-container", height=400)
+                                output_image2 = gr.Image(label="이미지 #2", elem_classes="image-container", height=400, type="filepath")
+                                output_image4 = gr.Image(label="이미지 #4", elem_classes="image-container", height=400, type="filepath")
                         
                         gr.HTML('<div class="section-title"><i class="fas fa-info-circle"></i> 결과 정보</div>')
                         output_text = gr.Textbox(label="상태 메시지", lines=2, elem_classes="custom-input")
@@ -959,14 +1116,15 @@ with gr.Blocks(css=custom_css) as demo:
                     ["down/모델.jpg", "down/와인잔.png", None, "(#1의 여성모델)이(#2의 와인잔)을 들고 있는 자연스러운 모습"],
                     ["down/모델.jpg", "down/카페전경.png", None, "(#1의 여성모델)이 (#2 카페)에서 자연스럽게 있는 모습"],
                     ["down/상어레고모형.png", None, None, "(#1의 레고모형)에서 청색상어레고를 제거한 후, 그 자리를 주변 배경과 자연스럽게 어우러지도록 채워주세요. 단, 이미지의 다른 부분의 주요 요소는 동일하게 유지해 해야한다."],
-                    ["down/카페전경.png", None, None, "(#1 이미지)를 원본그대로 중앙에 두고 비율로 유지한 체 위아래 및 좌우로 크게 확장하라."]
+                    ["down/카페전경.png", None, None, "(#1 이미지)를 원본그대로 중앙에 두고 비율로 유지한 체 위아래 및 좌우로 크게 확장하라."],
+                    ["down/샐러드.png", "down/용기.png", None, "(#1의 샐러드)가 음식만으로 반드시(#2의 동일한 용기)에 꽉 차도록 상업적인 각도로 어울리는 소품과 함께 플레이팅한 모습"]
                 ]
                 
                 gr.Examples(
                     examples=examples,
                     inputs=[image1_input, image2_input, image3_input, prompt_input]
                 )
-            
+        
         with gr.TabItem("🔍 이미지 업스케일러", elem_classes="tab-content"):
             with gr.Row():
                 with gr.Column(elem_classes="panel"):
@@ -1017,6 +1175,82 @@ with gr.Blocks(css=custom_css) as demo:
                     with gr.Row():
                         bg_remove_status = gr.Textbox(label="상태", lines=2, elem_classes="custom-input")
                         bg_remove_btn = gr.Button('✂️ 배경 제거하기', elem_classes="custom-button primary")
+        
+        # 이미지 필터 탭 UI 개선
+        with gr.TabItem("🖌️ 이미지 필터", elem_classes="tab-content"):
+            with gr.Column(elem_classes="panel"):
+                gr.HTML('<div class="section-title"><i class="fas fa-sliders-h"></i> 이미지 필터</div>')
+                gr.HTML("""
+                <p style="margin-bottom: 1rem; font-size: 1.05rem;">
+                     이미지 필터를 적용하여 색상과 톤을 조절합니다.
+                </p>
+                <div class="tip-box">
+                    <i class="fas fa-lightbulb"></i> <strong>팁:</strong> 슬라이더를 조절하여 이미지의 밝기, 대비, 채도 등을 변경할 수 있습니다.
+                </div>
+                """)
+                
+                # 입력/출력 이미지 섹션
+                with gr.Row():
+                    with gr.Column():
+                        gr.HTML('<div class="section-title"><i class="fas fa-upload"></i> 입력 이미지</div>')
+                        filter_input_image = gr.Image(type="pil", label="업로드", elem_classes="image-container")
+                    
+                    with gr.Column():
+                        gr.HTML('<div class="section-title"><i class="fas fa-image"></i> 결과 이미지</div>')
+                        filter_output_image = gr.Image(type="pil", label="필터 적용 결과", elem_classes="image-container")
+                
+                # 필터 조절 슬라이더들
+                gr.HTML('<div class="section-title"><i class="fas fa-sliders-h"></i> 필터 조절</div>')
+                with gr.Group():
+                    brightness_slider = gr.Slider(0.0, 2.0, value=1.0, step=0.1, label="밝기 조절")
+                    contrast_slider = gr.Slider(0.5, 1.5, value=1.0, step=0.1, label="대비 조절")
+                    saturation_slider = gr.Slider(0.0, 2.0, value=1.0, step=0.1, label="채도 조절")
+                    temperature_slider = gr.Slider(-1.0, 1.0, value=0.0, step=0.1, label="색온도 조절")
+                    tint_slider = gr.Slider(-100, 100, value=0, step=1, label="색조 조절")
+                    exposure_slider = gr.Slider(-5.0, 5.0, value=0.0, step=0.1, label="노출 조절")
+                    vibrance_slider = gr.Slider(-100.0, 100.0, value=0.0, step=1.0, label="활기 조절")
+                    color_mixer_blues_slider = gr.Slider(-100.0, 100.0, value=0.0, step=1.0, label="컬러 믹서 (블루)")
+                    shadows_slider = gr.Slider(-100.0, 100.0, value=0.0, step=1.0, label="그림자 조절")
+
+            # 이미지 처리 함수 연결
+            inputs = [
+                filter_input_image,
+                brightness_slider,
+                contrast_slider,
+                saturation_slider,
+                temperature_slider,
+                tint_slider,
+                exposure_slider,
+                vibrance_slider,
+                color_mixer_blues_slider,
+                shadows_slider
+            ]
+            
+            input_components = [
+                brightness_slider,
+                contrast_slider,
+                saturation_slider,
+                temperature_slider,
+                tint_slider,
+                exposure_slider,
+                vibrance_slider,
+                color_mixer_blues_slider,
+                shadows_slider
+            ]
+            
+            for input_component in input_components:
+                input_component.change(
+                    fn=process_image,
+                    inputs=inputs,
+                    outputs=filter_output_image
+                )
+            
+            filter_input_image.change(
+                fn=lambda x, *args: process_image(x, *args) if x is not None else None,
+                inputs=inputs,
+                outputs=filter_output_image
+            )
+    
     
     # 버튼 이벤트 연결
     image_change_btn1.click(
@@ -1070,6 +1304,12 @@ with gr.Blocks(css=custom_css) as demo:
         inputs=[],
         outputs=prompt_input
     )
+
+    food_plating_btn.click(
+        fn=lambda: "(#1의 샐러드)가 음식만으로 반드시(#2의 동일한 용기)에 꽉 차도록 상업적인 각도로 어울리는 소품과 함께 플레이팅한 모습",
+        inputs=[],
+        outputs=prompt_input
+    )
     
     # 단일 이미지 생성 버튼 이벤트 연결
     submit_single_btn.click(