geomind-ai 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

geomind/tools/stac_search.py

@@ -1,231 +1,234 @@
-"""
-STAC API search tools for querying Sentinel-2 imagery.
-
-Uses the EOPF STAC API at https://stac.core.eopf.eodc.eu
-"""
-
-from typing import Optional, List
-from datetime import datetime, timedelta
-from pystac_client import Client
-
-from ..config import (
-    STAC_API_URL,
-    STAC_COLLECTION,
-    DEFAULT_MAX_CLOUD_COVER,
-    DEFAULT_MAX_ITEMS,
-)
-
-
-def _get_stac_client() -> Client:
-    """Get a STAC API client instance."""
-    return Client.open(STAC_API_URL)
-
-
-def _format_item(item) -> dict:
-    """Format a STAC item into a simplified dictionary."""
-    props = item.properties
-
-    return {
-        "id": item.id,
-        "datetime": props.get("datetime"),
-        "cloud_cover": props.get("eo:cloud_cover"),
-        "platform": props.get("platform"),
-        "bbox": item.bbox,
-        "geometry": item.geometry,
-        "assets": {
-            key: {
-                "title": asset.title,
-                "href": asset.href,
-                "type": asset.media_type,
-            }
-            for key, asset in item.assets.items()
-            if key in ["SR_10m", "SR_20m", "SR_60m", "TCI_10m", "product"]
-        },
-        "stac_url": f"{STAC_API_URL}/collections/{STAC_COLLECTION}/items/{item.id}",
-    }
-
-
-def search_imagery(
-    bbox: Optional[List[float]] = None,
-    start_date: Optional[str] = None,
-    end_date: Optional[str] = None,
-    max_cloud_cover: Optional[float] = None,
-    max_items: Optional[int] = None,
-) -> dict:
-    """
-    Search for Sentinel-2 L2A imagery in the EOPF STAC catalog.
-
-    Args:
-        bbox: Bounding box [min_lon, min_lat, max_lon, max_lat]
-        start_date: Start date in YYYY-MM-DD format
-        end_date: End date in YYYY-MM-DD format
-        max_cloud_cover: Maximum cloud cover percentage (0-100)
-        max_items: Maximum number of items to return
-
-    Returns:
-        Dictionary with search results including items found
-
-    Example:
-        >>> search_imagery(
-        ...     bbox=[-74.0, 40.7, -73.9, 40.8],
-        ...     start_date="2024-12-01",
-        ...     end_date="2024-12-20",
-        ...     max_cloud_cover=20
-        ... )
-    """
-    if max_cloud_cover is None:
-        max_cloud_cover = DEFAULT_MAX_CLOUD_COVER
-    if max_items is None:
-        max_items = DEFAULT_MAX_ITEMS
-
-    # Build datetime string
-    datetime_str = None
-    if start_date or end_date:
-        start = start_date or "2015-01-01"
-        end = end_date or datetime.now().strftime("%Y-%m-%d")
-        datetime_str = f"{start}/{end}"
-
-    try:
-        client = _get_stac_client()
-
-        # Build search parameters
-        search_params = {
-            "collections": [STAC_COLLECTION],
-            "max_items": max_items,
-        }
-
-        if bbox:
-            search_params["bbox"] = bbox
-
-        if datetime_str:
-            search_params["datetime"] = datetime_str
-
-        # Execute search
-        search = client.search(**search_params)
-        items = list(search.items())
-
-        # Filter by cloud cover (post-filter since API may not support query param)
-        filtered_items = [
-            item
-            for item in items
-            if item.properties.get("eo:cloud_cover", 100) <= max_cloud_cover
-        ]
-
-        # Sort by date (newest first)
-        filtered_items.sort(
-            key=lambda x: x.properties.get("datetime", ""), reverse=True
-        )
-
-        # Format results
-        formatted_items = [_format_item(item) for item in filtered_items]
-
-        return {
-            "success": True,
-            "total_found": len(items),
-            "filtered_count": len(filtered_items),
-            "items": formatted_items,
-            "search_params": {
-                "bbox": bbox,
-                "datetime": datetime_str,
-                "max_cloud_cover": max_cloud_cover,
-            },
-        }
-
-    except Exception as e:
-        return {
-            "success": False,
-            "error": str(e),
-            "items": [],
-        }
-
-
-def get_item_details(item_id: str) -> dict:
-    """
-    Get detailed information about a specific STAC item.
-
-    Args:
-        item_id: The STAC item ID (e.g., "S2B_MSIL2A_20251218T110359_...")
-
-    Returns:
-        Dictionary with full item details including all assets
-    """
-    try:
-        # Get the item
-        item_url = f"{STAC_API_URL}/collections/{STAC_COLLECTION}/items/{item_id}"
-
-        import requests
-
-        response = requests.get(item_url)
-        response.raise_for_status()
-        item_data = response.json()
-
-        return {
-            "success": True,
-            "item": item_data,
-        }
-
-    except Exception as e:
-        return {
-            "success": False,
-            "error": str(e),
-        }
-
-
-def list_recent_imagery(
-    location_name: Optional[str] = None,
-    days: int = 7,
-    max_cloud_cover: Optional[float] = None,
-    max_items: Optional[int] = None,
-) -> dict:
-    """
-    List recent Sentinel-2 imagery, optionally for a specific location.
-
-    This is a convenience function that combines geocoding and search.
-
-    Args:
-        location_name: Optional place name to search around
-        days: Number of days to look back (default: 7)
-        max_cloud_cover: Maximum cloud cover percentage
-        max_items: Maximum items to return
-
-    Returns:
-        Dictionary with recent imagery items
-    """
-    from .geocoding import get_bbox_from_location
-
-    # Calculate date range
-    end_date = datetime.now()
-    start_date = end_date - timedelta(days=days)
-
-    # Get bbox if location provided
-    bbox = None
-    location_info = None
-
-    if location_name:
-        bbox_result = get_bbox_from_location(location_name)
-        if bbox_result["success"]:
-            bbox = bbox_result["bbox"]
-            location_info = {
-                "name": location_name,
-                "center": bbox_result["center"],
-                "address": bbox_result["address"],
-            }
-        else:
-            return {
-                "success": False,
-                "error": f"Could not geocode location: {location_name}",
-            }
-
-    # Search for imagery
-    result = search_imagery(
-        bbox=bbox,
-        start_date=start_date.strftime("%Y-%m-%d"),
-        end_date=end_date.strftime("%Y-%m-%d"),
-        max_cloud_cover=max_cloud_cover,
-        max_items=max_items,
-    )
-
-    if location_info:
-        result["location"] = location_info
-
-    return result
+"""
+STAC API search tools for querying Sentinel-2 imagery.
+
+Uses the EOPF STAC API at https://stac.core.eopf.eodc.eu
+"""
+
+from typing import Optional, List
+from datetime import datetime, timedelta
+from pystac_client import Client
+
+from ..config import (
+    STAC_API_URL,
+    STAC_COLLECTION,
+    DEFAULT_MAX_CLOUD_COVER,
+    DEFAULT_MAX_ITEMS,
+)
+
+
+def _get_stac_client() -> Client:
+    """Get a STAC API client instance."""
+    return Client.open(STAC_API_URL)
+
+
+def _format_item(item) -> dict:
+    """Format a STAC item into a simplified dictionary."""
+    props = item.properties
+
+    return {
+        "id": item.id,
+        "datetime": props.get("datetime"),
+        "cloud_cover": props.get("eo:cloud_cover"),
+        "platform": props.get("platform"),
+        "bbox": item.bbox,
+        "geometry": item.geometry,
+        "assets": {
+            key: {
+                "title": asset.title,
+                "href": asset.href,
+                "type": asset.media_type,
+            }
+            for key, asset in item.assets.items()
+            if key in ["SR_10m", "SR_20m", "SR_60m", "TCI_10m", "product"]
+        },
+        "stac_url": f"{STAC_API_URL}/collections/{STAC_COLLECTION}/items/{item.id}",
+    }
+
+
+def search_imagery(
+    bbox: Optional[List[float]] = None,
+    start_date: Optional[str] = None,
+    end_date: Optional[str] = None,
+    max_cloud_cover: Optional[float] = None,
+    max_items: Optional[int] = None,
+) -> dict:
+    """
+    Search for Sentinel-2 L2A imagery in the EOPF STAC catalog.
+
+    Args:
+        bbox: Bounding box [min_lon, min_lat, max_lon, max_lat]
+        start_date: Start date in YYYY-MM-DD format
+        end_date: End date in YYYY-MM-DD format
+        max_cloud_cover: Maximum cloud cover percentage (0-100)
+        max_items: Maximum number of items to return
+
+    Returns:
+        Dictionary with search results including items found
+
+    Example:
+        >>> search_imagery(
+        ...     bbox=[-74.0, 40.7, -73.9, 40.8],
+        ...     start_date="2024-12-01",
+        ...     end_date="2024-12-20",
+        ...     max_cloud_cover=20
+        ... )
+    """
+    if max_cloud_cover is None:
+        max_cloud_cover = DEFAULT_MAX_CLOUD_COVER
+    if max_items is None:
+        max_items = DEFAULT_MAX_ITEMS
+
+    # Build datetime string
+    datetime_str = None
+    if start_date or end_date:
+        start = start_date or "2015-01-01"
+        end = end_date or datetime.now().strftime("%Y-%m-%d")
+        datetime_str = f"{start}/{end}"
+
+    try:
+        client = _get_stac_client()
+
+        # Build search parameters
+        search_params = {
+            "collections": [STAC_COLLECTION],
+            "max_items": max_items,
+        }
+
+        if bbox:
+            search_params["bbox"] = bbox
+
+        if datetime_str:
+            search_params["datetime"] = datetime_str
+
+        # Execute search
+        search = client.search(**search_params)
+        items = list(search.items())
+
+        # Filter by cloud cover (post-filter since API may not support query param)
+        filtered_items = [
+            item
+            for item in items
+            if item.properties.get("eo:cloud_cover", 100) <= max_cloud_cover
+        ]
+
+        # Sort by date (newest first)
+        filtered_items.sort(
+            key=lambda x: x.properties.get("datetime", ""), reverse=True
+        )
+
+        # Format results
+        formatted_items = [_format_item(item) for item in filtered_items]
+
+        return {
+            "success": True,
+            "total_found": len(items),
+            "filtered_count": len(filtered_items),
+            "items": formatted_items,
+            "search_params": {
+                "bbox": bbox,
+                "datetime": datetime_str,
+                "max_cloud_cover": max_cloud_cover,
+            },
+        }
+
+    except Exception as e:
+        return {
+            "success": False,
+            "error": str(e),
+            "items": [],
+        }
+
+
+def get_item_details(item_id: str) -> dict:
+    """
+    Get detailed information about a specific STAC item.
+
+    Args:
+        item_id: The STAC item ID (e.g., "S2B_MSIL2A_20251218T110359_...")
+
+    Returns:
+        Dictionary with full item details including all assets
+    """
+    try:
+        client = _get_stac_client()
+        collection = client.get_collection(STAC_COLLECTION)
+
+        # Get the item
+        item_url = f"{STAC_API_URL}/collections/{STAC_COLLECTION}/items/{item_id}"
+
+        import requests
+
+        response = requests.get(item_url)
+        response.raise_for_status()
+        item_data = response.json()
+
+        return {
+            "success": True,
+            "item": item_data,
+        }
+
+    except Exception as e:
+        return {
+            "success": False,
+            "error": str(e),
+        }
+
+
+def list_recent_imagery(
+    location_name: Optional[str] = None,
+    days: int = 7,
+    max_cloud_cover: Optional[float] = None,
+    max_items: Optional[int] = None,
+) -> dict:
+    """
+    List recent Sentinel-2 imagery, optionally for a specific location.
+
+    This is a convenience function that combines geocoding and search.
+
+    Args:
+        location_name: Optional place name to search around
+        days: Number of days to look back (default: 7)
+        max_cloud_cover: Maximum cloud cover percentage
+        max_items: Maximum items to return
+
+    Returns:
+        Dictionary with recent imagery items
+    """
+    from .geocoding import get_bbox_from_location
+
+    # Calculate date range
+    end_date = datetime.now()
+    start_date = end_date - timedelta(days=days)
+
+    # Get bbox if location provided
+    bbox = None
+    location_info = None
+
+    if location_name:
+        bbox_result = get_bbox_from_location(location_name)
+        if bbox_result["success"]:
+            bbox = bbox_result["bbox"]
+            location_info = {
+                "name": location_name,
+                "center": bbox_result["center"],
+                "address": bbox_result["address"],
+            }
+        else:
+            return {
+                "success": False,
+                "error": f"Could not geocode location: {location_name}",
+            }
+
+    # Search for imagery
+    result = search_imagery(
+        bbox=bbox,
+        start_date=start_date.strftime("%Y-%m-%d"),
+        end_date=end_date.strftime("%Y-%m-%d"),
+        max_cloud_cover=max_cloud_cover,
+        max_items=max_items,
+    )
+
+    if location_info:
+        result["location"] = location_info
+
+    return result

