Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import json
 import operator
 import streamlit as st
+from dotenv import load_dotenv
 from typing import TypedDict, Annotated, Sequence
 from langchain_openai import ChatOpenAI
 from langchain_core.tools import tool
@@ -16,6 +17,7 @@ model = ChatOpenAI(temperature=0)
 
 @tool
 def multiply(first_number: int, second_number: int):
+    """Multiplies two numbers together and returns the result."""
     return first_number * second_number
 
 model_with_tools = model.bind(tools=[convert_to_openai_tool(multiply)])
@@ -70,8 +72,6 @@ with tab1:
         output = app_graph.invoke({"messages": [question]})
         st.success(output['messages'][-1])
 
-    st.info("The app intelligently routes queries through a graph.")
-
 with tab2:
     st.subheader("General Query")
     user_input = st.text_input("Enter your question here")
@@ -85,5 +85,4 @@ with tab2:
             except Exception as e:
                 st.error("Something went wrong. Try again!")
         else:
-            st.warning("Please enter a valid input.")
-
+            st.warning("Please enter a valid input.")