Update app.py

app.py

@@ -243,6 +243,10 @@ import gradio as gr
 
 # user_response, stage_history, conversation_history, pre_conversation_history = "", "", """""", """"""
  
+stage_description = ""
+for key, value in conversation_stages_dict.items():
+    stage_description += f"{key}.{value}\n"
+
 with gr.Blocks(css='#chatbot .overflow-y-auto{height:750px}') as demo:
  
     with gr.Row():
@@ -256,10 +260,11 @@ with gr.Blocks(css='#chatbot .overflow-y-auto{height:750px}') as demo:
         </div>""")
  
     chatbot = gr.Chatbot()
-    msg = gr.Textbox()
-    stage_history = gr.Textbox(visible=False)
+    msg = gr.Textbox(label='User input')
+    stage_history = gr.Textbox(value="stage history: ", interactive=False, label='stage history')
     submit_btn = gr.Button("전송")
-    clear_btn = gr.ClearButton([msg, chatbot])
+    clear_btn = gr.ClearButton([msg, chatbot, stage_history])
+    stage_info = gr.Textbox(value=stage_description, interactive=False, label='stage description')
 
     def answer(user_response, chat_history, stage_history):
         chat_history = chat_history or []
@@ -271,7 +276,7 @@ with gr.Blocks(css='#chatbot .overflow-y-auto{height:750px}') as demo:
         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_number[-1]
-        stage_history += stage_number if stage_history == "" else ", " + stage_number
+        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})
         # conversation_history += "이우선: " + response + "\n"
         response = response.split('<END_OF_TURN>')[0]