Update app.py

app.py

@@ -270,17 +270,19 @@ with gr.Blocks(css='#chatbot .overflow-y-auto{height:750px}') as demo:
  
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label='User input')
-    init_examples = [["이번 주에 친구들과 모임이 있는데, 훌륭한 와인 한 병을 추천해줄래?"], ["입문자에게 좋은 와인을 추천해줄래?"], ["보르도와 부르고뉴 와인의 차이점은 뭐야?"]]
-    user_response_examples = gr.Dataset(samples=init_examples, components=[msg])
+    samples = [["이번 주에 친구들과 모임이 있는데, 훌륭한 와인 한 병을 추천해줄래?"], ["입문자에게 좋은 와인을 추천해줄래?"], ["보르도와 부르고뉴 와인의 차이점은 뭐야?"]]
+    user_response_examples = gr.Dataset(samples=samples, components=[msg], type="index")
     stage_history = gr.Textbox(value="stage history: ", interactive=False, label='stage history')
     submit_btn = gr.Button("전송")
-    clear_btn = gr.ClearButton([msg, chatbot, stage_history, ])
+    clear_btn = gr.ClearButton([msg, chatbot, stage_history])
     stage_info = gr.Textbox(value=stage_description, interactive=False, label='stage description')
 
-    def load_example(example):
-        return example[0]
+    def load_example(example_id):
+        global samples
+        return samples[example_id][0]
 
-    def answer(user_response, chat_history, stage_history, user_response_examples):
+    def answer(user_response, chat_history, stage_history):
+        global samples
         chat_history = chat_history or []
         stage_history = stage_history or ""
         pre_conversation_history = ""
@@ -288,7 +290,7 @@ with gr.Blocks(css='#chatbot .overflow-y-auto{height:750px}') as demo:
             pre_conversation_history += f"User: {chat[0]} <END_OF_TURN>\n"
             pre_conversation_history += f"이우선: {chat[1]} <END_OF_TURN>\n"
         conversation_history = pre_conversation_history + f"User: {user_response} <END_OF_TURN>\n"
-        stage_number = stage_analyzer_chain.run({'conversation_history': conversation_history, 'stage_history': stage_history})
+        stage_number = stage_analyzer_chain.run({'conversation_history': conversation_history, 'stage_history': stage_history[1:]})
         stage_number = stage_number[-1]
         stage_history += stage_number if stage_history == "stage history: " else ", " + stage_number
         response = agent_executor.run({'input':user_response, 'conversation_history': pre_conversation_history, 'stage_number': stage_number})
@@ -298,15 +300,18 @@ with gr.Blocks(css='#chatbot .overflow-y-auto{height:750px}') as demo:
         user_response_examples = []
         for user_response_example in user_response_chain.run({'conversation_history': conversation_history}).split('|'):
             user_response_examples.append([user_response_example])
+        samples = user_response_examples
  
-        return "", chat_history, stage_history, gr.Dataset.update(samples=user_response_examples)
+        return "", chat_history, stage_history, gr.Dataset.update(samples=samples)
 
     def clear(user_response_examples):
-        return gr.Dataset.update(samples=init_examples)
+        global samples
+        samples = [["이번 주에 친구들과 모임이 있는데, 훌륭한 와인 한 병을 추천해줄래?"], ["입문자에게 좋은 와인을 추천해줄래?"], ["보르도와 부르고뉴 와인의 차이점은 뭐야?"]]
+        return gr.Dataset.update(samples=samples)
 
     clear_btn.click(fn=clear, inputs=[user_response_examples], outputs=[user_response_examples])
     user_response_examples.click(load_example, inputs=[user_response_examples], outputs=[msg])      
-    submit_btn.click(answer, [msg, chatbot, stage_history, user_response_examples], [msg, chatbot, stage_history, user_response_examples])
-    msg.submit(answer, [msg, chatbot, stage_history, user_response_examples], [msg, chatbot, stage_history, user_response_examples])
+    submit_btn.click(answer, [msg, chatbot, stage_history], [msg, chatbot, stage_history, user_response_examples])
+    msg.submit(answer, [msg, chatbot, stage_history], [msg, chatbot, stage_history, user_response_examples])
 
 demo.launch()