Upload tools/pokemon_info.py with huggingface_hub

tools/pokemon_info.py

@@ -0,0 +1,73 @@
+import gradio as gr
+import json
+from PIL import Image
+import numpy as np
+
+# --- User Defined Logic ---
+import requests
+from typing import Dict, Any
+
+def pokemon_info(name: str) -> Dict[str, Any]:
+    """Fetch detailed information about a Pokémon from the PokeAPI.
+    
+    Given a Pokémon name, returns comprehensive data including types, abilities,
+    base stats, height, weight, and sprite images.
+    
+    Args:
+        name: The name of the Pokémon to look up (case-insensitive).
+    
+    Returns:
+        A dictionary containing:
+            - name: The Pokémon's name
+            - id: National Pokédex number
+            - types: List of type names
+            - abilities: List of ability names
+            - stats: Dictionary mapping stat names to base values
+            - height: Height in decimetres
+            - weight: Weight in hectograms
+            - sprites: Dictionary of sprite image URLs
+            - base_experience: Base experience yield
+            - order: Order for sorting
+    
+    Raises:
+        ValueError: If the Pokémon is not found or the API request fails.
+    """
+    url = f"https://pokeapi.co/api/v2/pokemon/{name.lower()}"
+    try:
+        response = requests.get(url, timeout=10)
+        response.raise_for_status()
+        data = response.json()
+    except requests.exceptions.HTTPError as e:
+        if e.response.status_code == 404:
+            raise ValueError(f"Pokémon '{name}' not found.")
+        raise ValueError(f"API request failed: {e}")
+    except requests.exceptions.RequestException as e:
+        raise ValueError(f"Network error: {e}")
+    
+    types = [t["type"]["name"] for t in data["types"]]
+    abilities = [a["ability"]["name"] for a in data["abilities"]]
+    stats = {s["stat"]["name"]: s["base_stat"] for s in data["stats"]}
+    sprites = {k: v for k, v in data["sprites"].items() if isinstance(v, str)}
+    
+    return {
+        "name": data["name"],
+        "id": data["id"],
+        "types": types,
+        "abilities": abilities,
+        "stats": stats,
+        "height": data["height"],
+        "weight": data["weight"],
+        "sprites": sprites,
+        "base_experience": data.get("base_experience"),
+        "order": data["order"]
+    }
+
+# --- Interface Factory ---
+def create_interface():
+    return gr.Interface(
+        fn=pokemon_info,
+        inputs=[gr.Textbox(label=k) for k in ['name']],
+        outputs=gr.JSON(label="Detailed Pokémon information including types, abilities, stats, height, weight, and sprite URLs"),
+        title="pokemon-info",
+        description="Auto-generated tool: pokemon-info"
+    )