Update app.py

app.py

@@ -181,18 +181,37 @@ def generate_text_better_sampling(model, prompt, max_len=100, max_gen=98, top_k=
 
         yield decoded_text
 
-nickname = "사용자1"
 
-def respond(message, chat_history):
-    message = message.replace("@사용자1@", nickname)
-    response = ""
-    for partial in generate_text_better_sampling(model, message):
-        response = partial
-        yield response
+import gradio as gr
 
-chat = gr.ChatInterface(
-    fn=respond,
-    title="InteractGPT",
-)
+nickname = "사용자"
 
-chat.launch()
+def respond(message, history):
+    message = message.replace("@사용자1@", nickname)
+    history = history or []
+    bot_response = ""
+    for partial in generate_text_better_sampling(model, message):
+        bot_response = partial
+        # history에 메시지와 봇 응답을 넣고, 실시간 업데이트용으로 yield
+        yield history + [(nickname, message), ("봇", bot_response)]
+
+with gr.Blocks() as demo:
+    chat_history = gr.State([])  # 대화 기록 저장용
+    chatbot = gr.Chatbot()  # 안 씀! 그냥 빈 변수로 선언만 함, 실제 UI는 직접 만듦
+    chat_box = gr.Markdown()
+    user_input = gr.Textbox(show_label=False, placeholder="메시지를 입력하세요...")
+
+    def update_chatbox(history):
+        # history를 markdown 형식으로 변환
+        return "\n\n".join([f"**{who}**: {text}" for who, text in history])
+
+    # 실시간 스트리밍으로 대화 내역 업데이트
+    def stream_response(message, history):
+        # respond 함수 호출해서 스트리밍 yield 처리
+        for updated_history in respond(message, history):
+            yield update_chatbox(updated_history), updated_history
+
+    user_input.submit(stream_response, inputs=[user_input, chat_history], outputs=[chat_box, chat_history])
+    user_input.submit(lambda: "", None, user_input)  # 입력창 초기화
+
+demo.launch()