Update app.py

app.py

@@ -10,9 +10,8 @@ import numpy as np
 from io import BytesIO
 from datetime import datetime, timedelta
 import random
-import requests
-import base64
 from dotenv import load_dotenv
+from gradio_client import Client, handle_file
 
 load_dotenv()
 
@@ -20,316 +19,177 @@ load_dotenv()
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
-# ------------------- API 엔드포인트 설정 -------------------
-def get_api_endpoint():
-    """환경변수에서 API 엔드포인트를 가져오는 함수"""
-    endpoint = os.getenv('API_ENDPOINT')
-    
-    # 환경변수가 없으면 None 반환
-    if not endpoint:
+# 클라이언트 정보 로그에서 숨기기
+logging.getLogger('gradio_client').setLevel(logging.WARNING)
+
+# API 엔드포인트 설정 (환경변수에서만 설정)
+API_ENDPOINT = os.environ.get("API_ENDPOINT", "")
+
+if not API_ENDPOINT:
+    raise ValueError("API_ENDPOINT 환경변수가 설정되지 않았습니다.")
+
+# 클라이언트 초기화
+try:
+    client = Client(API_ENDPOINT)
+    logger.info("클라이언트 초기화 완료")
+except Exception as e:
+    logger.error(f"클라이언트 초기화 실패: {e}")
+    raise
+
+def safe_client_call(api_name, **kwargs):
+    """안전한 클라이언트 호출 함수"""
+    try:
+        return client.predict(api_name=api_name, **kwargs)
+    except Exception as e:
+        logger.error(f"API 호출 실패 ({api_name}): {e}")
         return None
-    
-    # 허깅페이스 스페이스 형식 처리
-    if "/" in endpoint and not endpoint.startswith("http"):
-        # "username/spacename" 형식을 전체 URL로 변환
-        endpoint = f"https://{endpoint.replace('/', '-')}.hf.space"
-    elif not endpoint.startswith("http"):
-        # 단순 도메인인 경우 https 추가
-        endpoint = f"https://{endpoint}"
-    
-    return endpoint
 
-def encode_image_to_base64(image):
-    """PIL 이미지를 base64로 인코딩"""
+def convert_image_to_file(image):
+    """PIL Image 객체를 임시 파일로 변환"""
     if image is None:
         return None
     
-    buffered = BytesIO()
-    image.save(buffered, format="PNG")
-    img_str = base64.b64encode(buffered.getvalue()).decode()
-    return img_str
-
-def decode_base64_to_image(base64_str):
-    """base64 문자열을 PIL 이미지로 디코딩"""
-    if not base64_str:
-        return None
-    
-    image_data = base64.b64decode(base64_str)
-    image = Image.open(BytesIO(image_data))
-    return image
-
-def call_api_endpoint(endpoint_url, data):
-    """API 엔드포인트 호출 함수"""
     try:
-        response = requests.post(endpoint_url, json=data, timeout=300)
-        response.raise_for_status()
-        return response.json()
-    except requests.exceptions.RequestException as e:
-        logger.error(f"API 호출 중 오류 발생: {str(e)}")
-        return {"error": f"API 호출 실패: {str(e)}"}
-
-# ========== 이미지 생성기 관련 함수 ==========
-def get_api_endpoint():
-    """환경변수에서 API 엔드포인트를 가져오는 함수"""
-    endpoint = os.getenv('API_ENDPOINT')
-    
-    # 환경변수가 없으면 기본값 사용 (테스트용)
-    if not endpoint:
-        endpoint = "https://happydoggg-03hd74jd.hf.space"
-        logger.info("환경변수가 설정되지 않아 기본 엔드포인트를 사용합니다.")
-    
-    # 허깅페이스 스페이스 형식 처리
-    if "/" in endpoint and not endpoint.startswith("http"):
-        # "username/spacename" 형식을 전체 URL로 변환
-        endpoint = f"https://{endpoint.replace('/', '-')}.hf.space"
-    elif not endpoint.startswith("http"):
-        # 단순 도메인인 경우 https 추가
-        endpoint = f"https://{endpoint}"
-    
-    return endpoint
+        # PIL Image 객체인 경우
+        if hasattr(image, 'save'):
+            # 임시 파일 생성
+            temp_file = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
+            image.save(temp_file.name, format='PNG')
+            temp_file.close()
+            return temp_file.name
+        # 이미 파일 경로인 경우
+        elif isinstance(image, str):
+            return image
+        # 기타 경우
+        else:
+            return image
+    except Exception as e:
+        logger.error(f"이미지 파일 변환 실패: {e}")
+        return None
 
