Update app.py

app.py

@@ -52,8 +52,7 @@ def postprocess(text):
     return text
 
 
-def predict(input, chatbot, max_length, top_p, temperature, history):
-    chatbot.append((postprocess(input), ""))
+def predict(input, max_length, top_p, temperature, history):
     response = ""
     history.append(input)
 
@@ -75,12 +74,11 @@ def predict(input, chatbot, max_length, top_p, temperature, history):
     )
     for response_piece in generator:
         response += response_piece
-        chatbot[-1] = (chatbot[-1][0], postprocess(response))
 
-        yield chatbot, history
+        yield postprocess(response), history
 
     history.append(response)
-    yield chatbot, history
+    yield response, history
 
 
 def reset_user_input():
@@ -89,7 +87,7 @@ def reset_user_input():
 
 def reset_state():
     history = gr.State(init_history)
-    return [], []
+    return '', '', []
 
 
 max_length = 2048
@@ -116,9 +114,8 @@ with gr.Blocks() as demo:
     submitBtn.click(
         predict, [
             '請為『%s』選取5個公式' %user_input,
-            chatbot, max_length, top_p,
-            temperature, history],
-        [chatbot, history], show_progress=True
+            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])
@@ -128,6 +125,9 @@ with gr.Blocks() as demo:
                           '你需要為故事%轉寫一段故事大綱'
                           '提供一個完整的龍骨給之後的編劇用。') %expand,
                   outputs=final_sample)
-    emptyBtn.click(reset_state, outputs=[chatbot, history], show_progress=True)
+
+    emptyBtn.click(reset_state,
+                   outputs=[five_sample, final_sample, history],
+                   show_progress=True)
 
 demo.queue().launch(share=False, inbrowser=True)