"""
설정 및 상수 정의 파일
"""

from dotenv import load_dotenv
import os

# .env 파일 로드
load_dotenv()

# 월드컵 관련 상수
FINAL_ROUND = "## <center>결승전</center>"
VS_IMAGE = "<img src='/gradio_api/file=static/image/vs.png' style='margin-top: 100px;'>"
WORLDCUPABLE_IMAGE_CNT = [4, 8, 16]
BATTLE_IMAGE_INIT = "battle-image fade-in-visibility"

# 이미지 생성 관련 상수
HIGH_PRICE_OPTION = "비싼거(60원/장)"
LOW_PRICE_OPTION = "싼거(5원/장)"
HIGH_PRICE = 60
LOW_PRICE = 5
CUSTOM = "custom(비싼모델전용)"

# 예제 이미지 URL
EXAMPLE_SPRING_URLS = [
    "static/examples/out-1.webp",
    "static/examples/out-2.webp",
    "static/examples/out-3.webp",
    "static/examples/out-4.webp",
    "static/examples/out-5.webp",
    "static/examples/out-6.webp",
    "static/examples/out-7.webp",
    "static/examples/out-0.webp",
]

EXAMPLE_BEAUTY_URLS = [
    "static/beauty/out-1.webp",
    "static/beauty/out-2.webp",
    "static/beauty/out-3.webp",
    "static/beauty/out-4.webp",
    "static/beauty/out-5.webp",
    "static/beauty/out-6.webp",
    "static/beauty/out-7.webp",
    "static/beauty/out-0.webp",
]

EXAMPLE_ENTRY_URLS = []
# EXAMPLE_ENTRY_URLS = EXAMPLE_BEAUTY_URLS + EXAMPLE_SPRING_URLS
# shuffle(EXAMPLE_ENTRY_URLS)

# 샘플 프롬프트
examplePrompt = "A breathtaking beautiful Korean woman."

# 이미지 스타일 맵핑
imageStyleMap = {
    "알아서": "Something that you think is the best.",
    "실사": "photo-realistic",
    "2D애니": "Classic 90s Disney animation style, 2D hand-drawn aesthetic, rich color palette, traditional cel animation look, expressive character design, detailed painted backgrounds, dramatic lighting, Disney Renaissance era, vibrant scenery, soft shadows, cinematic composition, animated features, warm color tones, characterized by Glen Keane and Mark Henn artistic influence, Disney traditional animation techniques, theatrical quality.",
    "일본2D애니": "Studio Ghibli style, Hayao Miyazaki aesthetic, hand-drawn animation, whimsical natural environments, soft pastel color palette, detailed background art, nostalgic lighting, ethereal atmosphere, fantastical elements blended with realism, delicate line work",
    "3D애니메이션": "3d_animation, Disney Pixar",
    "웹툰": "webtoon, manga",
    "일러스트": "illustration, Hand-drawn art, Artwork",
}

# 이미지 모델 맵핑
imageModelMap = {
    HIGH_PRICE_OPTION: "black-forest-labs/flux-1.1-pro",
    LOW_PRICE_OPTION: "black-forest-labs/flux-schnell",
}

# 에플리케이션 UI 텍스트
APP_TITLE = """
      <div style="display: flex; align-items: center; justify-content: space-between; width: 100%">
          <div style="display: flex; align-items: center; gap: 10px">
              <picture>
                <source srcset="/gradio_api/file=static/image/daekyo_logo_kr_white.png" media="(prefers-color-scheme: dark)">
                <img src="/gradio_api/file=static/image/daekyo_logo_kr.png" width='70' height='70'/>
              </picture>
              <span style="font-size: 40px; font-weight: bold; margin-left: 10px;"> 대교 AI 콘텐츠 이미지 월드컵 🏆</span>
          </div>
          <a href="/logout" style="text-decoration: none;">
              <button style="padding: 8px 16px; font-size: 16px; background-color: #446699; color: white; border: none; border-radius: 4px; cursor: pointer;">Logout</button>
          </a>
      </div>
    """

# 이미지 생성 샘플 텍스트
GEN_IMAGE_DESC = """봄을 담은 정원
It was the start of spring.
Our teacher had a great idea.
"Let's make a vegetable garden behind the school!"
We all worked together.
First, we dug holes.
Next, we planted different seeds.
After that, we watered the garden every day.
Tiny plants soon appeared in the soil.
Slowly, they got bigger and bigger.
Finally, we picked the vegetables!
There were tomatoes, lettuce, and carrots.
We cut them into small pieces and made a colorful salad!
It was fresh and delicious. """