Update app.py

app.py

@@ -191,28 +191,11 @@ def respond(message, chat_history):
     response = ""
     for partial in generate_text_better_sampling(model, message):
         response = partial
-        chat_history = chat_history + [(message, response)]
-        yield chat_history, ""
+        yield response
 
-with gr.Blocks() as demo:
-    chat_history = gr.State([])
+chat = gr.ChatInterface(
+    fn=respond,
+    title="InteractGPT",
+)
 
-    with gr.Column():
-        chat_display = gr.Markdown(value="", elem_id="chat_display", label="대화창")
-        msg = gr.Textbox(placeholder="말 걸어봐!", label="입력")
-        clear = gr.Button("초기화")
-
-    def update_chat_display(history):
-        formatted = ""
-        for user_msg, bot_msg in history:
-            formatted += f"**사용자:** {user_msg}\n\n"
-            formatted += f"**InteractGPT:** {bot_msg}\n\n---\n\n"
-        return formatted
-
-    msg.submit(respond, inputs=[msg, chat_history], outputs=[chat_history, msg]).then(
-        update_chat_display, inputs=chat_history, outputs=chat_display
-    )
-
-    clear.click(lambda: ([], ""), None, [chat_history, chat_display])
-
-demo.launch()
+chat.launch()