-def call_api_endpoint(endpoint_url, data):
-    """API 엔드포인트 호출 함수"""
-    try:
-        response = requests.post(endpoint_url, json=data, timeout=300)
-        response.raise_for_status()
-        return response.json()
-    except requests.exceptions.RequestException as e:
-        # 엔드포인트 정보를 숨기고 일반적인 오류 메시지만 표시
-        logger.error("API 호출 중 오류 발생")
-        return {"error": "API 호출 실패"}
-
-def translate_prompt_to_english(prompt):
-    """프롬프트 번역 - API 엔드포인트로 전달"""
-    endpoint_url = get_api_endpoint()
-    if not endpoint_url:
-        return prompt
-    
-    data = {
-        "function": "translate_prompt",
-        "prompt": prompt
-    }
-    
+# ========== 이미지 생성기 관련 함수 (클라이언트 API 활용) ==========
+def generate_single_image(image1, image2, image3, prompt):
+    """단일 이미지 생성 - 원본 함수명 유지"""
     try:
-        result = call_api_endpoint(f"{endpoint_url}/translate", data)
-        if "error" in result:
-            logger.warning("번역 API 오류 발생")
-            return prompt
-        return result.get("translated_prompt", prompt)
-    except Exception as e:
-        logger.warning("번역 중 오류 발생")
-        return prompt
-
-def generate_with_images(prompt, images, variation_index=0):
-    """API 엔드포인트로 이미지 생성 요청"""
-    endpoint_url = get_api_endpoint()
-    if not endpoint_url:
-        return None, "API 엔드포인트가 설정되지 않았습니다."
-    
-    # 이미지들을 base64로 인코딩
-    encoded_images = []
-    for img in images:
-        if img is not None:
-            encoded_img = encode_image_to_base64(img)
-            encoded_images.append(encoded_img)
+        # 이미지 파일 처리 - PIL Image를 임시 파일로 변환
+        image1_path = convert_image_to_file(image1)
+        image2_path = convert_image_to_file(image2)
+        image3_path = convert_image_to_file(image3)
+        
+        # handle_file로 처리
+        image1_file = handle_file(image1_path) if image1_path else None
+        image2_file = handle_file(image2_path) if image2_path else None
+        image3_file = handle_file(image3_path) if image3_path else None
+        
+        result = safe_client_call(
+            "/generate_single_image",
+            image1=image1_file,
+            image2=image2_file,
+            image3=image3_file,
+            prompt=prompt
+        )
+        
+        if result:
+            return result[0], result[1], result[2]
         else:
-            encoded_images.append(None)
-    
-    data = {
-        "function": "generate_with_images",
-        "prompt": prompt,
-        "images": encoded_images,
-        "variation_index": variation_index
-    }
-    
+            return None, "API 호출 실패", ""
+            
+    except Exception as e:
+        logger.error(f"단일 이미지 생성 실패: {e}")
+        return None, f"오류 발생: {str(e)}", ""
+
+def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progress()):
+    """다중 이미지 생성 - 원본 함수명 유지"""
     try:
-        result = call_api_endpoint(f"{endpoint_url}/generate", data)
-        if "error" in result:
-            return None, "이미지 생성 중 오류가 발생했습니다."
+        progress(0, desc="이미지 생성 준비 중...")
+        
+        # 이미지 파일 처리 - PIL Image를 임시 파일로 변환
+        image1_path = convert_image_to_file(image1)
+        image2_path = convert_image_to_file(image2)
+        image3_path = convert_image_to_file(image3)
+        
+        # handle_file로 처리
+        image1_file = handle_file(image1_path) if image1_path else None
+        image2_file = handle_file(image2_path) if image2_path else None
+        image3_file = handle_file(image3_path) if image3_path else None
+        
+        progress(0.5, desc="이미지 생성 중...")
+        
+        result = safe_client_call(
+            "/generate_multiple_images",
+            image1=image1_file,
+            image2=image2_file,
+            image3=image3_file,
+            prompt=prompt
+        )
+        
+        progress(1.0, desc="이미지 생성 완료!")
         
