Update utils/chatbot_interface.py

utils/chatbot_interface.py

@@ -2,7 +2,6 @@ import os
 import json
 import logging
 from typing import Optional
-from uuid import uuid4
 import gradio as gr
 from utils.response_manager import ResponseManager 
 
@@ -47,17 +46,16 @@ 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)  
-
+            self.generate_response = self.response_manager.generate_response  
             logging.info(
                 "ChatbotInterface initialized with the following parameters:\n"
                 f"  - Model: {model}\n"
                 f"  - Temperature: {temperature}\n"
                 f"  - Max Output Tokens: {max_output_tokens}\n"
                 f"  - Max Number of Results: {max_num_results}\n"
+                f"  - Vector Store ID: {vector_store_id}\n"
+                f"  - API Key: {'Provided' if api_key else 'Not Provided'}\n"
+                f"  - Meta Prompt File: {meta_prompt_file or 'Default'}"
             )
         except Exception as e:
             logging.error(f"Failed to initialize ResponseManager: {e}")
@@ -92,13 +90,12 @@ class ChatbotInterface:
         logging.info("Configuration loaded successfully.")
         return config
 
-    def reset_output(self) -> tuple:
+    def reset_output(self) -> list:
         """
-        Reset the chatbot output and start a new session.
-        :return: An empty list and a new session ID to reset the output.
+        Reset the chatbot output.
+        :return: An empty list to reset the output.
         """
-        new_session_id = self.response_manager.start_session()
-        return [], new_session_id
+        return []
 
     def create_interface(self) -> gr.Blocks:
         """
@@ -121,33 +118,15 @@ class ChatbotInterface:
                 placeholder=self.input_placeholder
             )
 
-            # Session state
-            session_state = gr.State(str(uuid4()))#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=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)
+            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