wbgapi360 0.2.1__py3-none-any.whl → 0.2.8__py3-none-any.whl

wbgapi360/__init__.py

@@ -3,7 +3,8 @@ from .search.engine import SearchEngine
 from .data.builder import DataBuilder
 from .ai.agent import DataAgent
 from .metadata.builder import MetadataBuilder
-from .visual import viz
+from .metadata.builder import MetadataBuilder
+# from .visual import viz # LAZY LOADED
 
 
 class API:
@@ -34,13 +35,22 @@ class API:
 
     @property
     def visual(self):
-        return viz
+        try:
+            from .visual import viz
+            return viz
+        except ImportError as e:
+            if "seaborn" in str(e) or "matplotlib" in str(e):
+                 raise ImportError(
+                    "Optional dependency 'seaborn' not found. "
+                    "Install with: pip install wbgapi360[visual]"
+                ) from e
+            raise e
         
     async def close(self):
         if self._client:
             await self._client.close()
 
-__version__ = "0.2.1"
+__version__ = "0.2.8"
 __author__ = "Maykol Medrano"
 __email__ = "mmedrano2@uc.cl"
 __credits__ = ["Applied Economist Policy Data Scientist"]

wbgapi360/api.py

@@ -10,7 +10,9 @@ from typing import List, Dict, Any, Union, Optional
 from wbgapi360.core.client import Data360Client
 from wbgapi360.search.engine import SearchEngine
 from wbgapi360.data.builder import DataBuilder
-from wbgapi360.visual.charts import Visualizer
+from wbgapi360.data.builder import DataBuilder
+# from wbgapi360.visual.charts import Visualizer # LAZY LOADED
+from wbgapi360.core.utils import normalize_codes, resolve_economies
 from wbgapi360.core.utils import normalize_codes, resolve_economies
 from wbgapi360.core.transformers import DataStandardizer
 from wbgapi360.core.auditor import DataAuditor
@@ -21,7 +23,9 @@ logger.addHandler(logging.NullHandler())
 
 # Singletons for Sync API
 # _client = Data360Client() -> PREVIOUSLY GLOBAL, NOW INSTANTIATED PER CALL TO FIX ASYNCIO LIFECYCLE
-_viz = Visualizer()
+# Singletons for Sync API
+# _client = Data360Client() -> PREVIOUSLY GLOBAL, NOW INSTANTIATED PER CALL TO FIX ASYNCIO LIFECYCLE
+# _viz = Visualizer() -> NOW LAZY LOADED
 
 def _run_sync(coro):
     """
@@ -43,6 +47,8 @@ def _run_sync(coro):
 
 # --- INTERNAL ASYNC LOGIC (Replicated from server to decouple) ---
 
+# --- INTERNAL ASYNC LOGIC (Replicated from server to decouple) ---
+
 async def _async_get_data(
     indicator: Union[str, List[str]], 
     economies: Union[str, List[str]], 
@@ -192,6 +198,23 @@ def get_data(
         
     return df
 
+# Helper for lazy loading visualization
+_viz_instance = None
+def _get_viz():
+    global _viz_instance
+    if _viz_instance is None:
+        try:
+            from wbgapi360.visual.charts import Visualizer
+            _viz_instance = Visualizer()
+        except ImportError as e:
+            if "seaborn" in str(e) or "matplotlib" in str(e):
+                raise ImportError(
+                    "Optional dependency 'seaborn' or 'matplotlib' not found. "
+                    "Please install with: pip install wbgapi360[visual]"
+                ) from e
+            raise e
+    return _viz_instance
+
 def plot(chart_type: str, data: Union[str, pd.DataFrame], title: str = "", subtitle: str = "", **kwargs) -> str:
     """
     Generate Financial Times-style chart with editorial aesthetics.
