Update app.py

app.py

@@ -83,12 +83,167 @@ prompt_python_execute = """You are a python code execution agent.
                            After you're done with your tasks, respond to the supervisor directly
                            Respond ONLY with the results of your work, do NOT include ANY other text."""
 
+prompt_botanical_classification = """You are a vegetable botanical classification agent.
+                           INSTRUCTIONS:
+                           Assist ONLY with tasks related to classificating vegatables, DO NOT do any math
+                           After you're done with your tasks, respond to the supervisor directly
+                           Respond ONLY with the results of your work, do NOT include ANY other text."""
+
 #TOOLS
 web_search = TavilySearch(
         max_results=5,
         topic="general",
     )
 
+def get_botanical_classification_tool(item_name):
+    """
+    Provides the botanical classification (fruit, vegetable, or other)
+    for a given food item, adhering to botanical definitions.
+    Args:
+        item_name (str): The name of the food item (e.g., "bell pepper", "sweet potato").
+    Returns:
+        dict: A dictionary containing:
+              - 'item': The normalized item name.
+              - 'botanical_category': 'fruit', 'vegetable', 'other', or 'unclassified'.
+              - 'botanical_part': The botanical part of the plant (e.g., 'matured ovary', 'root', 'leaf'),
+                                  or 'N/A' if not applicable/unknown.
+              - 'notes': Any additional botanical notes or clarifications.
+    """
+
+    # --- Curated Botanical Database ---
+    # This dictionary holds the botanical classifications.
+    # It's crucial this data is accurate according to botanical science.
+    # You will need to expand this as needed for your full grocery list.
+    botanical_data = {
+        "sweet potatoes": {
+            "botanical_category": "vegetable",
+            "botanical_part": "root/tuber",
+            "notes": "Edible root of the plant."
+        },
+        "fresh basil": {
+            "botanical_category": "vegetable",
+            "botanical_part": "leaf",
+            "notes": "Edible leaves of the herb."
+        },
+        "plums": {
+            "botanical_category": "fruit",
+            "botanical_part": "matured ovary",
+            "notes": "Simple fleshy fruit (drupe)."
+        },
+        "green beans": {
+            "botanical_category": "fruit",
+            "botanical_part": "matured ovary (legume)",
+            "notes": "Botanically a fruit (legume) containing seeds."
+        },
+        "rice": {
+            "botanical_category": "fruit",
+            "botanical_part": "matured ovary (caryopsis)",
+            "notes": "A grain, which is botanically a dry fruit (caryopsis)."
+        },
+        "corn": {
+            "botanical_category": "fruit",
+            "botanical_part": "matured ovary (caryopsis)",
+            "notes": "A grain, which is botanically a dry fruit (caryopsis)."
+        },
+        "bell pepper": {
+            "botanical_category": "fruit",
+            "botanical_part": "matured ovary",
+            "notes": "Developed from the flower's ovary and contains seeds."
+        },
+        "whole allspice": { # Allspice berries
+            "botanical_category": "fruit",
+            "botanical_part": "dried berry",
+            "notes": "Dried unripe berries of the Pimenta dioica plant."
+        },
+        "acorns": {
+            "botanical_category": "fruit",
+            "botanical_part": "nut (type of dry fruit)",
+            "notes": "A nut, which is botanically a type of dry fruit."
+        },
+        "broccoli": {
+            "botanical_category": "vegetable",
+            "botanical_part": "flower/stem",
+            "notes": "Edible flower heads and stalks."
+        },
+        "celery": {
+            "botanical_category": "vegetable",
+            "botanical_part": "stem/petiole",
+            "notes": "Edible leaf stalks."
+        },
+        "zucchini": {
+            "botanical_category": "fruit",
+            "botanical_part": "matured ovary",
+            "notes": "A type of berry (pepo) from a flowering plant."
+        },
+        "lettuce": {
+            "botanical_category": "vegetable",
+            "botanical_part": "leaf",
+            "notes": "Edible leaves."
+        },
+        "peanuts": {
+            "botanical_category": "fruit",
+            "botanical_part": "legume (matured ovary)",
+            "notes": "Botanically a fruit (legume), despite growing underground."
+        },
+        # Non-plant items or items not strictly fruit/vegetable botanically
+        "milk": {
+            "botanical_category": "other",
+            "botanical_part": "N/A",
+            "notes": "Dairy product (animal)."
+        },
+        "eggs": {
+            "botanical_category": "other",
+            "botanical_part": "N/A",
+            "notes": "Animal product."
+        },
+        "flour": {
+            "botanical_category": "other",
+            "botanical_part": "N/A",
+            "notes": "Processed grain product (typically wheat, which is a fruit)."
+        },
+        "whole bean coffee": {
+            "botanical_category": "other", # The bean itself is a seed, not the entire fruit
+            "botanical_part": "seed",
+            "notes": "The coffee 'bean' is botanically the seed of the coffee cherry (a fruit)."
+        },
+        "oreos": {
+            "botanical_category": "other",
+            "botanical_part": "N/A",
+            "notes": "Processed food item."
+        }
+    }
+
+    # Normalize the input item name for lookup
+    normalized_item = item_name.strip().lower()
+
+    # Handle pluralization/singularization for common cases if not explicitly in data
+    # This is a simple approach; for more robustness, you'd need a proper NLP library.
+    if normalized_item.endswith("s") and normalized_item[:-1] in botanical_data:
+        normalized_item = normalized_item[:-1]
+    elif normalized_item + "s" in botanical_data:
+        # Check if the plural form exists if input is singular
+        if item_name.strip().lower() + "s" in botanical_data:
+            normalized_item = item_name.strip().lower() + "s"
+
+    # Retrieve classification
+    classification = botanical_data.get(normalized_item)
+
+    if classification:
+        return {
+            "item": item_name,
+            "botanical_category": classification["botanical_category"],
+            "botanical_part": classification["botanical_part"],
+            "notes": classification["notes"]
+        }
+    else:
+        # If the item is not found in the database
+        return {
+            "item": item_name,
+            "botanical_category": "unclassified",
+            "botanical_part": "N/A",
+            "notes": "Classification not found in the current database."
+        }
+
 def python_code_runner_tool(task_id:str) -> str:
     """
         Download and run python code, capturing the output.
@@ -325,9 +480,16 @@ python_code_runner_agent = create_react_agent(
     name="python_code_runner_agent"
 )
 
+vegetable_botanical_classification_agent = create_react_agent(
+    model=gemini_llm,
+    tools=[get_botanical_classification_tool],
+    prompt=prompt_botanical_classification,
+    name="vegetable_botanical_classification_agent"
+)
+
 supervisor = create_supervisor(
     model=gemini_llm,
-    agents=[web_research_agent,chess_position_review_agent,python_code_runner_agent],
+    agents=[web_research_agent,chess_position_review_agent,python_code_runner_agent,vegetable_botanical_classification_agent],
     prompt=prompt_recomendado,
     add_handoff_back_messages=True,
     output_mode="full_history"