Update app.py

app.py

@@ -186,33 +186,16 @@ import gradio as gr
 
 nickname = "사용자"
 
-def respond_stream(message, chat_history):
+def respond(message, chat_history):
     message = message.replace("@사용자1@", nickname)
-    history = chat_history or []
-    bot_response = ""
+    response = ""
     for partial in generate_text_better_sampling(model, message):
-        bot_response = partial
-        # 마지막 대화만 스트리밍 중이라 업데이트
-        if history and len(history) > 0:
-            history[-1] = (history[-1][0], bot_response)
-        else:
-            history.append((message, bot_response))
-        yield history
-
-with gr.Blocks() as demo:
-    chat_history = gr.State([])
-
-    txt_input = gr.TextArea(placeholder="메시지를 입력하세요...", lines=2)
-    chat_display = gr.Markdown()
-
-    def update_display(history):
-        formatted = ""
-        for user_msg, bot_msg in history:
-            formatted += f"**사용자:** {user_msg}\n\n"
-            formatted += f"**봇:** {bot_msg}\n\n"
-        return formatted
-
-    txt_input.submit(respond_stream, inputs=[txt_input, chat_history], outputs=[chat_history], queue=True)\
-            .then(update_display, inputs=chat_history, outputs=chat_display)
-
-demo.launch()
+        response = partial
+        yield response
+
+chat = gr.ChatInterface(
+    fn=respond,
+    title="InteractGPT",
+)
+
+chat.launch()