app.py

import gradio as gr
# 이미지 생성 함수들 import
from animagine_xl import generate_animagine_xl
from animesai import generate_animesai
from generate_sdxl_with_refiner import generate_sdxl_with_refiner
from ghibli import generate_ghibli
from realistic import generate_realistic
from sd_turbo import generate_sd_turbo
from waifu import generate_waifu

# 한국어 이야기 생성 함수 import
from Text2Long_text import generate_korean_story

# 모델 함수 매핑
MODEL_FUNCTIONS = {
    "Animagine XL": generate_animagine_xl,
    "Animesai": generate_animesai,
    "SDXL+Refiner": generate_sdxl_with_refiner,
    "Ghibli": generate_ghibli,
    "Realistic": generate_realistic,
    "SD Turbo": generate_sd_turbo,
    "Waifu Diffusion": generate_waifu,
    "한국어 긴 이야기 생성": lambda prompt: generate_korean_story(prompt, max_length=500),
}

def generate_selected_results(prompt, selected_models):
    results = []
    for name in selected_models:
        gen_func = MODEL_FUNCTIONS.get(name)
        if gen_func:
            result = gen_func(prompt)
            results.append((name, result))
    # 결과별로 적절히 구분(이미지는 gallery용, 텍스트는 아래에 표시)
    images = []
    texts = []
    for name, result in results:
        if hasattr(result, "save") or hasattr(result, "show"):
            images.append(result)  # PIL.Image로 간주
        else:
            texts.append(f"[{name}]\n{result}")
    return images, "\n\n".join(texts)

with gr.Blocks() as demo:
    gr.Markdown("## 원하는 생성 모델로 이미지를 생성하거나, 긴 한국어 이야기를 받아보세요")
    prompt = gr.Textbox(label="프롬프트(텍스트를 입력하세요)")
    models = gr.CheckboxGroup(
        choices=list(MODEL_FUNCTIONS.keys()),
        label="사용할 모델을 선택하세요"
    )
    gallery = gr.Gallery(label="생성 이미지 (선택한 모델별)").style(grid=3)
    long_textbox = gr.Textbox(label="생성된 이야기(텍스트)", lines=10, interactive=False)
    generate_btn = gr.Button("생성하기")

    generate_btn.click(
        fn=generate_selected_results,
        inputs=[prompt, models],
        outputs=[gallery, long_textbox]
    )

demo.launch()