-        # base64 이미지를 PIL 이미지로 변환
-        if "image" in result:
-            generated_image = decode_base64_to_image(result["image"])
-            return generated_image, result.get("status", "이미지가 성공적으로 생성되었습니다.")
+        if result:
+            return result[0], result[1], result[2], result[3]
         else:
-            return None, "API에서 이미지를 생성하지 못했습니다."
+            return None, None, "API 호출 실패", ""
+            
     except Exception as e:
-        logger.error("이미지 생성 API 호출 중 오류")
-        return None, "이미지 생성 중 오류가 발생했습니다."
-    """프롬프트 전처리 - 기존 로직 유지"""
-    has_img1 = image1 is not None
-    has_img2 = image2 is not None
-    has_img3 = image3 is not None
-
-    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 = "첫 번째 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
+        logger.error(f"다중 이미지 생성 실패: {e}")
+        return None, None, f"오류 발생: {str(e)}", ""
 
-    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 = "첫 번째 이미지에서 모든 텍스트를 찾아 자연스럽게 제거해주세요. 깔끔한 이미지로 만들어주세요."
+# ========== 프롬프트 템플릿 함수들 - 클라이언트 API 활용 ==========
+def get_prompt_template_1():
+    """프롬프트 템플릿 1 - 부분변경-1"""
+    return "(#1의 여성)이 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라."
 
-    elif "4. 옷바꾸기" in prompt:
-        prompt = "첫 번째 이미지의 인물 의상을 두 번째 이미지의 의상으로 변경해주세요. 의상의 스타일과 색상은 두 번째 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
+def get_prompt_template_2():
+    """프롬프트 템플릿 2 - 부분변경-2"""
+    return "(#1 레모모형)에서 청색상어레고만 검은색 고래레고로 변경하고 나머지 부분은 seed를 변경을 하지마라."
 
-    elif "5. 배경바꾸기" in prompt:
-        prompt = "첫 번째 이미지의 배경을 두 번째 이미지의 배경으로 변경해주세요. 첫 번째 이미지의 주요 피사체는 유지하고, 두 번째 이미지의 배경과 조화롭게 합성해주세요."
+def get_prompt_template_3():
+    """프롬프트 템플릿 3 - 부분변경-3"""
+    return "(#1 여행용 얼음박스)앞에 얼음이 담긴 3잔의 콜라가 놓여있는 이미지를 생성하라."
 
-    elif "6. 이미지 합성(상품포함)" in prompt:
-        prompt = "첫 번째 이미지와 두 번째 이미지(또는 세 번째 이미지)를 자연스럽게 합성해주세요. 모든 이미지의 주요 요소를 포함하고, 특히 상품이 돋보이도록 조화롭게 통합해주세요."
+def get_prompt_template_4():
+    """프롬프트 템플릿 4 - 글자지우기"""
+    return "(#1 이미지)에 있는 중국어를 모두 제거하라."
 
-    prompt += " 이미지를 생성해주세요. 이미지에 텍스트나 글자를 포함하지 마세요."
-    return prompt
+def get_prompt_template_5():
+    """프롬프트 템플릿 5 - 글자변경"""
+    return '(#1의 텍스트)를 스타일을 유지한체 텍스트만 "Hello"로 바꿔라'
 
