Update space

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
-
-# 각 파일에서 함수 import
+# 이미지 생성 함수들 import
 from animagine_xl import generate_animagine_xl
 from animesai import generate_animesai
 from generate_sdxl_with_refiner import generate_sdxl_with_refiner
@@ -9,7 +8,10 @@ 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,
@@ -18,32 +20,41 @@ MODEL_FUNCTIONS = {
     "Realistic": generate_realistic,
     "SD Turbo": generate_sd_turbo,
     "Waifu Diffusion": generate_waifu,
+    "한국어 긴 이야기 생성": lambda prompt: generate_korean_story(prompt, max_length=500),
 }
 
-def generate_selected_images(prompt, selected_models):
+def generate_selected_results(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
+            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("## 원하는 이미지 생성 모델만 선택해서 프롬프트로 이미지를 생성하세요")
+    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("이미지 생성하기")
+    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_images,
+        fn=generate_selected_results,
         inputs=[prompt, models],
-        outputs=gallery
+        outputs=[gallery, long_textbox]
     )
 
 demo.launch()

dsavv.py

@@ -1 +1 @@
-fdagabb
+fdagabb