Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from langchain_google_genai import GoogleGenerativeAIEmbeddings, ChatGoogleGenerativeAI
 from langchain_community.agent_toolkits.load_tools import load_tools
 from langchain.agents import create_react_agent
-from langchain.prompts import PromptTemplate  # Import PromptTemplate
+from langchain.prompts import PromptTemplate
 
 # Set up Google API keys from environment variables
 GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
@@ -20,10 +20,16 @@ llm = ChatGoogleGenerativeAI(model="gemini-1.5-pro")
 # Load tools
 tools = load_tools(["serpapi", "llm-math"], llm=llm, serpapi_api_key=SERPER_API_KEY)
 
-# Define a prompt using PromptTemplate
+# Define a prompt using PromptTemplate with required variables
 prompt = PromptTemplate(
-    input_variables=["query"],
-    template="You are a helpful assistant that answers questions based on the provided tools. Question: {query}"
+    input_variables=["query", "agent_scratchpad", "tools", "tool_names"],
+    template=(
+        "You are a helpful assistant that answers questions based on the provided tools.\n"
+        "Tools available: {tool_names}\n"
+        "Current tools: {tools}\n"
+        "Scratchpad: {agent_scratchpad}\n"
+        "Question: {query}"
+    )
 )
 
 # Initialize the agent with the prompt