Update app.py

app.py

@@ -25,15 +25,7 @@ init_history = [
 4. 反轉结局：這是一個比較特別的结局，它通常在故事的最後出現，並且會對讀者產生一種意外的冲击和影響。
 5. 寓言故事：這是一種用故事來說明一個道理或概念的故事，通常包含一個寓意或象徵性的元素。
 
-這些故事公式只是故事創作中的一部分，不同的故事可能会有不同的結構和情節，但這些公式可以作為一個起點，幫助你開始創作出一個引人入勝的故事。'''),
-    myChatMessage(role="user", content='''請為『發生在香港的現代版本《聖誕頌歌》電影，背景從西方的聖誕節換成中式的團年飯。
-
-原版《聖誕頌歌》講述了一位名叫厄尼斯特·斯克魯奇（Ebenezer Scrooge）的富有商人的故事。
-斯克魯奇是一個極度吝嗇和自私的人，在三個幽靈的拜訪下，斯克魯奇被帶回他的童年和青少年時期，看到了他曾經擁有的喜悅和幸福，也看到了他的貪婪和自私對他人造成的傷害。
-他還看到了自己的墳墓，意識到他的小氣將使他成為一個被遺忘和不受人愛戴的人。
-這些經歷改變了斯克魯奇的心態。最終，斯克魯奇成為一個讓人尊敬和受人愛戴的人，並體會到了聖誕節的真正意義。
-
-』提議五個適合的公式。'''),
+這些故事公式只是故事創作中的一部分，不同的故事可能会有不同的結構和情節，但這些公式可以作為一個起點，幫助你開始創作出一個引人入勝的故事。''')
 ]
 
 os.system("wget https://huggingface.co/npc0/chatglm3-6b-int4/resolve/main/chatglm3-ggml-q4_0.bin")
@@ -68,7 +60,7 @@ def predict(input, max_length, top_p, temperature, history):
     elif input.isdigit():
         input = '現在你作為一個專業的編劇，你需要為故事%s轉寫一段故事大綱提供一個完整的龍骨給之後的編劇用。' %input
     else:
-        input = '請為『%s』選取5個公式' %input
+        input = '請為『%s』提議五個適合的公式。' %input
     response = ""
     history.append(myChatMessage(role="user", content=input))
 
@@ -93,8 +85,8 @@ def predict(input, max_length, top_p, temperature, history):
 
         yield postprocess(response), history
 
-    history.append(response)
-    yield response.content, history
+    history.append(response_piece)
+    yield response_piece.content, history
 
 
 def reset_user_input():
@@ -103,11 +95,11 @@ def reset_user_input():
 
 def reset_state():
     history = gr.State(init_history)
-    return '', '', [], gr.Dropdown(value=None, interactive=False)
+    return '', '', [], gr.Dropdown(["1", "2", "3", "4", "5"], label="Step2：進一步擴展的範例", value=None, interactive=False)
 
 
 def allow_user_input():
-    return gr.Dropdown(interactive=True)
+    return gr.Dropdown(["1", "2", "3", "4", "5"], label="Step2：進一步擴展的範例", interactive=True)
 
 
 with gr.Blocks() as demo: