Update app.py

app.py

@@ -7,6 +7,7 @@ from PIL import Image
 
 # Import our custom tools from their modules
 from tools import evaluate_consumption, evaluate_consumption_example
+from retriever import FrugalAI_methods
 
 HF_TOKEN = os.environ.get("HF_TOKEN") 
 API_KEY = os.environ.get("API_KEY") 
@@ -24,10 +25,11 @@ headers = {
 # Initialize the weather tool
 evaluate_consumption = evaluate_consumption()
 evaluate_consumption_example = evaluate_consumption_example()
+FrugalAI_methods=FrugalAI_methods()
 
 # Create Alfred with all the tools
 alfred = CodeAgent(
-    tools=[evaluate_consumption, evaluate_consumption_example], 
+    tools=[evaluate_consumption, evaluate_consumption_example, FrugalAI_methods], 
     model=model,
     additional_authorized_imports=['os'],
     add_base_tools=True,  # Add any additional base tools