Update app.py

app.py

@@ -199,26 +199,22 @@ def generate_text_topkp_stream(model, prompt, max_len=100, max_gen=98, p=0.9, k=
             break
 
         yield decoded_text
-    
-history_text = ""
 
-def chat_stream(user_input):
-    global history_text
+def chat_stream(user_input, history_text):
     partial_text = ""
     for partial_response in generate_text_topkp_stream(model, user_input):
         partial_text = partial_response
-        yield history_text + f"사용자: {user_input}\nKeraLux: {partial_text}\n"
-
-    # 최종 응답 업데이트
-    history_text += f"사용자: {user_input}\nKeraLux: {partial_text}\n"
+        yield history_text + f"사용자: {user_input}\nKeraLux: {partial_text}\n", \
+              history_text + f"사용자: {user_input}\nKeraLux: {partial_text}\n"
 
 with gr.Blocks() as demo:
     gr.Markdown("### 📟 KeraLux Textbot with Streaming")
 
     textbox = gr.Textbox(placeholder="메시지를 입력하세요", lines=1)
     output_area = gr.Textbox(label="대화 기록", lines=20, interactive=False)
-    
-    textbox.submit(chat_stream, inputs=textbox, outputs=output_area)
+    state = gr.State("")  # 세션별 저장소
+
+    textbox.submit(chat_stream, inputs=[textbox, state], outputs=[output_area, state])
     textbox.submit(lambda: "", None, textbox)
 
 demo.launch()