-def generate_with_images(prompt, images, variation_index=0):
-    """API 엔드포인트로 이미지 생성 요청"""
-    endpoint_url = get_api_endpoint()
-    if not endpoint_url:
-        return None, "API 엔드포인트가 설정되지 않았습니다."
-    
-    # 이미지들을 base64로 인코딩
-    encoded_images = []
-    for img in images:
-        if img is not None:
-            encoded_img = encode_image_to_base64(img)
-            encoded_images.append(encoded_img)
-        else:
-            encoded_images.append(None)
-    
-    data = {
-        "function": "generate_with_images",
-        "prompt": prompt,
-        "images": encoded_images,
-        "variation_index": variation_index
-    }
-    
-    try:
-        result = call_api_endpoint(f"{endpoint_url}/generate", data)
-        if "error" in result:
-            return None, result["error"]
-        
-        # base64 이미지를 PIL 이미지로 변환
-        if "image" in result:
-            generated_image = decode_base64_to_image(result["image"])
-            return generated_image, result.get("status", "이미지가 성공적으로 생성되었습니다.")
-        else:
-            return None, "API에서 이미지를 생성하지 못했습니다."
-    except Exception as e:
-        logger.error(f"이미지 생성 API 호출 중 오류: {str(e)}")
-        return None, f"이미지 생성 중 오류 발생: {str(e)}"
+def get_prompt_template_6():
+    """프롬프트 템플릿 6 - 상품착용-1"""
+    return "(#1의 여성모델)이 신체 비율과 포즈는 유지한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 모습을 생성하라."
 
-def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0, max_retries=3):
-    """이미지 처리 메인 함수"""
-    retry_count = 0
-    last_error = None
-    
-    while retry_count < max_retries:
-        try:
-            images = [image1, image2, image3]
-            valid_images = [img for img in images if img is not None]
-            
-            if not valid_images:
-                if prompt and prompt.strip():
-                    logger.info("텍스트 프롬프트만으로 이미지 생성 시도")
-                    try:
-                        result_img, status = generate_with_images(prompt, [], variation_index)
-                        if result_img is not None:
-                            return result_img, status, prompt
-                    except:
-                        pass
-                return None, "이미지를 업로드하거나 텍스트 프롬프트를 입력해주세요.", ""           
-
-            if prompt and prompt.strip():
-                processed_prompt = preprocess_prompt(prompt, image1, image2, image3)
-                if re.search("[가-힣]", processed_prompt):
-                    final_prompt = translate_prompt_to_english(processed_prompt)
-                else:
-                    final_prompt = processed_prompt
-            else:
-                if len(valid_images) == 1:
-                    final_prompt = "Please creatively transform this image into a more vivid and artistic version. Do not include any text or watermarks in the generated image."
-                    logger.info("Default prompt generated for single image")
-                elif len(valid_images) == 2:
-                    final_prompt = "Please seamlessly composite these two images, integrating their key elements harmoniously into a single image. Do not include any text or watermarks in the generated image."
-                    logger.info("Default prompt generated for two images")
-                else:
-                    final_prompt = "Please creatively composite these three images, combining their main elements into a cohesive and natural scene. Do not include any text or watermarks in the generated image."
-                    logger.info("Default prompt generated for three images")
-
-            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
-                retry_count += 1
-                logger.warning(f"이미지 생성 실패, 재시도 {retry_count}/{max_retries}")
-                time.sleep(1)
-        except Exception as e:
-            last_error = "처리 중 오류가 발생했습니다."
-            retry_count += 1
-            logger.exception(f"이미지 처리 중 오류 발생, 재시도 {retry_count}/{max_retries}")
-            time.sleep(1)
-    
-    return None, f"최대 재시도 횟수({max_retries}회) 초과 후 실패: {last_error}", prompt
+def get_prompt_template_7():
+    """프롬프트 템플릿 7 - 상품착용-2"""
+    return "(#1의 여성모델)이 (#2의 선글라스)을 착용하고 (#3의 뒷배경의 카페전체가 보이며) 의자에 앉아 있는 모습을 생성하라."
 
