Update app.py

app.py

@@ -1,23 +1,20 @@
 
-# app.py —— 完整修正版（現代中文故事生成器）
+
 from transformers import pipeline, set_seed
 import gradio as gr
 import torch
 import re
 
-# 🎯 使用現代中文訓練的模型（CLUE 語料，含新聞/百科/論壇）
 MODEL_NAME = "uer/gpt2-chinese-cluecorpussmall"
 
-# 安全載入模型
 generator = pipeline(
     "text-generation",
     model=MODEL_NAME,
-    device=-1,           # 強制使用 CPU
+    device=-1,
     torch_dtype=torch.float32
 )
 
 def is_mostly_chinese(text, threshold=0.7):
-    """檢查文字是否至少 70% 是中文字或中文標點"""
     if not text.strip():
         return False
     total = len(text)
@@ -26,24 +23,24 @@ def is_mostly_chinese(text, threshold=0.7):
     )
     return (chinese_count / total) >= threshold if total > 0 else False
 
-def generate_story(starter, max_length=80, temperature=0.9):
+def generate_story(starter, max_length=100, temperature=0.9):
     if not starter.strip():
         return "請輸入一個故事開頭喔！"
 
-    # ✅ 關鍵：用「現代童話」風格引導，明確要求「迪士尼風格」
+    # ✅ 要求更長、更豐富的故事
     prompt = f"""請用現代繁體中文寫一個迪士尼風格的童話故事。
 開頭：「{starter}」
-要求：60字以內，有魔法、有動物朋友、結尾溫馨。
+要求：150字以內，要有魔法、動物朋友、情節轉折、溫馨結局。
 故事："""
 
     try:
-        set_seed(42)  # 固定種子，讓結果較穩定
+        set_seed(42)
         outputs = generator(
             prompt,
-            max_new_tokens=min(int(max_length), 300),
+            max_new_tokens=int(max_length),  # ✅ 完全跟隨滑桿值
             temperature=float(temperature),
             do_sample=True,
-            top_k=60,        # 提高創意
+            top_k=60,
             top_p=0.9,
             repetition_penalty=1.2,
             pad_token_id=generator.tokenizer.eos_token_id
@@ -51,59 +48,52 @@ def generate_story(starter, max_length=80, temperature=0.9):
 
         full_text = outputs[0]['generated_text']
 
-        # 提取「故事：」之後的內容
         if "故事：" in full_text:
             story = full_text.split("故事：", 1)[-1].strip()
         else:
             story = full_text.replace(prompt, "").strip()
 
-        # 清理多餘空格與符號
         story = re.sub(r'\s+', ' ', story).strip()
         story = re.sub(r'[「」]+', '', story)
-        story = story.split('\n')[0]  # 只取第一句，避免後面亂加
 
-        # 防呆：內容太短或非中文
-        if len(story) < 10 or not is_mostly_chinese(story):
-            return "AI 今天靈感枯竭... 試試換個開頭或調高『創意溫度』！"
+        if len(story) < 20 or not is_mostly_chinese(story):  # ✅ 放寬最小字數
+            return "AI 今天文思泉湧... 試試調高『創意溫度』或再按一次！"
 
         return story
 
     except Exception as e:
-        return f"❌ 錯誤：{str(e)}\n建議降低『最大長度』或重試"
+        return f"❌ 錯誤：{str(e)}\n建議降低長度或重試"
 
-# 建立 Gradio 介面
 demo = gr.Interface(
     fn=generate_story,
     inputs=[
         gr.Textbox(
             lines=2,
-            placeholder="例如：迪士尼公主的魔法口紅不見了...",
+            placeholder="例如：小矮人發現了一張藏寶圖，指向雲端的糖果城堡...",
             label="📖 輸入你的故事開頭"
         ),
-        gr.Slider(40, 300, value=70, step=10, label="📏 故事長度"),
+        gr.Slider(50, 200, value=120, step=10, label="📏 故事長度（token數）"),  # ✅ 最高 200
         gr.Slider(0.8, 1.3, value=1.0, step=0.1, label="🎨 創意溫度")
     ],
     outputs=gr.Textbox(
         label="📜 AI 生成的童話故事",
-        lines=8,
-        max_lines=20,
-        placeholder="魔法故事即將誕生...",
+        lines=12,        # ✅ 增加顯示行數
+        max_lines=30,    # ✅ 增加最大行數
+        placeholder="長篇魔法故事即將誕生...",
         autoscroll=True
     ),
-    title="🐉 現代中文 AI 說書人",
-    description="✨ 專為「迪士尼」「現代詞彙」優化！輸入開頭，AI 幫你寫出溫馨奇幻故事～",
+    title="🐉 長篇中文 AI 說書人",
+    description="✨ 支援更長故事！輸入開頭，AI 幫你寫出情節豐富的奇幻冒險～",
     examples=[
-        ["迪士尼公主的魔法口紅不見了...", 70, 1.0],
-        ["皮卡丘開了一家深夜拉麵店...", 80, 1.1],
-        ["我的手機殼裡住著一隻小精靈...", 75, 0.9]
+        ["小矮人發現了一張藏寶圖，指向雲端的糖果城堡...", 150, 1.1],
+        ["會說話的雲朵邀請我去參加天空動物園的開幕典禮...", 180, 1.2],
+        ["我的鬧鐘其實是時光機，每天早上帶我去不同年代...", 160, 1.0]
     ],
     theme=gr.themes.Soft()
 )
 
-# 可選：加大字體
 demo.css = """
 .output-text { font-size: 18px !important; line-height: 1.8; }
-.gr-button-primary { background: #ff6b6b !important; }
 """
 
 if __name__ == "__main__":