Update app.py

app.py

@@ -181,37 +181,41 @@ def generate_text_better_sampling(model, prompt, max_len=100, max_gen=98, top_k=
 
         yield decoded_text
 
+import gradio as gr
+
 nickname = "사용자1"
 
-def respond(message, chat_history):
+def stream_respond(message, history):
     message = message.replace("@사용자1@", nickname)
-    response = ""
+    history = history or []
+    bot_response = ""
+
     for partial in generate_text_better_sampling(model, message):
-        response = partial
-        yield response
+        bot_response = partial
+        temp = history + [(nickname, message), ("봇", bot_response)]
+        yield "", temp
+
+def format_history(history):
+    return "\n\n".join([f"**{who}**: {text}" for who, text in history])
 
 css = """
-.chatbot {
-    background-color: #222;
+#chatbox {
+    background-color: #111;
     color: #eee;
     border-radius: 10px;
     padding: 15px;
-    font-family: '나눔고딕', monospace;
-}
-.input-area textarea {
-    background-color: #333;
-    color: #eee;
-    border-radius: 8px;
-    border: none;
-    padding: 10px;
-    font-size: 16px;
+    font-family: 'D2Coding', monospace;
+    white-space: pre-wrap;
 }
 """
 
 with gr.Blocks(css=css) as demo:
-    chatbot = gr.Chatbot(elem_classes="chatbot")
-    txt = gr.Textbox(show_label=False, placeholder="메시지를 입력하세요...")
-    txt.submit(respond, [txt, chatbot], chatbot)
+    state = gr.State([])
+    chat_display = gr.Markdown(elem_id="chatbox")
+    txt = gr.Textbox(placeholder="메시지를 입력하세요...", show_label=False)
+
+    txt.submit(stream_respond, [txt, state], [txt, state])
+    state.change(fn=format_history, inputs=state, outputs=chat_display)
     txt.submit(lambda: "", None, txt)  # 입력창 클리어
 
 demo.launch()