-def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progress()):
-    """다중 이미지 생성 함수"""
-    results = []
-    statuses = []
-    prompts = []
-    
-    num_images = 2
-    max_retries = 3
-    
-    progress(0, desc="이미지 생성 준비 중...")
-    
-    for i in range(num_images):
-        progress((i / num_images), desc=f"{i+1}/{num_images} 이미지 생성 중...")
-        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}")
-        else:
-            results.append(None)
-            statuses.append(f"이미지 #{i+1} 생성 실패: {status}")
-            prompts.append(f"이미지 #{i+1}: {final_prompt}")
-        
-        time.sleep(1)
-    
-    progress(1.0, desc="이미지 생성 완료!")
-    
-    while len(results) < 2:
-        results.append(None)
-    
-    combined_status = "\n".join(statuses)
-    combined_prompts = "\n".join(prompts)
-    
-    return results[0], results[1], combined_status, combined_prompts
+def get_prompt_template_8():
+    """프롬프트 템플릿 8 - 상품들고"""
+    return "(#1의 여성모델)이(#2의 와인잔)을 들고 있는 자연스러운 모습을 생성하라."
+
+def get_prompt_template_9():
+    """프롬프트 템플릿 9 - 배경바꾸기"""
+    return "(#1의 여성모델)이 (#2 카페)에서 자연스럽게 있는 모습을 생성하라."
+
+def get_prompt_template_10():
+    """프롬프트 템플릿 10 - 부분지우기"""
+    return "(#1의 레고모형)에서 청색상어레고를 제거한 후, 그 자리를 주변 배경과 자연스럽게 어우���지도록 채워주세요. 단, 이미지의 다른 부분의 주요 요소는 동일하게 유지 해야한다."
+
+def get_prompt_template_11():
+    """프롬프트 템플릿 11 - 이미지확장"""
+    return "(#1 이미지)를 원본그대로 중앙에 두고 비율로 유지한 체 위아래 및 좌우로 크게 확장하라."
+
+def get_prompt_template_12():
+    """프롬프트 템플릿 12 - 플레이팅-1"""
+    return "(#1셀러드)에 담은 용기는 버리고 넓고 흰 예쁜 접시에 (#1셀러드)음식만 가득 채워서 상업적인 각도로 어울리는 소품과 함께 플레이팅 한 모습을 이미지로 생성하라. "
+
+def get_prompt_template_13():
+    """프롬프트 템플릿 13 - 플레이팅-2"""
+    return "(#2 플레이팅한 이미지)에 담긴 음식을  (#1 샐러드)로 바꾸고 나머지는 시드를 유지한 체 이미지를 생성하라."
+
+def get_prompt_template_14():
+    """프롬프트 템플릿 14 - 플레이팅-3"""
+    return "(#1컵)에 딸기, 바닐라, 초코 아이스크림을 담고 그 위에 초코 시럽이 흐르게 이미지를 생성하라."
 
 # 다크모드 자동 적용 커스텀 CSS 스타일
 custom_css = """
@@ -765,7 +625,6 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Default(
 )) as demo:
     gr.HTML(fontawesome_link)
     
-    # 탭 제거하고 이미지 생성기 내용을 직접 배치
     with gr.Row(equal_height=True):
         with gr.Column(scale=1):
             # ======== 이미지 업로드 및 설정 섹션 ========
@@ -854,85 +713,23 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Default(
     )
     
     # ========== 이미지 생성기 이벤트 연결 ==========
-    # 버튼 이벤트 연결
-    image_change_btn1.click(
-        fn=lambda: "(#1의 여성)이 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    image_change_btn2.click(
-        fn=lambda: "(#1 레모모형)에서 청색상어레고만 검은색 고래레고로 변경하고 나머지 부분은 seed를 변경을 하지마라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    image_change_btn3.click(
-        fn=lambda: "(#1 여행용 얼음박스)앞에 얼음이 담긴 3잔의 콜라가 놓여있는 이미지를 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    
-    text_remove_btn.click(
-        fn=lambda: "(#1 이미지)에 있는 중국어를 모두 제거하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    text_change_btn.click(
-        fn=lambda: '(#1의 텍스트)를 스타일을 유지한체 텍스트만 "Hello"로 바꿔라',
-        inputs=[],
-        outputs=prompt_input
-    )
-    clothes_change_btn1.click(
-        fn=lambda: "(#1의 여성모델)이 신체 비율과 포즈는 유지한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 모습을 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    clothes_change_btn2.click(
-        fn=lambda: "(#1의 여성모델)이 (#2의 선글라스)을 착용하고 (#3의 뒷배경의 카페전체가 보이며) 의자에 앉아 있는 모습을 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    holding_product_btn.click(
-        fn=lambda: "(#1의 여성모델)이(#2의 와인잔)을 들고 있는 자연스러운 모습을 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    background_change_btn.click(
-        fn=lambda: "(#1의 여성모델)이 (#2 카페)에서 자연스럽게 있는 모습을 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    composite_product_btn.click(
-        fn=lambda: "(#1의 레고모형)에서 청색상어레고를 제거한 후, 그 자리를 주변 배경과 자연스럽게 어우러지도록 채워주세요. 단, 이미지의 다른 부분의 주요 요소는 동일하게 유지 해야한다.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    outpainting_btn.click(
-        fn=lambda: "(#1 이미지)를 원본그대로 중앙에 두고 비율로 유지한 체 위아래 및 좌우로 크게 확장하라.",
-        inputs=[],
-        outputs=prompt_input
-
-    )
-    food_btn_1.click(
-        fn=lambda: "(#1셀러드)에 담은 용기는 버리고 넓고 흰 예쁜 접시에 (#1셀러드)음식만 가득 채워서 상업적인 각도로 어울리는 소품과 함께 플레이팅 한 모습을 이미지로 생성하라. ",
-        inputs=[],
-        outputs=prompt_input
-    )
-    food_btn_2.click(
-        fn=lambda: "(#2 플레이팅한 이미지)에 담긴 음식을  (#1 샐러드)로 바꾸고 나머지는 시드를 유지한 체 이미지를 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )  
-
-    food_btn_3.click(
-        fn=lambda: "(#1컵)에 딸기, 바닐라, 초코 아이스크림을 담고 그 위에 초코 시럽이 흐르게 이미지를 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )    
+    # 버튼 이벤트 연결 - 프롬프트 템플릿
+    image_change_btn1.click(fn=get_prompt_template_1, outputs=prompt_input)
+    image_change_btn2.click(fn=get_prompt_template_2, outputs=prompt_input)
+    image_change_btn3.click(fn=get_prompt_template_3, outputs=prompt_input)
+    text_remove_btn.click(fn=get_prompt_template_4, outputs=prompt_input)
+    text_change_btn.click(fn=get_prompt_template_5, outputs=prompt_input)
+    clothes_change_btn1.click(fn=get_prompt_template_6, outputs=prompt_input)
+    clothes_change_btn2.click(fn=get_prompt_template_7, outputs=prompt_input)
+    holding_product_btn.click(fn=get_prompt_template_8, outputs=prompt_input)
+    background_change_btn.click(fn=get_prompt_template_9, outputs=prompt_input)
+    composite_product_btn.click(fn=get_prompt_template_10, outputs=prompt_input)
+    outpainting_btn.click(fn=get_prompt_template_11, outputs=prompt_input)
+    food_btn_1.click(fn=get_prompt_template_12, outputs=prompt_input)
+    food_btn_2.click(fn=get_prompt_template_13, outputs=prompt_input)
+    food_btn_3.click(fn=get_prompt_template_14, outputs=prompt_input)
     
     # 단일 이미지 생성 버튼 이벤트 연결
-    def generate_single_image(image1, image2, image3, prompt):
-        return process_images_with_prompt(image1, image2, image3, prompt, 0, 3)
-    
     submit_single_btn.click(
         fn=generate_single_image,
         inputs=[image1_input, image2_input, image3_input, prompt_input],
@@ -947,13 +744,5 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Default(
     )
 
 # 애플리케이션 실행
-if __name__ == "__main__":
-    # API 엔드포인트 확인 (로그에는 민감정보 출력하지 않음)
-    endpoint = get_api_endpoint()
-    if endpoint:
-        logger.info("컨트롤 타워가 시작되었습니다.")
-    else:
-        logger.warning("API 설정을 확인해주세요.")
-    
-    demo.queue()
-    demo.launch(share=False, inbrowser=True, width="100%")
+demo.queue()
+demo.launch(share=False, inbrowser=True, width="100%")