Update src/chatbot.py

src/chatbot.py

@@ -7,11 +7,9 @@ from .models import route_llm, prompt
 def process_input(user_input, session_id='1'):
     """
     Processes the user input and generates a response using the conversation chain.
-    
     Parameters:
         user_input (str): The user's input message.
         session_id (str): The session ID for the chat (default is "1").
-    
     Returns:
         str: The generated response from the chatbot.
     """
@@ -35,27 +33,27 @@ def process_input(user_input, session_id='1'):
     
     return response
 
-# Function for Gradio interface
+# Gradio interface function to handle input
 def chatbot_interface(user_input, chat_history=None, session_id="1"):
     """
     Interface function for Gradio to handle input and output between the user and the chatbot.
-    
     Parameters:
         user_input (str): The user's input message.
         session_id (str): The session ID for the chat (default is "1").
         chat_history (list): List of previous chat messages in the format [[user, bot], ...]
-    
     Returns:
         list: Updated chat history including the new user and bot messages.
     """
     if chat_history is None:
         chat_history = []
     
+    # Greeting at the start of the chat
     if user_input == "":
         bot_response = "Hi there! How can I help you today?"
     else:
         bot_response = process_input(user_input, session_id)
     
+    # Add user input and bot response to chat history
     chat_history.append([user_input, bot_response])
     
     return chat_history
@@ -68,5 +66,4 @@ def launch_gradio_interface():
         outputs=gr.Chatbot(label="Chat History"),
         title="AI Chatbot",
         live=False
-    ).launch()
-
+    ).launch()