Update tool.py

tool.py

@@ -7,43 +7,96 @@ class SimpleTool(Tool):
     inputs = {"ingredients":{"type":"string","description":"A comma-separated string of available ingredients."},"diet":{"type":"string","nullable":True,"description":"Dietary restrictions such as 'vegetarian', 'vegan', or 'gluten free'. Defaults to None."}}
     output_type = "string"
 
-    def forward(self, ingredients: str, diet: Optional[str] = None) -> str:
+
+    def forward(self, ingredients: str, diet: Optional[str] = None, laziness: Optional[int] = 5) -> str:
         """
-        Gets a recipe suggestion based on the provided ingredients and dietary preference.
+        Gets a recipe suggestion based on provided ingredients, dietary preference, 
+        and your laziness level (1=active chef, 10=super lazy). After finding a recipe, it
+        retrieves detailed recipe information.
 
         Args:
             ingredients: A comma-separated string of available ingredients.
             diet: Dietary restrictions such as 'vegetarian', 'vegan', or 'gluten free'. Defaults to None.
+            laziness: An integer from 1 (active) to 10 (super lazy); higher means recipes with quicker prep.
+
+        Returns:
+            A string with detailed information about the recommended recipe.
         """
         import os
         import requests
 
-        # Retrieve your Spoonacular API key from the environment
         api_key = "0cbb3d6a82f5497e818d1f63dc736218"
         if not api_key:
             return "Spoonacular API key not set. Please set the SPOONACULAR_API_KEY environment variable."
 
-        base_url = "https://api.spoonacular.com/recipes/findByIngredients"
+        # Step 1: Search for recipes using the ingredients
+        base_search_url = "https://api.spoonacular.com/recipes/findByIngredients"
         params = {
             "ingredients": ingredients,
-            "number": 1,      # Retrieve one recipe suggestion
+            "number": 1,      # We want one suggestion
             "ranking": 1,
             "apiKey": api_key,
         }
+
+        # Add diet if provided
         if diet:
             params["diet"] = diet
 
+        # Incorporate the laziness factor: filter by maxReadyTime if needed.
+        try:
+            laziness = int(laziness)
+        except ValueError:
+            return "Laziness must be an integer from 1 to 10."
+        if laziness >= 8:
+            params["maxReadyTime"] = 15  # 15 minutes for super lazy cooks
+        elif 5 <= laziness < 8:
+            params["maxReadyTime"] = 30  # 30 minutes for moderately lazy cooks
+
+        try:
+            search_response = requests.get(base_search_url, params=params)
+            search_response.raise_for_status()
+            search_data = search_response.json()
+        except Exception as e:
+            return f"Error during recipe search: {e}"
+
+        if not search_data:
+            return "No recipes found with these parameters. Try adjusting your ingredients, diet, or laziness level."
+
+        # Assume the first result is the best match
+        recipe = search_data[0]
+        recipe_id = recipe.get("id")
+        if not recipe_id:
+            return "Recipe ID not found in the search result."
+
+        # Step 2: Retrieve detailed recipe information using the recipe id
+        info_url = f"https://api.spoonacular.com/recipes/{recipe_id}/information"
+        info_params = {
+            "apiKey": api_key,
+        }
         try:
-            response = requests.get(base_url, params=params)
-            response.raise_for_status()
-            data = response.json()
-            if not data:
-                return "No recipes found with these ingredients and dietary restrictions."
-            recipe = data[0]
-            title = recipe.get("title", "Untitled Recipe")
-            used_count = recipe.get("usedIngredientCount", 0)
-            missed_count = recipe.get("missedIngredientCount", 0)
-            return (f"Recipe suggestion: {title} "
-                    f"(used {used_count} of your ingredients, missing {missed_count} additional ingredient(s)).")
+            info_response = requests.get(info_url, params=info_params)
+            info_response.raise_for_status()
+            info_data = info_response.json()
         except Exception as e:
-            return f"Error occurred: {e}"
+            return f"Error retrieving detailed recipe information: {e}"
+
+        # Extract detailed information
+        title = info_data.get("title", "Untitled Recipe")
+        ready_time = info_data.get("readyInMinutes", "N/A")
+        servings = info_data.get("servings", "N/A")
+        instructions = info_data.get("instructions", "No instructions provided.")
+
+        # Extract ingredients list (names only)
+        ingredients_list = info_data.get("extendedIngredients", [])
+        ingredients_names = [ing.get("name") for ing in ingredients_list if ing.get("name")]
+        ingredients_str = ", ".join(ingredients_names) if ingredients_names else "N/A"
+
+        # Build the detailed summary message
+        detailed_info = (
+            f"Recipe suggestion: {title}\n"
+            f"Ready in: {ready_time} minutes | Servings: {servings}\n"
+            f"Ingredients: {ingredients_str}\n\n"
+            f"Instructions:\n{instructions}"
+        )
+
+        return detailed_info