Update app.py

app.py

@@ -14,12 +14,102 @@ import pytz
 import yaml
 import os
 
-from tools.final_answer import FinalAnswerTool  # Adjust import if you store final_answer differently
+from tools.final_answer import FinalAnswerTool  # Adjust import path if needed
 
 ########################################
-# TOOLS
+# UTILITY FUNCTION:
+# Restructure Europeana JSON response
 ########################################
+def restructure_europeana_response(europeana_json: dict) -> str:
+    """
+    Takes a Europeana Search API JSON response and returns a
+    simplified string summary, focusing on child-friendly information.
+    """
+    # Check if Europe's response is successful
+    if not europeana_json.get("success"):
+        return f"Oops! There was a problem with Europeana: {europeana_json.get('error', 'Unknown error')}"
+
+    items = europeana_json.get("items", [])
+    if not items:
+        return "Sorry, I couldn’t find anything on Europeana for that topic!"
+
+    # Build a short list of results
+    result_lines = []
+    for idx, item in enumerate(items, start=1):
+        # Try to get a title
+        title_list = item.get("title") or []
+        if not title_list:
+            # fallback: check dcTitleLangAware if "title" is missing
+            dc_title_lang = item.get("dcTitleLangAware", {})
+            if dc_title_lang:
+                first_lang = next(iter(dc_title_lang))
+                title_list = dc_title_lang[first_lang]
+        title_str = title_list[0] if title_list else "[No title found]"
+
+        # Provider (museum/institution)
+        provider_list = item.get("dataProvider") or []
+        provider_str = provider_list[0] if provider_list else "Unknown provider"
+
+        # Object type (IMAGE, VIDEO, TEXT, SOUND, etc.)
+        obj_type = item.get("type") or "Unknown type"
+
+        # A short description
+        desc_list = item.get("dcDescription") or []
+        desc_str = desc_list[0] if desc_list else "No description available."
+
+        # Year (if present)
+        year_list = item.get("year") or []
+        year_str = year_list[0] if year_list else "N/A"
+
+        # Construct a child-friendly summary
+        # Feel free to reword for an even more "kid-friendly" vibe
+        summary_text = (
+            f"{idx}) **Title**: {title_str}\n"
+            f"   **Where it’s from**: {provider_str}\n"
+            f"   **Type of item**: {obj_type}\n"
+            f"   **Approx. Year**: {year_str}\n"
+            f"   **Fun Fact/Description**: {desc_str}\n"
+        )
+        result_lines.append(summary_text)
+
+    # Combine the lines into a single string
+    intro = "Here are some cool things I found in Europeana:\n"
+    return intro + "\n".join(result_lines)
+
+########################################
+# EUROPEANA TOOL
+########################################
+EUROPEANA_API_KEY = "YOUR_EUROPEANA_API_KEY"  # <-- Replace with your real key!
+
+@tool
+def query_europeana(query: str) -> str:
+    """
+    A tool that queries the Europeana Search API for a given query
+    and returns up to 5 results in a kid-friendly summary.
+    """
+    endpoint = "https://api.europeana.eu/record/v2/search.json"
+    params = {
+        "query": query,
+        "wskey": EUROPEANA_API_KEY,
+        "rows": 5,
+    }
+    try:
+        response = requests.get(endpoint, params=params)
+        data = response.json()
+        
+        if response.status_code != 200:
+            return f"Oops, something went wrong: {data.get('error', 'Unknown HTTP error')}"
+
+        # Use our restructure function for a nice summary
+        return restructure_europeana_response(data)
 
+    except Exception as e:
+        return f"Error calling Europeana API: {str(e)}"
+
+
+########################################
+# TIME TOOL
+########################################
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """
@@ -35,6 +125,9 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+########################################
+# WIKIPEDIA CULTURAL INFO
+########################################
 @tool
 def get_cultural_info(topic: str) -> str:
     """
@@ -45,7 +138,6 @@ def get_cultural_info(topic: str) -> str:
         A short summary text from Wikipedia for the specified topic.
     """
     try:
-        # Query the Wikipedia API for an extract
         url = (
             "https://en.wikipedia.org/w/api.php?"
             "action=query&prop=extracts&exintro&explaintext&format=json&titles=" + topic
@@ -55,9 +147,8 @@ def get_cultural_info(topic: str) -> str:
 
         pages = data.get("query", {}).get("pages", {})
         if not pages:
-            return f"No Wikipedia pages found for topic: {topic}"
+            return f"I couldn't find anything on Wikipedia for '{topic}'."
 
-        # The keys in 'pages' are page IDs
         page_id = next(iter(pages))
         page_content = pages[page_id]
         if "missing" in page_content:
@@ -78,52 +169,45 @@ def get_cultural_info(topic: str) -> str:
 ########################################
 final_answer = FinalAnswerTool()
 
-
 ########################################
-# MODEL (choose whichever model/endpoint you prefer)
+# MODEL 
 ########################################
 model = HfApiModel(
     max_tokens=1024,
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # or any other model ID / HF endpoint
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # or your chosen HF endpoint
     custom_role_conversions=None
 )
 
 ########################################
-# LOAD ANY OPTIONAL TOOLS FROM HUB (e.g. text-to-image)
+# OPTIONAL: IMAGE GENERATION TOOL
 ########################################
-# NOTE: If you want to see if it works, you can skip or keep it
 try:
     image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 except Exception as e:
     print("Error loading text-to-image tool:", e)
     image_generation_tool = None
 
-
 ########################################
-# PROMPT TEMPLATES
+# PROMPT TEMPLATES (for kids style)
 ########################################
-# We read from prompts.yaml. Make sure you have a prompts.yaml in the same folder.
-# It can contain specialized instructions for your "kid-friendly" museum agent.
 this_dir = os.path.dirname(os.path.abspath(__file__))
 prompts_path = os.path.join(this_dir, "prompts.yaml")
-
 with open(prompts_path, 'r', encoding='utf-8') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-
 ########################################
 # BUILD THE AGENT
 ########################################
 tools_list = [
-    final_answer,           # Important: must remain in the tools
+    final_answer,  
     DuckDuckGoSearchTool(),
     VisitWebpageTool(),
     get_current_time_in_timezone,
-    get_cultural_info
+    get_cultural_info,
+    query_europeana,  # <--- Our new Europeana tool
 ]
 
-# If the image_generation_tool loaded successfully, add it:
 if image_generation_tool:
     tools_list.append(image_generation_tool)
 
@@ -139,10 +223,7 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
-########################################
-# (OPTIONAL) GRADIO UI LAUNCH
-########################################
 if __name__ == "__main__":
+    # (OPTIONAL) Launch Gradio Chat UI
     from Gradio_UI import GradioUI
     GradioUI(agent).launch()