Update app.py

app.py

@@ -53,6 +53,10 @@ def postprocess(text):
 
 
 def predict(input, max_length, top_p, temperature, history):
+    if input.isdigit():
+        input = '現在你作為一個專業的編劇，你需要為故事%s轉寫一段故事大綱提供一個完整的龍骨給之後的編劇用。' %input
+    else:
+        input = '請為『%s』選取5個公式' %input
     response = ""
     history.append(input)
 
@@ -112,19 +116,13 @@ with gr.Blocks() as demo:
     history = gr.State(init_history)
 
     submitBtn.click(
-        predict, [
-            '請為『%s』選取5個公式' %user_input,
-            max_length, top_p, temperature, history],
+        predict, [user_input, max_length, top_p, temperature, history],
         [five_sample, history], show_progress=True
     )
     submitBtn.click(reset_user_input, [], [user_input])
     submitBtn.click(reset_user_input, [], [final_sample])
 
-    expand.select(fn=predict,
-                  inputs=('現在你作為一個專業的編劇，'
-                          '你需要為故事%轉寫一段故事大綱'
-                          '提供一個完整的龍骨給之後的編劇用。') %expand,
-                  outputs=final_sample)
+    expand.select(fn=predict, inputs=expand, outputs=final_sample)
 
     emptyBtn.click(reset_state,
                    outputs=[five_sample, final_sample, history],