Update utils/chatbot_interface.py

utils/chatbot_interface.py

@@ -46,10 +46,10 @@ class ChatbotInterface:
                 api_key=api_key,
                 meta_prompt_file=meta_prompt_file
             )
-            # Start a user session
-            self.session_id = self.response_manager.start_session()  
-            logging.info(f"New session started with session id {self.session_id}.")
-            self.generate_response = lambda query, history: self.response_manager.generate_response(query, self.session_id)  
+            # # Start a user session
+            # self.session_id = self.response_manager.start_session()  
+            # logging.info(f"New session started with session id {self.session_id}.")
+            # self.generate_response = lambda query, history: self.response_manager.generate_response(query, self.session_id)  
 
             logging.info(
                 "ChatbotInterface initialized with the following parameters:\n"
@@ -91,14 +91,13 @@ class ChatbotInterface:
         logging.info("Configuration loaded successfully.")
         return config
 
-    def reset_output(self) -> list:
+    def reset_output(self) -> tuple:
         """
         Reset the chatbot output and start a new session.
-        :return: An empty list to reset the output.
+        :return: An empty list and a new session ID to reset the output.
         """
-        # Restart session on reset
-        self.session_id = self.response_manager.start_session()  
-        return []
+        new_session_id = self.response_manager.start_session()
+        return [], new_session_id
 
     def create_interface(self) -> gr.Blocks:
         """
@@ -121,15 +120,33 @@ class ChatbotInterface:
                 placeholder=self.input_placeholder
             )
 
+            # Session state
+            session_state = gr.State(self.response_manager.start_session())  
+
             # Buttons
             with gr.Row():
                 reset = gr.Button(self.reset_button, variant="secondary")
                 submit = gr.Button(self.submit_button, variant="primary")
 
             # Button actions
-            submit.click(fn=self.generate_response, inputs=[user_input, chatbot_output], outputs=chatbot_output)
-            user_input.submit(fn=self.generate_response, inputs=[user_input, chatbot_output], outputs=chatbot_output)
-            reset.click(fn=self.reset_output, inputs=None, outputs=chatbot_output)
+            submit.click(
+                fn=lambda query, session_id: self.response_manager.generate_response(query, session_id),
+                inputs=[user_input, session_state], outputs=[chatbot_output]
+            )
+            user_input.submit(
+                fn=lambda query, session_id: self.response_manager.generate_response(query, session_id),
+                inputs=[user_input, session_state], outputs=[chatbot_output]
+            )
+            reset.click(
+                fn=self.reset_output,
+                inputs=None,
+                outputs=[chatbot_output, session_state]  
+            )
+
+            # # Button actions
+            # submit.click(fn=self.generate_response, inputs=[user_input, chatbot_output], outputs=chatbot_output)
+            # user_input.submit(fn=self.generate_response, inputs=[user_input, chatbot_output], outputs=chatbot_output)
+            # reset.click(fn=self.reset_output, inputs=None, outputs=chatbot_output)
 
         logging.info("Gradio interface created successfully.")
         return demo