Update app.py

app.py

@@ -13,6 +13,8 @@ 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 +44,140 @@ 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 download_image(image, input_image_name):
+    """이미지를 JPG 형식으로 저장하고 경로 반환"""
+    if image is None:
+        return None
+
+    # 한국 시간 타임스탬프 생성
+    def get_korean_timestamp():
+        korea_time = datetime.utcnow() + timedelta(hours=9)
+        return korea_time.strftime('%Y%m%d_%H%M%S')
+
+    timestamp = get_korean_timestamp()
+    if input_image_name and hasattr(input_image_name, 'name'):
+        base_name = input_image_name.name.split('.')[0]  # 파일 객체에서 이름 추출
+    else:
+        base_name = "이미지"
+
+    file_name = f"[끝장AI]끝장필터_{base_name}_{timestamp}.jpg"
+
+    # 파일 저장
+    temp_file_path = tempfile.gettempdir() + "/" + file_name
+    image.save(temp_file_path, format="JPEG")
+    return temp_file_path
+
+
+
 def translate_prompt_to_english(prompt):
     # 기존 함수 유지
     if not re.search("[가-힣]", prompt):
@@ -906,7 +1042,39 @@ with gr.Blocks(css=custom_css) as demo:
                     </p>
                 </div>
             </div>
-            """)
+    
+    # 사용 설명서 탭의 HTML에 이미지 필터 기능 설명 추가
+    # 기존 <div class="manual-section"> 패턴과 동일하게 작성했습니다
+    
+    # 이 부분을 사용 설명서 HTML의 적절한 위치 (다른 섹션 다음)에 추가
+    """
+    <div class="manual-section">
+        <div class="manual-section-title"><i class="fas fa-sliders-h"></i> 이미지 필터 기능</div>
+        <p class="manual-text">
+            <span style="font-size: 1.1rem; font-weight: 600; color: #5561e9;">이미지 편집 도구</span><br>
+            이미지 필터 탭에서는 다양한 슬라이더를 사용하여 이미지를 쉽게 편집할 수 있습니다.<br>
+            주요 기능:
+            <ul>
+                <li>밝기, 대비, 채도, 색온도 등 다양한 조절 옵션</li>
+                <li>모든 변경사항 실시간 미리보기</li>
+                <li>편집 완료 후 JPG 형식으로 저장 가능</li>
+            </ul>
+            
+            <div class="tip-box">
+                <i class="fas fa-lightbulb"></i> <strong>팁:</strong> 각 슬라이더를 조금씩 조절하며 원하는 효과를 찾아보세요.
+            </div>
+        </p>
+    </div>
+    """
+    
+    # 또한 "주요 기능" 섹션의 그리드에 이미지 필터 항목 추가
+    # 기존 그리드 항목들 다음에 이 항목을 추가하세요
+    """
+    <div>
+        <p style="font-size: 1.1rem; font-weight: 600; color: #ff6b6b;"><i class="fas fa-sliders-h"></i> 이미지 필터</p>
+        밝기, 대비, 채도 등을 조절하여 이미지를 전문적으로 편집합니다.
+    </div>
+    """)
             
         with gr.TabItem("✨ 이커머스 이미지 생성기", elem_classes="tab-content"):
             with gr.Row(equal_height=True):
@@ -1027,6 +1195,86 @@ 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.Row(equal_height=True):
+                # 왼쪽 열: 비율 3
+                with gr.Column(scale=3, elem_classes="panel"):
+                    gr.HTML('<div class="section-title"><i class="fas fa-sliders-h"></i> 이미지 필터 설정</div>')
+                    # 이미지 업로드
+                    filter_input_image = gr.Image(type="pil", label="이미지 업로드", elem_classes="image-container")
+        
+                    # 조정 슬라이더
+                    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="그림자 조절")
+        
+                # 오른쪽 열: 비율 7
+                with gr.Column(scale=7, elem_classes="panel"):
+                    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")
+        
+                    # 변환된 이미지 다운로드 버튼
+                    with gr.Row():
+                        gr.HTML('<div class="section-title"><i class="fas fa-download"></i> 이미지 저장</div>')
+                        download_button = gr.Button("JPG로 변환하기", elem_classes="custom-button primary")
+                        download_output = gr.File(label="JPG 이미지 다운로드")
+        
+            # 이미지 처리 함수 연결
+            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
+            )
+        
+            # 다운로드 버튼 기능
+            download_button.click(
+                fn=download_image,
+                inputs=[filter_output_image, filter_input_image],
+                outputs=download_output
+            )
     
     # 버튼 이벤트 연결
     image_change_btn1.click(