Update utils/chatbot_interface_test.py

utils/chatbot_interface_test.py

@@ -17,18 +17,23 @@ class ChatbotInterface:
                  meta_prompt_file: Optional[str] = None):
         """
         Initialize the ChatbotInterface with configuration and custom parameters for ResponseManager.
+        :param config_path: Path to the configuration JSON file.
+        :param model: The OpenAI model to use (default: 'gpt-4o-mini').
+        :param temperature: The temperature for response generation (default: 0).
+        :param max_output_tokens: The maximum number of output tokens (default: 800).
+        :param max_num_results: The maximum number of search results to return (default: 15).
+        :param vector_store_id: The ID of the vector store to use for file search.
+        :param api_key: The OpenAI API key for authentication.
+        :param meta_prompt_file: Path to the meta prompt file .
         """
         self.config = self.load_config(config_path)
-        
         self.title = self.config["chatbot_title"]
         self.description = self.config["chatbot_description"]
-
-        # Use a placeholder in the textbox for a unified look.
-        self.input_placeholder = "How can the CIS agent help you?"  
-
+        self.input_label = self.config["chatbot_input_label"]
+        self.input_placeholder = self.config["chatbot_input_placeholder"]
         self.output_label = self.config["chatbot_output_label"]
-        self.submit_button = self.config["chatbot_submit_button"]
 
+        # Initialize ResponseManager with custom parameters
         try:
             self.response_manager = ResponseManager(
                 model=model,
@@ -47,6 +52,7 @@ class ChatbotInterface:
                 f"  - Max Output Tokens: {max_output_tokens}\n"
                 f"  - Max Number of Results: {max_num_results}\n"
             )
+            
         except Exception as e:
             logging.error(f"Failed to initialize ResponseManager: {e}")
             raise
@@ -95,6 +101,7 @@ class ChatbotInterface:
 
             # Chatbot output area.
             chatbot_output = gr.Chatbot(label=self.output_label, type="messages")
+            
             # Use a gr.Row container as the input box with an integrated submit button.
             with gr.Row(elem_id="input-container", equal_height=True):
                 user_input = gr.Textbox(
@@ -128,7 +135,6 @@ class ChatbotInterface:
                 updated_history = self.response_manager.generate_response(user_message, chat_history)
                 return updated_history, ""
 
-            
             # Bind the reset button click to the reset function
             reset.click(
                 fn=reset_output,
@@ -136,7 +142,6 @@ class ChatbotInterface:
                 outputs=[chatbot_output, user_input]
             )
             
-            
             # Bind the Enter key (textbox submit) to the same processing function
             user_input.submit(
                 fn=process_input,