Update app.py

app.py

@@ -181,41 +181,18 @@ 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 stream_respond(message, history):
+def respond(message, chat_history):
     message = message.replace("@사용자1@", nickname)
-    history = history or []
-    bot_response = ""
-
+    response = ""
     for partial in generate_text_better_sampling(model, message):
-        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 = """
-#chatbox {
-    background-color: #111;
-    color: #eee;
-    border-radius: 10px;
-    padding: 15px;
-    font-family: 'D2Coding', monospace;
-    white-space: pre-wrap;
-}
-"""
-
-with gr.Blocks(css=css) as demo:
-    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()
+        response = partial
+        yield response
+
+chat = gr.ChatInterface(
+    fn=respond,
+    title="InteractGPT",
+)
+
+chat.launch()