Update space

app.py

@@ -1,15 +1,49 @@
 import gradio as gr
-from Text2Long_text import generate_korean_story
-
-def make_story(user_prompt):
-    # generate_korean_story 함수에 유저 입력과 길이 전달
-    return generate_korean_story(user_prompt, max_length=500)
-
-# Gradio 인터페이스
-demo = gr.Interface(
-    fn=make_story,
-    inputs=gr.Textbox(label="프롬프트(이야기 주제 또는 첫 문장) 입력"),
-    outputs=gr.Textbox(label="AI가 생성한 긴 한국어 이야기")
-)
+
+# 각 파일에서 함수 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
+
+# 모델 이름과 함수사이 매핑
+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,
+}
+
+def generate_selected_images(prompt, selected_models):
+    results = []
+    for name in selected_models:
+        gen_func = MODEL_FUNCTIONS.get(name)
+        if gen_func:
+            img = gen_func(prompt)
+            results.append((name, img))  # 모델이름, 이미지 묶어서
+    # Gallery는 (label, image) 튜플 또는 이미지 리스트 모두 지원
+    return [img for name, img in results]  # label 형태 써도 되고, 그냥 이미지만도 ok
+
+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)
+    generate_btn = gr.Button("이미지 생성하기")
+
+    generate_btn.click(
+        fn=generate_selected_images,
+        inputs=[prompt, models],
+        outputs=gallery
+    )
 
 demo.launch()