Update app.py

app.py

@@ -104,24 +104,10 @@ def rag_response(query):
 # Define tools using LangChain's `tool` decorator
 @tool
 def knowledge_base_tool(query: str):
-    """
-    Tool function to query the knowledge base and retrieve a response.
-    Args:
-        query (str): The query to search the knowledge base.
-    Returns:
-        str: The response retrieved from the knowledge base.
-    """
     return rag_response(query)
 
 @tool
 def google_search_tool(query: str):
-    """
-    Tool function to perform a Google search using the SERPER API.
-    Args:
-        query (str): The query to search on Google.
-    Returns:
-        list: List of snippets extracted from search results.
-    """
     return google_search(query)
 
 tools = [knowledge_base_tool, google_search_tool]
@@ -185,9 +171,10 @@ def chatbot_response(message, history):
 
 # Streamlit app
 
-# User input
+# Create input field for user message
 user_input = st.text_input("You:", "")
 
+# Create a button for submitting the message
 if st.button("Submit"):
     if user_input:
         response = chatbot_response(user_input, chat_history)