Update app.py

app.py

@@ -182,6 +182,8 @@ def generate_text_better_sampling(model, prompt, max_len=100, max_gen=98, top_k=
 
         yield decoded_text
 
+import gradio as gr
+
 nickname = "사용자"
 
 def respond(message, chat_history):
@@ -189,14 +191,28 @@ def respond(message, chat_history):
     response = ""
     for partial in generate_text_better_sampling(model, message):
         response = partial
-        yield chat_history + [(message, response)]
+        chat_history = chat_history + [(message, response)]
+        yield chat_history, ""
 
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot(elem_id="chatbot", label="InteractGPT 어시스턴트")
-    msg = gr.Textbox(placeholder="말 걸어봐!", label="입력")
-    clear = gr.Button("초기화")
+    chat_history = gr.State([])
+
+    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
+    )
 
-    msg.submit(respond, [msg, chatbot], chatbot)
-    clear.click(lambda: [], None, chatbot)
+    clear.click(lambda: ([], ""), None, [chat_history, chat_display])
 
 demo.launch()