@@ -209,26 +232,29 @@ def plot(chart_type: str, data: Union[str, pd.DataFrame], title: str = "", subti
     """
     logger.info(f"Plotting chart type: {chart_type}")
     
+    # Lazy load visualizer
+    viz = _get_viz()
+    
     # Logic adapted for local dispatch
     dispatch_table = {
-        'trend': _viz.plot_trend,
-        'line': _viz.plot_trend,
-        'bar': _viz.plot_bar,
-        'column': _viz.plot_column,
-        'scatter': _viz.plot_scatter,
-        'map': _viz.plot_map,
-        'map_bubble': _viz.plot_map_bubble,
-        'map_diverging': _viz.plot_map_diverging,
-        'map_categorical': _viz.plot_map_categorical,
-        'dumbbell': _viz.plot_dumbbell,
-        'stacked': _viz.plot_stacked_bar,
-        'stacked_bar': _viz.plot_stacked_bar,
-        'area': _viz.plot_area,
-        'heatmap': _viz.plot_heatmap,
-        'bump': _viz.plot_bump,
-        'treemap': _viz.plot_treemap,
-        'donut': _viz.plot_donut,
-        'pie': _viz.plot_donut
+        'trend': viz.plot_trend,
+        'line': viz.plot_trend,
+        'bar': viz.plot_bar,
+        'column': viz.plot_column,
+        'scatter': viz.plot_scatter,
+        'map': viz.plot_map,
+        'map_bubble': viz.plot_map_bubble,
+        'map_diverging': viz.plot_map_diverging,
+        'map_categorical': viz.plot_map_categorical,
+        'dumbbell': viz.plot_dumbbell,
+        'stacked': viz.plot_stacked_bar,
+        'stacked_bar': viz.plot_stacked_bar,
+        'area': viz.plot_area,
+        'heatmap': viz.plot_heatmap,
+        'bump': viz.plot_bump,
+        'treemap': viz.plot_treemap,
+        'donut': viz.plot_donut,
+        'pie': viz.plot_donut
     }
     
     func = dispatch_table.get(chart_type)
@@ -252,6 +278,13 @@ def plot(chart_type: str, data: Union[str, pd.DataFrame], title: str = "", subti
     # Execute the plot function with all kwargs
     func(data, title=title, subtitle=subtitle, save_path=path, **kwargs)
     
+    # Auto-display in Notebooks (Colab/Jupyter)
+    try:
+        from IPython.display import Image, display
+        display(Image(filename=path))
+    except (ImportError, Exception):
+        pass
+
     return path
 
 

wbgapi360/mcp/server.py

@@ -14,7 +14,8 @@ from fastmcp import FastMCP
 from wbgapi360.core.client import Data360Client
 from wbgapi360.data.builder import DataBuilder
 from wbgapi360.search.engine import SearchEngine
-from wbgapi360.visual.charts import Visualizer
+from wbgapi360.search.engine import SearchEngine
+# from wbgapi360.visual.charts import Visualizer # LAZY LOADED
 
 # Setup logger
 logger = logging.getLogger("wbgapi360.server")
@@ -25,7 +26,9 @@ mcp = FastMCP("wbgapi360-agent")
 _client_instance: Optional[Data360Client] = None
 _search_engine_instance: Optional[SearchEngine] = None
 
-viz = Visualizer()  # Single instance for visualization is safe as it's synchronous
+_search_engine_instance: Optional[SearchEngine] = None
+
+# viz = Visualizer()  # LAZY LOADED
 
 def _get_client() -> Data360Client:
     """Lazy initialize the Data360Client."""
@@ -42,6 +45,23 @@ def _get_search_engine() -> SearchEngine:
         _search_engine_instance = SearchEngine(client)
     return _search_engine_instance
 
+# Helper for lazy loading visualization
+_viz_instance = None
+def _get_viz():
+    global _viz_instance
+    if _viz_instance is None:
+        try:
+            from wbgapi360.visual.charts import Visualizer
+            _viz_instance = Visualizer()
+        except ImportError as e:
+            if "seaborn" in str(e) or "matplotlib" in str(e):
+                raise ImportError(
+                    "Optional dependency 'seaborn' not found. "
+                    "Please install with: pip install wbgapi360[visual]"
+                ) from e
+            raise e
+    return _viz_instance
+
 # --- RAW IMPLEMENTATIONS (Core Functions) ---
 
 async def _search_indicators(query: str, limit: int = 10, database_id: str = "WB_WDI") -> List[Dict[str, Any]]:
@@ -387,6 +407,7 @@ def _plot_chart(chart_type: str, data: Union[str, pd.DataFrame], title: str = "C
         os.close(fd)
 
         # 3. Dispatcher Logic (Clean Architecture)
+        viz = _get_viz()
         dispatch_table = {
             'trend': viz.plot_trend,
             'line': viz.plot_trend,

{wbgapi360-0.2.1.dist-info → wbgapi360-0.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbgapi360
-Version: 0.2.1
+Version: 0.2.8
 Summary: Enterprise-grade Async Python Client for World Bank Data360 API
 Author: Maykol Medrano
 Author-email: Maykol Medrano <mmedrano2@uc.cl>
@@ -35,7 +35,7 @@ Dynamic: requires-python
 [![PyPI version](https://img.shields.io/badge/pypi-v0.2.1-0052CC.svg)](https://pypi.org/project/wbgapi360/)
 [![Python](https://img.shields.io/badge/python-3.9%2B-blue)](https://www.python.org/downloads/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-grey.svg)](https://opensource.org/licenses/MIT)
-[![Build Status](https://github.com/mmedrano2/mcp_wbgapi360/actions/workflows/ci.yml/badge.svg)](https://github.com/mmedrano2/mcp_wbgapi360/actions)
+[![Build Status](https://github.com/MaykolMedrano/mcp_wbgapi360/actions/workflows/ci.yml/badge.svg)](https://github.com/MaykolMedrano/mcp_wbgapi360/actions)
 [![Status](https://img.shields.io/badge/status-enterprise%20ready-green)]()
 
 **wbgapi360** is a high-performance Python client designed to interact with the World Bank's Data APIs. It bridges the gap between traditional econometric analysis and modern Artificial Intelligence workflows.
@@ -62,18 +62,22 @@ We acknowledge and thank Tim for establishing the developer-friendly patterns th
 
 ## Installation
 
-The library is available on PyPI.
+Install using pip:
 
 ```bash
-# Core installation (Lightweight)
 pip install wbgapi360
+```
 
+**For Visualization Features (Plots & Maps):**
+The visualization module is optional to keep the library lightweight. To use `wb.plot()`, install with extras:
+```bash
+pip install "wbgapi360[visual]"
+```
 # With Visualization support (adds Matplotlib/Seaborn)
 pip install wbgapi360[visual]
 
 # With Mapping support (adds GeoPandas - Heavy dependency)
 pip install wbgapi360[map]
-```
 
 ---
 
@@ -92,29 +96,36 @@ results = wb.search("inflation") # Returns top 10 matches
 
 for r in results[:3]:
     print(f"[{r['code']}] {r['name']}")
-    # Output:
-    # [FP.CPI.TOTL.ZG] Inflation, consumer prices (annual %)
-    # [FP.CPI.TOTL] Consumer price index (2010 = 100)
-    # [NY.GDP.DEFL.KD.ZG] Inflation, GDP deflator (annual %)
-```
-
-### 2. Robust Data Retrieval
-Fetch data with automatic ID correction and intelligent time window handling.
-
-```python
-# Fetch GDP Growth for USA, China, and Peru (Last 10 years)
+# 1. Fetch GDP Growth (Last 10 years)
 df = wb.get_data(
-    indicator="NY.GDP.MKTP.KD.ZG", 
+    indicator="NY.GDP.MKTP.KD.ZG",
     economies=["USA", "CHN", "PER"],
-    years=10,
-    labels=True,      # Adds 'Country Name' column
-    as_frame=True     # Returns pandas DataFrame instead of JSON
+    years=10
 )
-
 print(df.head())
+
+# 2. Fetch FDI Data (Last 20 years)
+fdi_df = wb.get_data(
+    indicator="BX.KLT.DINV.CD.WD",
+    economies=["CHL", "USA", "CHN"],
+    years=20
+)
+print("Foreign Direct Investment Data:")
+print(fdi_df.head())
+
+# 3. Plot Trend (Financial Times Style)
+# Requires: pip install wbgapi360[visual]
+data = wb.get_data("NY.GDP.PCAP.CD", ["CHL", "PER", "COL", "MEX"], years=15)
+
+wb.plot(
+    chart_type="trend",
+    data=data,
+    title="GDP Per Capita Trend",
+    subtitle="USD Current"
+)
 ```
 
-### 3. Advanced Analysis: Trends & Statistics
+### 4. Advanced Analysis: Trends & Statistics
 Quickly assess the economic trajectory of a country without writing boilerplate pandas code.
 
 ```python

{wbgapi360-0.2.1.dist-info → wbgapi360-0.2.8.dist-info}/RECORD

@@ -1,5 +1,5 @@
-wbgapi360/__init__.py,sha256=7BEkh1I7oGv0Afx9y1Bgno-k0g_FO6CwUU2K0HAwpco,1224
-wbgapi360/api.py,sha256=TxgFoxwaLTlJugIH0iNFdHVJWgseLb3SRYrapTB-dIA,9508
+wbgapi360/__init__.py,sha256=wK5c-PxlT9xw3ytZ9AJDAh4YAMzduFJRjFhDym8uwro,1655
+wbgapi360/api.py,sha256=l0V10hzCI2tZT1_RXCDBLV1lUbMhEus2V9mye3Lz9TM,10720
 wbgapi360/cli.py,sha256=DxToczs3yUsSt8vANnMTmB2A3TwAjWBZTY5VJF50-Co,3117
 wbgapi360/config.py,sha256=iOzQgBcBQ-dZWpP1eVgMeFUw3CPtzoxnsDQ7ySYHjB8,1693
 wbgapi360/ai/agent.py,sha256=KXZky-VmKTRIqrPe_RYsNUnpUk9sw5KX0GY2nRWcTlU,5267
@@ -9,16 +9,16 @@ wbgapi360/core/models.py,sha256=awiKgADvyCI595-ietE-kOg82PwwsmHZMinR7qn7Nvk,742
 wbgapi360/core/transformers.py,sha256=zrey9b7S-XHgjOFjK2p-Jjkrw1ijrfX4ypbfZHdpFYw,2931
 wbgapi360/core/utils.py,sha256=TCQDEcHg_alqA5fFqQlVA6NttatLNyyFKsqv744c-T0,1316
 wbgapi360/data/builder.py,sha256=2Jlua89FKxtjiYE0I1h5iviDTvrMb5vElvOEBeN48vM,7519
-wbgapi360/mcp/server.py,sha256=tAradOdhU5iUavu0eQpRMmYTUByWzikfWdxx6_K7WKY,21271
+wbgapi360/mcp/server.py,sha256=TL87vECObaeIzTp_eJy1H1KC2rIjDWu1o9sIM8tFKhI,21973
 wbgapi360/metadata/builder.py,sha256=8L05WshpcbmzZFHeGH38pZKt36XsB4-VFMn63Qbrtg8,3149
 wbgapi360/metadata/iso_mapping.py,sha256=55psZTI53_nMsFmvnN21szvO_gTh6noki5waKr8H5NU,5674
 wbgapi360/metadata/resolver.py,sha256=YNZOcbCnlJ3EJ-dbgTIUitLgSwcmLV3Oz63OaSX1Ozk,4957
 wbgapi360/search/engine.py,sha256=zTS1We-07-La1DjEkMLeKbIrNbeqbiV6fdZQpzGnOT8,5970
 wbgapi360/visual/__init__.py,sha256=94LZyoNR1OSWdItQ1CwgeULnwmMN6HSJNHf0oom2970,25
 wbgapi360/visual/charts.py,sha256=hZSUDbGnRe95LwQUHEERhSqAvvRrIeO6kS2T_YAkyZo,46773
-wbgapi360-0.2.1.dist-info/licenses/LICENSE,sha256=A260AC9u7jC7SfQOJ21JOQYqsIJUtgC-6iM2dm3BsNw,1092
-wbgapi360-0.2.1.dist-info/METADATA,sha256=G-hG-_xog8MVxAHHJoPAbnmD_YOlX1CsyrnUZGeNtko,8604
-wbgapi360-0.2.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-wbgapi360-0.2.1.dist-info/entry_points.txt,sha256=9nNopsIlKEKw9rNe_6zW-ERL6fWu6eXymmKLBIZzzqI,49
-wbgapi360-0.2.1.dist-info/top_level.txt,sha256=q3vycsvqQsLZqAFyCzrPw392QV6wAYbVamr39Z-n2MQ,10
-wbgapi360-0.2.1.dist-info/RECORD,,
+wbgapi360-0.2.8.dist-info/licenses/LICENSE,sha256=A260AC9u7jC7SfQOJ21JOQYqsIJUtgC-6iM2dm3BsNw,1092
+wbgapi360-0.2.8.dist-info/METADATA,sha256=iDmB3t2ofOaILceLauQ3Y4Zji6S_iMQa26WgpeT1Tts,8802
+wbgapi360-0.2.8.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+wbgapi360-0.2.8.dist-info/entry_points.txt,sha256=9nNopsIlKEKw9rNe_6zW-ERL6fWu6eXymmKLBIZzzqI,49
+wbgapi360-0.2.8.dist-info/top_level.txt,sha256=q3vycsvqQsLZqAFyCzrPw392QV6wAYbVamr39Z-n2MQ,10
+wbgapi360-0.2.8.dist-info/RECORD,,