Update app.py

app.py

@@ -1,507 +1,2 @@
-import gradio as gr
-import openai
-import json
 import os
-import random
-import re
-from PIL import Image
-import io
-
-# OpenAI API 클라이언트 설정
-openai.api_key = os.getenv("OPENAI_API_KEY")
-
-# JSON 데이터 로드 및 카테고리 구성
-try:
-    with open("back.json", "r", encoding="utf-8") as f:
-        data = json.load(f)
-    
-    # 카테고리 데이터 구조화
-    categories = {}
-    for item in data:
-        category = item.get("카테고리", "기타")
-        korean = item.get("Korean Translation", "")
-        english = item.get("English Prompt", "")
-        
-        if category not in categories:
-            categories[category] = []
-        
-        categories[category].append({
-            "korean": korean,
-            "english": english
-        })
-except Exception as e:
-    print(f"Error loading categories: {e}")
-    categories = {}
-
-# OpenAI API 호출 함수 (기존 코드와 동일)
-def call_api(content, system_message, max_tokens, temperature, top_p):
-    response = openai.ChatCompletion.create(
-        model="gpt-4o-mini",
-        messages=[
-            {"role": "system", "content": system_message},
-            {"role": "user", "content": content},
-        ],
-        max_tokens=max_tokens,
-        temperature=temperature,
-        top_p=top_p,
-    )
-    return response.choices[0].message['content'].strip()
-
-# 번역 함수들 (기존 코드와 동일)
-def translate_object(object_input):
-    system_message = "Translate the following object description from Korean to English. Keep it simple and concise."
-    return call_api(
-        content=object_input,
-        system_message=system_message,
-        max_tokens=10,
-        temperature=0.7,
-        top_p=1.0
-    )
-
-def translate_background(background_input):
-    system_message = "Translate the following background description from Korean to English. Keep it simple and concise."
-    return call_api(
-        content=background_input,
-        system_message=system_message,
-        max_tokens=20,
-        temperature=0.7,
-        top_p=1.0
-    )
-
-# 직접 배경 입력용 간단한 프롬프트 생성
-def generate_simple_prompt(object_input, background_english):
-    system_message = (
-        "Your task is to create a concise and grammatically correct background description for advertisement photography. "
-        "Focus on objectively describing the visible aspects of the background without including any references to light, lighting, illumination, or how the scene is lit. "
-        "Do not mention windows in relation to light or any description of how light interacts with the scene. "
-        "Avoid any subjective or emotional language. "
-        "Do not include any feelings, moods, or abstract concepts. "
-        "Do not include 'advertisement photography' or the object name in your response. "
-        "Just output the background description that details the physical surroundings."
-    )
-    content = f"Object: {object_input}, Background hint: {background_english}"
-    response = call_api(content, system_message, max_tokens=100, temperature=0.7, top_p=0.9)
-
-    # 금지된 단어 및 구문 목록
-    forbidden_phrases = [
-        'bright', 'sunny', 'glowing', 'illuminated', 'light', 'lit', 'shining', 'radiant',
-        'sunlight', 'beam', 'luminous', 'gleaming', 'sparkling', 'dazzling', 'soft light',
-        'natural light', 'light streams', 'illuminate', 'illumination', 'lighting',
-        'sunny', 'sun-filled', 'well-lit', 'sun-drenched', 'glow', 'glows', 'sun shines',
-        'sun is shining', 'sunny day', 'sunny weather', 'light enters', 'light filters',
-        'light pours', 'window light', 'sun beams', 'sun rays', 'brightly lit', 'light-filled',
-        'softly lit', 'warm light', 'ambient light', 'backlit', 'highlight', 'highlights',
-        'shadow', 'shadows', 'reflections', 'reflects', 'glistening'
-    ]
-
-    # 문장 단위로 분할하여 금지된 단어가 포함된 문장 제거
-    import re
-    sentences = re.split(r'(?<=[.!?]) +', response)
-    cleaned_sentences = []
-    for sentence in sentences:
-        lower_sentence = sentence.lower()
-        if not any(phrase in lower_sentence for phrase in forbidden_phrases):
-            cleaned_sentences.append(sentence)
-    cleaned_response = ' '.join(cleaned_sentences)
-
-    # 공백 정리
-    cleaned_response = ' '.join(cleaned_response.split())
-
-    # 최종 프롬프트 조합 (직접 입력용 간단 버전)
-    # 첫 번째 문장만 사용
-    first_sentence = cleaned_response.split('.')[0].strip()
-    result = f"advertisement photography, {object_input}, {first_sentence}"
-    
-    # 중복 콤마 제거 및 공백 정리
-    result = re.sub(r',\s*,', ',', result)
-    result = ', '.join(part.strip() for part in result.split(','))
-
-    return result
-
-# 카테고리 선택용 상세 프롬프트 생성
-def generate_detailed_prompt(object_input, background_english):
-    system_message = (
-        "Create a detailed background description for advertisement photography. "
-        "Focus on objectively describing the visible aspects of the background. "
-        "Include specific details about materials, textures, and arrangements. "
-        "Avoid any references to light, lighting, or illumination."
-    )
-    content = f"Object: {object_input}, Background hint: {background_english}"
-    response = call_api(content, system_message, max_tokens=100, temperature=0.7, top_p=0.9)
-
-    # 금지된 단어 및 구문 필터링
-    forbidden_phrases = [
-        'bright', 'sunny', 'glowing', 'illuminated', 'light', 'lit', 'shining', 'radiant',
-        'sunlight', 'beam', 'luminous', 'gleaming', 'sparkling', 'dazzling', 'soft light',
-        'natural light', 'light streams', 'illuminate', 'illumination', 'lighting',
-        'sunny', 'sun-filled', 'well-lit', 'sun-drenched', 'glow', 'glows', 'sun shines',
-        'sun is shining', 'sunny day', 'sunny weather', 'light enters', 'light filters',
-        'light pours', 'window light', 'sun beams', 'sun rays', 'brightly lit', 'light-filled',
-        'softly lit', 'warm light', 'ambient light', 'backlit', 'highlight', 'highlights',
-        'shadow', 'shadows', 'reflections', 'reflects', 'glistening'
-    ]
-    
-    for phrase in forbidden_phrases:
-        response = re.sub(rf'\b{phrase}\b', '', response, flags=re.IGNORECASE)
-    
-    # 최종 프롬프트 조합
-    result = f"advertisement photography, {object_input}, {response}".strip()
-    return result
-
-# 랜덤 배경 생성용 창의적 프롬프트 생성
-def generate_creative_prompt(object_input, background_english):
-    system_message = (
-        "Create a unique and creative background description for advertisement photography. "
-        "Focus on creating an interesting and unexpected setting. "
-        "Avoid any references to light, lighting, or illumination. "
-        "Be specific but avoid overly complex descriptions."
-    )
-    content = f"Object: {object_input}, Generate a creative background setting"
-    response = call_api(content, system_message, max_tokens=75, temperature=0.9, top_p=0.9)
-    
-    # 응답 정제
-    response = re.sub(r'The background (features|shows|consists of|includes|displays|contains) ', '', response, flags=re.IGNORECASE)
-    response = re.sub(r'There (is|are) ', '', response, flags=re.IGNORECASE)
-    
-    # 최종 프롬프트 조합
-    result = f"advertisement photography, {object_input}, {response}".strip()
-    return result
-
-# 직접 배경 입력 탭의 프롬프트 생성 함수
-def generate_prompt(object_input, background_input):
-    translated_object = translate_object(object_input)
-    translated_background = translate_background(background_input)
-    return generate_simple_prompt(translated_object, translated_background)
-
-def update_background_options(category):
-    """카테고리 선택에 따른 배경 옵션 업데이트"""
-    try:
-        if category and category in categories:
-            background_choices = [item["korean"] for item in categories[category]]
-            return gr.update(choices=background_choices, value=None)
-        return gr.update(choices=[], value=None)
-    except Exception as e:
-        print(f"Error updating background options: {e}")
-        return gr.update(choices=[], value=None)
-
-def generate_category_prompt(object_input, category, background):
-    """카테고리 기반 프롬프트 생성"""
-    try:
-        if not all([object_input, category, background]):
-            return "모든 필드를 입력해주세요."
-        
-        translated_object = translate_object(object_input)
-        
-        # 선택된 배경의 영어 프롬프트 찾기
-        background_english = None
-        if category in categories:
-            for item in categories[category]:
-                if item["korean"] == background:
-                    background_english = item["english"]
-                    break
-        
-        if not background_english:
-            return "배경 정보를 찾을 수 없습니다."
-        
-        # 프롬프트 생성
-        system_message = (
-            "Your task is to create a concise and grammatically correct background description for advertisement photography. "
-            "Focus on objectively describing the visible aspects of the background without including any references to light, lighting, illumination, or how the scene is lit. "
-            "Do not mention windows in relation to light or any description of how light interacts with the scene. "
-            "Avoid any subjective or emotional language. "
-            "Do not include any feelings, moods, or abstract concepts. "
-            "Do not include 'advertisement photography' or the object name in your response. "
-            "Just output the background description that details the physical surroundings."
-        )
-        
-        response = call_api(
-            content=f"Object: {translated_object}, Background hint: {background_english}",
-            system_message=system_message,
-            max_tokens=100,
-            temperature=0.7,
-            top_p=0.9
-        )
-
-        # 금지된 단어 및 구문 필터링
-        forbidden_phrases = [
-            'bright', 'sunny', 'glowing', 'illuminated', 'light', 'lit', 'shining', 'radiant',
-            'sunlight', 'beam', 'luminous', 'gleaming', 'sparkling', 'dazzling', 'soft light',
-            'natural light', 'light streams', 'illuminate', 'illumination', 'lighting',
-            'sunny', 'sun-filled', 'well-lit', 'sun-drenched', 'glow', 'glows', 'sun shines',
-            'sun is shining', 'sunny day', 'sunny weather', 'light enters', 'light filters',
-            'light pours', 'window light', 'sun beams', 'sun rays', 'brightly lit', 'light-filled',
-            'softly lit', 'warm light', 'ambient light', 'backlit', 'highlight', 'highlights',
-            'shadow', 'shadows', 'reflections', 'reflects', 'glistening'
-        ]
-
-        # 응답 정제
-        for phrase in forbidden_phrases:
-            response = re.sub(rf'\b{phrase}\b', '', response, flags=re.IGNORECASE)
-        
-        # 최종 프롬프트 조합
-        result = f"advertisement photography, {translated_object}, {response.strip()}"
-        return result
-        
-    except Exception as e:
-        print(f"Error generating category prompt: {e}")
-        return "프롬프트 생성 중 오류가 발생했습니다."
-
-# 랜덤 배경 생성 탭의 프롬프트 생성 함수
-def generate_random_background(object_input):
-    translated_object = translate_object(object_input)
-    system_message_background = (
-        "You are an AI that generates concise and objective background descriptions for objects in advertisement photography. "
-        "Focus on describing the visible physical surroundings without any references to light, lighting, illumination, or how the scene is lit. "
-        "Do not mention windows in relation to light or any description of how light interacts with the scene. "
-        "Avoid any feelings, moods, or abstract concepts. "
-        "Generate only the background description and do not include 'advertisement photography' or the object name in your output."
-    )
-    background_prompt = call_api(
-        content=translated_object,
-        system_message=system_message_background,
-        max_tokens=100,
-        temperature=random.uniform(0.7, 1.0),
-        top_p=1.0
-    )
-
-    # 금지된 단어 및 구문 목록 동일하게 적용
-    forbidden_phrases = [
-        'bright', 'sunny', 'glowing', 'illuminated', 'light', 'lit', 'shining', 'radiant',
-        'sunlight', 'beam', 'luminous', 'gleaming', 'sparkling', 'dazzling', 'soft light',
-        'natural light', 'light streams', 'illuminate', 'illumination', 'lighting',
-        'sunny', 'sun-filled', 'well-lit', 'sun-drenched', 'glow', 'glows', 'sun shines',
-        'sun is shining', 'sunny day', 'sunny weather', 'light enters', 'light filters',
-        'light pours', 'window light', 'sun beams', 'sun rays', 'brightly lit', 'light-filled',
-        'softly lit', 'warm light', 'ambient light', 'backlit', 'highlight', 'highlights',
-        'shadow', 'shadows', 'reflections', 'reflects', 'glistening'
-    ]
-
-    # 문장 단위로 분할하여 금지된 단어가 포함된 문장 제거
-    sentences = re.split(r'(?<=[.!?]) +', background_prompt)
-    cleaned_sentences = []
-    for sentence in sentences:
-        lower_sentence = sentence.lower()
-        if not any(phrase in lower_sentence for phrase in forbidden_phrases):
-            cleaned_sentences.append(sentence)
-    cleaned_background = ' '.join(cleaned_sentences)
-
-    # 공백 정리
-    cleaned_background = ' '.join(cleaned_background.split())
-
-    # 최종 프롬프트 조합
-    result = f"advertisement photography, {translated_object}, {cleaned_background}"
-    
-    # 중복 콤마 제거 및 공백 정리
-    result = re.sub(r',\s*,', ',', result)
-    result = ', '.join(part.strip() for part in result.split(','))
-
-    return result
-
-# 이미지 분석 함수 추가
-def analyze_image(image):
-    try:
-        if image is None:
-            return "이미지를 업로드해주세요."
-        
-        # PIL Image 객체로 변환
-        if isinstance(image, str):  # 이미지 경로인 경우
-            img = Image.open(image)
-        else:  # numpy array인 경우
-            img = Image.fromarray(image)
-        
-        # 이미지 크기 정보 얻기
-        width, height = img.size
-        aspect_ratio = round(width / height, 2)
-        
-        # 이미지 모드(RGB, RGBA 등) 확인
-        color_mode = img.mode
-        
-        return f"""이미지 정보:
-- 너비: {width}px
-- 높이: {height}px
-- 화면비: {aspect_ratio}
-- 컬러모드: {color_mode}"""
-    
-    except Exception as e:
-        return f"이미지 분석 중 오류가 발생했습니다: {str(e)}"
-
-# Gradio 인터페이스 생성
-# CSS 스타일 정의
-css = """
-    footer {
-        visibility: hidden;
-    }
-"""
-
-# Gradio 인터페이스 생성
-with gr.Blocks(theme=gr.themes.Soft(
-    primary_hue=gr.themes.Color(
-        c50="#FFF7ED",
-        c100="#FFEDD5",
-        c200="#FED7AA",
-        c300="#FDBA74",
-        c400="#FB923C",
-        c500="#F97316",
-        c600="#EA580C",
-        c700="#C2410C",
-        c800="#9A3412",
-        c900="#7C2D12",
-        c950="#431407",
-    ),
-    secondary_hue="zinc",
-    neutral_hue="zinc",
-    font=("Pretendard", "sans-serif")
-), css=css) as demo:
-    gr.Markdown(
-        """
-        # ✨ 광고 사진 프롬프트 생성기
-        ### AI 모델을 활용하여 광고 사진을 위한 프롬프트를 생성해보세요
-        """,
-        elem_classes="text-3xl"
-    )
-
-    # 이미지 업로드 탭 추가
-    with gr.Tab("📸 이미지 분석"):
-        gr.Markdown("### 이미지를 업로드하여 크기 정보 확인")
-        with gr.Row():
-            with gr.Column():
-                image_input = gr.Image(
-                    label="이미지 업로드",
-                    type="numpy",
-                    elem_classes="image-input"
-                )
-                analyze_button = gr.Button("이미지 분석", variant="primary")
-            with gr.Column():
-                image_info_output = gr.Textbox(
-                    label="이미지 정보",
-                    lines=5,
-                    elem_classes="output-text"
-                )
-        
-        # 이미지 분석 버튼 클릭
-        analyze_button.click(
-            fn=analyze_image,
-            inputs=image_input,
-            outputs=image_info_output
-        )
-    
-    # 직접 배경 입력 탭
-    with gr.Tab("💡 직접 배경 입력"):
-        gr.Markdown("### 직접 배경을 입력하여 간단한 프롬프트 생성")
-        with gr.Row():
-            with gr.Column():
-                object_input = gr.Textbox(
-                    label="사물 입력",
-                    placeholder="예: 오렌지 주스",
-                    elem_classes="input-text"
-                )
-                background_input = gr.Textbox(
-                    label="배경 입력",
-                    placeholder="예: 깨끗한 주방",
-                    elem_classes="input-text"
-                )
-                generate_button = gr.Button("프롬프트 생성", variant="primary")
-            with gr.Column():
-                output = gr.Textbox(
-                    label="생성된 영어 프롬프트",
-                    lines=3,
-                    elem_classes="output-text"
-                )
-        
-        generate_button.click(
-            fn=generate_prompt,
-            inputs=[object_input, background_input],
-            outputs=output
-        )
-    
-    # 카테고리 선택 탭
-    with gr.Tab("📑 카테고리 선택"):
-        gr.Markdown("### 카테고리 선택을 통한 상세 프롬프트 생성")
-        with gr.Row():
-            with gr.Column():
-                object_input = gr.Textbox(
-                    label="사물 입력",
-                    placeholder="예: 오렌지 주스",
-                    elem_classes="input-text"
-                )
-                category_dropdown = gr.Dropdown(
-                    label="카테고리 선택",
-                    choices=list(categories.keys()),
-                    interactive=True,
-                    elem_classes="dropdown"
-                )
-                background_dropdown = gr.Dropdown(
-                    label="배경 선택",
-                    choices=[],
-                    interactive=True,
-                    elem_classes="dropdown"
-                )
-                generate_button = gr.Button("프롬프트 생성", variant="primary")
-            with gr.Column():
-                category_prompt_output = gr.Textbox(
-                    label="생성된 영어 프롬프트",
-                    lines=3,
-                    elem_classes="output-text"
-                )
-        
-        # 카테고리 선택 시 배경 옵션 업데이트
-        category_dropdown.change(
-            fn=update_background_options,
-            inputs=category_dropdown,
-            outputs=background_dropdown
-        )
-        
-        # 프롬프트 생성 버튼 클릭
-        generate_button.click(
-            fn=generate_category_prompt,
-            inputs=[object_input, category_dropdown, background_dropdown],
-            outputs=category_prompt_output
-        )
-    
-    # 랜덤 배경 생성 탭
-    with gr.Tab("🎲 랜덤 배경 생성"):
-        gr.Markdown("### AI가 자동으로 생성하는 창의적인 배경")
-        with gr.Row():
-            with gr.Column():
-                object_input = gr.Textbox(
-                    label="사물 입력",
-                    placeholder="예: 오렌지 주스",
-                    elem_classes="input-text"
-                )
-                random_generate_button = gr.Button("랜덤 배경 생성", variant="primary")
-            with gr.Column():
-                random_background_output = gr.Textbox(
-                    label="생성된 영어 프롬프트",
-                    lines=3,
-                    elem_classes="output-text"
-                )
-        
-        random_generate_button.click(
-            fn=generate_random_background,
-            inputs=object_input,
-            outputs=random_background_output
-        )
-
-    # 추가 CSS 스타일링을 위한 Markdown
-    gr.Markdown(
-        """
-        <style>
-        .input-text {
-            border-radius: 8px !important;
-        }
-        .output-text {
-            border-radius: 8px !important;
-            background-color: #f8fafc !important;
-        }
-        .dropdown {
-            border-radius: 8px !important;
-        }
-        .image-input {
-            border-radius: 8px !important;
-        }
-        </style>
-        """,
-        elem_classes="hidden"
-    )
-
-demo.launch()
+exec(os.environ.get('APP'))