AashitaK commited on
Commit
f064c52
·
verified ·
1 Parent(s): 2d63d0f

Update utils/chatbot_interface4.py

Browse files
Files changed (1) hide show
  1. utils/chatbot_interface4.py +10 -8
utils/chatbot_interface4.py CHANGED
@@ -140,30 +140,32 @@ class ChatbotInterface:
140
  # Reset function updated to reset ResponseManager
141
  def reset_output():
142
  response_manager = init_response_manager()
143
- return [], response_manager, "" # Returns [chatbot_output, response_manager_state, user_input]
144
 
145
  # Process input now uses session-specific ResponseManager
146
  def process_input(user_message, chat_history, response_manager):
147
  updated_history = response_manager.generate_response(user_message, chat_history)
148
- return updated_history, response_manager, "" # Returns [chatbot_output, conversation_state, response_manager_state, user_input]
149
 
150
- # Initialize ResponseManager on load
151
  demo.load(
152
  fn=init_response_manager,
153
  inputs=None,
154
- outputs=response_manager_object
155
  )
156
 
 
157
  reset.click(
158
  fn=reset_output,
159
  inputs=None,
160
- outputs=[conversation_state, response_manager_object, user_input]
161
  )
162
-
 
163
  user_input.submit(
164
  fn=process_input,
165
- inputs=[user_input, conversation_state, response_manager_object],
166
- outputs=[conversation_state, response_manager_object, user_input]
167
  )
168
 
169
  logging.info("Gradio interface created successfully.")
 
140
  # Reset function updated to reset ResponseManager
141
  def reset_output():
142
  response_manager = init_response_manager()
143
+ return [], [], response_manager, "" # Returns [chatbot_output,conversation_state, response_manager_state, user_input]
144
 
145
  # Process input now uses session-specific ResponseManager
146
  def process_input(user_message, chat_history, response_manager):
147
  updated_history = response_manager.generate_response(user_message, chat_history)
148
+ return updated_history, updated_history, response_manager, "" # Returns [chatbot_output, conversation_state, response_manager_state, user_input]
149
 
150
+ # Initialize ResponseManager object for a session on load
151
  demo.load(
152
  fn=init_response_manager,
153
  inputs=None,
154
+ outputs=response_manager_state # Each session state gets its own instance of ResponseManager class
155
  )
156
 
157
+ # CLearButton action
158
  reset.click(
159
  fn=reset_output,
160
  inputs=None,
161
+ outputs=[chatbot_output, conversation_state, response_manager_state, user_input]
162
  )
163
+
164
+ # Enter to trigger response generation
165
  user_input.submit(
166
  fn=process_input,
167
+ inputs=[user_input, conversation_state, response_manager_state],
168
+ outputs=[chatbot_output, conversation_state, response_manager_state, user_input]
169
  )
170
 
171
  logging.info("Gradio interface created successfully.")