{geomind_ai-1.0.0.dist-info → geomind_ai-1.0.2.dist-info}/METADATA

@@ -1,85 +1,78 @@
-Metadata-Version: 2.4
-Name: geomind-ai
-Version: 1.0.0
-Summary: AI agent for geospatial analysis
-Author: Harsh Shinde
-License-Expression: MIT
-Project-URL: Homepage, https://harshshinde0.github.io/GeoMind/
-Project-URL: Repository, https://github.com/HarshShinde0/GeoMind
-Project-URL: Documentation, https://github.com/HarshShinde0/GeoMind#readme
-Project-URL: Issues, https://github.com/HarshShinde0/GeoMind/issues
-Keywords: geospatial,satellite-imagery,sentinel-2,ai-agent,remote-sensing,earth-observation
-Classifier: Development Status :: 4 - Beta
-Classifier: Intended Audience :: Science/Research
-Classifier: Intended Audience :: Developers
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Topic :: Scientific/Engineering
-Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: openai>=1.0.0
-Requires-Dist: pystac-client>=0.8.0
-Requires-Dist: pystac>=1.10.0
-Requires-Dist: xarray>=2024.1.0
-Requires-Dist: zarr>=2.18.0
-Requires-Dist: dask>=2024.1.0
-Requires-Dist: geopy>=2.4.0
-Requires-Dist: fsspec>=2024.1.0
-Requires-Dist: aiohttp>=3.9.0
-Requires-Dist: requests>=2.31.0
-Requires-Dist: s3fs>=2024.1.0
-Requires-Dist: matplotlib>=3.8.0
-Requires-Dist: numpy>=1.26.0
-Requires-Dist: python-dotenv>=1.0.0
-Dynamic: license-file
-
-### 1. Install Dependencies
-
-```bash
-pip install -r requirements.txt
-```
-
-### 2. Set Up API Key
-
-Set your HuggingFace API key in the environment or update `config.py`:
-
-```python
-# In geomind/config.py
-HF_API_KEY = "your_huggingface_api_key"
-```
-
-Get a free API key from [HuggingFace](https://huggingface.co/settings/tokens).
-
-### 3. Run the Agent
-
-```bash
-python main.py
-```
-
-## Example Queries
-
-```
-
-💬 "Create an RGB composite for the most recent image of London"
-
-💬 "Calculate NDVI for Central Park, New York"
-
-💬 "What images are available for Tokyo with less than 10% cloud cover?"
-```
-
-## Approach
-
-### Traditional Approach
-```
-Full Scene Download → Local Storage → Process → Result
-     ~720 MB            Disk I/O      Slow      
-```
-
-### GeoMind Approach (Zarr + fsspec)
-```
-HTTP Range Request → Stream Chunks → Process in Memory → Result
-     ~1-5 MB           No disk          Fast           
-```
+Metadata-Version: 2.4
+Name: geomind-ai
+Version: 1.0.2
+Summary: AI agent for geospatial analysis with Sentinel-2 satellite imagery
+Author: Harsh Shinde
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/HarshShinde0/GeoMind
+Project-URL: Repository, https://github.com/HarshShinde0/GeoMind
+Project-URL: Documentation, https://github.com/HarshShinde0/GeoMind#readme
+Project-URL: Issues, https://github.com/HarshShinde0/GeoMind/issues
+Keywords: geospatial,satellite-imagery,sentinel-2,ai-agent,remote-sensing,earth-observation
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Science/Research
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openai>=1.0.0
+Requires-Dist: pystac-client>=0.8.0
+Requires-Dist: pystac>=1.10.0
+Requires-Dist: xarray>=2024.1.0
+Requires-Dist: zarr>=2.18.0
+Requires-Dist: dask>=2024.1.0
+Requires-Dist: geopy>=2.4.0
+Requires-Dist: fsspec>=2024.1.0
+Requires-Dist: aiohttp>=3.9.0
+Requires-Dist: requests>=2.31.0
+Requires-Dist: s3fs>=2024.1.0
+Requires-Dist: matplotlib>=3.8.0
+Requires-Dist: numpy>=1.26.0
+Requires-Dist: python-dotenv>=1.0.0
+Dynamic: license-file
+
+###  Install Dependencies
+
+```bash
+pip install -r requirements.txt
+```
+
+###  Run the Agent
+
+```bash
+# Interactive mode
+geomind
+
+# Single query
+geomind --query "Find recent imagery of Paris"
+```
+
+## Example Queries
+
+```
+
+💬 "Create an RGB composite for the most recent image of London"
+
+💬 "Calculate NDVI for Central Park, New York"
+
+💬 "What images are available for Tokyo with less than 10% cloud cover?"
+```
+
+## Approach
+
+### Traditional Approach
+```
+Full Scene Download → Local Storage → Process → Result
+     ~720 MB            Disk I/O      Slow      
+```
+
+### GeoMind Approach (Zarr + fsspec)
+```
+HTTP Range Request → Stream Chunks → Process in Memory → Result
+     ~1-5 MB           No disk          Fast           
+```

geomind_ai-1.0.2.dist-info/RECORD

@@ -0,0 +1,14 @@
+geomind/__init__.py,sha256=mma_vqNjiOpc0WOvclIdFqULWUnc7P8cDfAfqkPOsd4,151
+geomind/agent.py,sha256=iE-mwRLuA5sn4YtCQYU30GxcwsJoDhbhxdUrYsUJU0c,15010
+geomind/cli.py,sha256=tmMkvO9scyvzpRtiKbcb1jtWn3fSgFoywQmuusmjDuI,3283
+geomind/config.py,sha256=hv3DNM7QbCuOSWuPmXeiVxEbGxH7fRLxOc8jexd7rrY,2011
+geomind/tools/__init__.py,sha256=8iumGwIFHh8Bj1VJNgZtmKnEBqCy6_cRkzYENDUH7x4,720
+geomind/tools/geocoding.py,sha256=hiLpzHpkJP6IgWAUtZMnHL6qpkWcYWVLpGe0yfYxXv8,3007
+geomind/tools/processing.py,sha256=vMp8PMb8h8QiBRBFRvI_TGRqEDBTDQvSV0zvC1Ji5bc,9976
+geomind/tools/stac_search.py,sha256=V6230l4aHjedPWXu-3Cjmfc6diSFh5zsycewUko0W8k,6452
+geomind_ai-1.0.2.dist-info/licenses/LICENSE,sha256=aveu0ERm7I3NnIu8rtpKdvd0eyRpmktXKU0PBABtSN0,1069
+geomind_ai-1.0.2.dist-info/METADATA,sha256=tF0kr6M6J_Vo1_RiU3sAqseL_76xU2rvJUT52yej8Mw,2233
+geomind_ai-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+geomind_ai-1.0.2.dist-info/entry_points.txt,sha256=2nPR3faYKl0-1epccvzMJ2xdi-Q1Vt7aOSvA84oIWnw,45
+geomind_ai-1.0.2.dist-info/top_level.txt,sha256=rjKWNSNRhq4R9xJoZGsG-eAaH7BmTVNvfrrbcaJMIIs,8
+geomind_ai-1.0.2.dist-info/RECORD,,