Spaces:
Sleeping
Sleeping
Update utils/chatbot_interface4.py
Browse files- 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=
|
155 |
)
|
156 |
|
|
|
157 |
reset.click(
|
158 |
fn=reset_output,
|
159 |
inputs=None,
|
160 |
-
outputs=[conversation_state,
|
161 |
)
|
162 |
-
|
|
|
163 |
user_input.submit(
|
164 |
fn=process_input,
|
165 |
-
inputs=[user_input, conversation_state,
|
166 |
-
outputs=[conversation_state,
|
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.")
|