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

geomind/cli.py

@@ -1,125 +1,125 @@
-"""
-Command-line interface for GeoMind.
-"""
-
-import sys
-import argparse
-from typing import Optional
-
-from .agent import GeoMindAgent
-
-
-def main():
-    """Main CLI entry point for the geomind package."""
-    parser = argparse.ArgumentParser(
-        description="GeoMind - AI agent for geospatial analysis with Sentinel-2 imagery",
-        formatter_class=argparse.RawDescriptionHelpFormatter,
-        epilog="""
-Examples:
-  # Interactive mode
-  geomind
-
-  # Single query
-  geomind --query "Find recent imagery of Paris"
-
-  # With custom model
-  geomind --model "anthropic/claude-3.5-sonnet"
-
-  # With API key
-  geomind --api-key "your-openrouter-api-key"
-
-Environment Variables:
-  OPENROUTER_API_KEY    Your OpenRouter API key
-  OPENROUTER_MODEL      Model to use (default: xiaomi/mimo-v2-flash:free)
-  OPENROUTER_API_URL    API endpoint (default: https://openrouter.ai/api/v1)
-        """,
-    )
-
-    parser.add_argument(
-        "--query",
-        "-q",
-        type=str,
-        help="Single query to run (if not provided, starts interactive mode)",
-    )
-    parser.add_argument(
-        "--model",
-        "-m",
-        type=str,
-        help="Model name to use (e.g., 'anthropic/claude-3.5-sonnet')",
-    )
-    parser.add_argument(
-        "--api-key",
-        "-k",
-        type=str,
-        help="OpenRouter API key (or set OPENROUTER_API_KEY env variable)",
-    )
-    parser.add_argument(
-        "--version", "-v", action="store_true", help="Show version and exit"
-    )
-
-    args = parser.parse_args()
-
-    if args.version:
-        from . import __version__
-
-        print(f"GeoMind version {__version__}")
-        sys.exit(0)
-
-    # Start interactive or single-query mode
-    try:
-        if args.query:
-            # Single query mode
-            agent = GeoMindAgent(model=args.model, api_key=args.api_key)
-            agent.chat(args.query)
-        else:
-            # Interactive mode
-            run_interactive(model=args.model, api_key=args.api_key)
-    except ValueError as e:
-        print(f"\n❌ Error: {e}")
-        sys.exit(1)
-    except KeyboardInterrupt:
-        print("\n\n👋 Goodbye!")
-        sys.exit(0)
-    except Exception as e:
-        print(f"\n❌ Unexpected error: {e}")
-        sys.exit(1)
-
-
-def run_interactive(model: Optional[str] = None, api_key: Optional[str] = None):
-    """Run interactive CLI mode."""
-    print("=" * 60)
-    print("🌍 GeoMind - Geospatial AI Agent")
-    print("=" * 60)
-    print("Powered by OpenRouter | Sentinel-2 Imagery")
-    print("Type 'quit' or 'exit' to end the session")
-    print("Type 'reset' to start a new conversation")
-    print("=" * 60)
-
-    agent = GeoMindAgent(model=model, api_key=api_key)
-
-    while True:
-        try:
-            user_input = input("\n💬 You: ").strip()
-
-            if not user_input:
-                continue
-
-            if user_input.lower() in ["quit", "exit", "q"]:
-                print("\n👋 Goodbye!")
-                break
-
-            if user_input.lower() == "reset":
-                agent.reset()
-                continue
-
-            agent.chat(user_input)
-
-        except KeyboardInterrupt:
-            print("\n\n👋 Goodbye!")
-            break
-        except Exception as e:
-            print(f"\n❌ Error: {e}")
-
-
-if __name__ == "__main__":
-    main()
+"""
+Command-line interface for GeoMind.
+"""
+
+import sys
+import argparse
+from typing import Optional
+
+from .agent import GeoMindAgent
+
+
+def main():
+    """Main CLI entry point for the geomind package."""
+    parser = argparse.ArgumentParser(
+        description="GeoMind - AI agent for geospatial analysis with Sentinel-2 imagery",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Examples:
+  # Interactive mode
+  geomind
+
+  # Single query
+  geomind --query "Find recent imagery of Paris"
+
+  # With custom model
+  geomind --model "anthropic/claude-3.5-sonnet"
+
+  # With API key
+  geomind --api-key "your-openrouter-api-key"
+
+Environment Variables:
+  OPENROUTER_API_KEY    Your OpenRouter API key
+  OPENROUTER_MODEL      Model to use (default: xiaomi/mimo-v2-flash:free)
+  OPENROUTER_API_URL    API endpoint (default: https://openrouter.ai/api/v1)
+        """,
+    )
+
+    parser.add_argument(
+        "--query",
+        "-q",
+        type=str,
+        help="Single query to run (if not provided, starts interactive mode)",
+    )
+    parser.add_argument(
+        "--model",
+        "-m",
+        type=str,
+        help="Model name to use (e.g., 'anthropic/claude-3.5-sonnet')",
+    )
+    parser.add_argument(
+        "--api-key",
+        "-k",
+        type=str,
+        help="OpenRouter API key (or set OPENROUTER_API_KEY env variable)",
+    )
+    parser.add_argument(
+        "--version", "-v", action="store_true", help="Show version and exit"
+    )
+
+    args = parser.parse_args()
+
+    if args.version:
+        from . import __version__
+
+        print(f"GeoMind version {__version__}")
+        sys.exit(0)
+
+    # Start interactive or single-query mode
+    try:
+        if args.query:
+            # Single query mode
+            agent = GeoMindAgent(model=args.model, api_key=args.api_key)
+            agent.chat(args.query)
+        else:
+            # Interactive mode
+            run_interactive(model=args.model, api_key=args.api_key)
+    except ValueError as e:
+        print(f"\n❌ Error: {e}")
+        sys.exit(1)
+    except KeyboardInterrupt:
+        print("\n\n👋 Goodbye!")
+        sys.exit(0)
+    except Exception as e:
+        print(f"\n❌ Unexpected error: {e}")
+        sys.exit(1)
+
+
+def run_interactive(model: Optional[str] = None, api_key: Optional[str] = None):
+    """Run interactive CLI mode."""
+    print("=" * 60)
+    print("🌍 GeoMind - Geospatial AI Agent")
+    print("=" * 60)
+    print("Powered by OpenRouter | Sentinel-2 Imagery")
+    print("Type 'quit' or 'exit' to end the session")
+    print("Type 'reset' to start a new conversation")
+    print("=" * 60)
+
+    agent = GeoMindAgent(model=model, api_key=api_key)
+
+    while True:
+        try:
+            user_input = input("\n💬 You: ").strip()
+
+            if not user_input:
+                continue
+
+            if user_input.lower() in ["quit", "exit", "q"]:
+                print("\n👋 Goodbye!")
+                break
+
+            if user_input.lower() == "reset":
+                agent.reset()
+                continue
+
+            agent.chat(user_input)
+
+        except KeyboardInterrupt:
+            print("\n\n👋 Goodbye!")
+            break
+        except Exception as e:
+            print(f"\n❌ Error: {e}")
+
+
+if __name__ == "__main__":
+    main()

geomind/config.py

@@ -1,55 +1,55 @@
-"""
-Configuration settings for GeoMind agent.
-"""
-
-import os
-from pathlib import Path
-from dotenv import load_dotenv
-
-# Load environment variables from .env file
-load_dotenv()
-
-# STAC API Configuration
-STAC_API_URL = "https://stac.core.eopf.eodc.eu"
-STAC_COLLECTION = "sentinel-2-l2a"
-
-# Sentinel-2 Band Configuration
-SENTINEL2_BANDS = {
-    "B01": {"name": "Coastal aerosol", "wavelength": 0.443, "resolution": 60},
-    "B02": {"name": "Blue", "wavelength": 0.490, "resolution": 10},
-    "B03": {"name": "Green", "wavelength": 0.560, "resolution": 10},
-    "B04": {"name": "Red", "wavelength": 0.665, "resolution": 10},
-    "B05": {"name": "Red Edge 1", "wavelength": 0.704, "resolution": 20},
-    "B06": {"name": "Red Edge 2", "wavelength": 0.740, "resolution": 20},
-    "B07": {"name": "Red Edge 3", "wavelength": 0.783, "resolution": 20},
-    "B08": {"name": "NIR", "wavelength": 0.842, "resolution": 10},
-    "B8A": {"name": "NIR Narrow", "wavelength": 0.865, "resolution": 20},
-    "B09": {"name": "Water Vapour", "wavelength": 0.945, "resolution": 60},
-    "B11": {"name": "SWIR 1", "wavelength": 1.610, "resolution": 20},
-    "B12": {"name": "SWIR 2", "wavelength": 2.190, "resolution": 20},
-}
-
-# Reflectance scale and offset (from STAC metadata)
-REFLECTANCE_SCALE = 0.0001
-REFLECTANCE_OFFSET = -0.1
-
-# RGB Band Mapping
-RGB_BANDS = {"red": "b04", "green": "b03", "blue": "b02"}
-
-# Default search parameters
-DEFAULT_MAX_CLOUD_COVER = 20  # percent
-DEFAULT_BUFFER_KM = 10  # km buffer around point for bbox
-DEFAULT_MAX_ITEMS = 10
-
-# Output directory for saved images
-OUTPUT_DIR = Path("outputs")
-OUTPUT_DIR.mkdir(exist_ok=True)
-
-# Geocoding configuration
-GEOCODER_USER_AGENT = "geomind_agent_v0.1"
-
-# OpenRouter API Configuration
-# OpenRouter provides access to multiple AI models via API
-OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
-OPENROUTER_API_URL = os.getenv("OPENROUTER_API_URL", "https://openrouter.ai/api/v1")
-OPENROUTER_MODEL = os.getenv("OPENROUTER_MODEL", "xiaomi/mimo-v2-flash:free")
+"""
+Configuration settings for GeoMind agent.
+"""
+
+import os
+from pathlib import Path
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
+
+# STAC API Configuration
+STAC_API_URL = "https://stac.core.eopf.eodc.eu"
+STAC_COLLECTION = "sentinel-2-l2a"
+
+# Sentinel-2 Band Configuration
+SENTINEL2_BANDS = {
+    "B01": {"name": "Coastal aerosol", "wavelength": 0.443, "resolution": 60},
+    "B02": {"name": "Blue", "wavelength": 0.490, "resolution": 10},
+    "B03": {"name": "Green", "wavelength": 0.560, "resolution": 10},
+    "B04": {"name": "Red", "wavelength": 0.665, "resolution": 10},
+    "B05": {"name": "Red Edge 1", "wavelength": 0.704, "resolution": 20},
+    "B06": {"name": "Red Edge 2", "wavelength": 0.740, "resolution": 20},
+    "B07": {"name": "Red Edge 3", "wavelength": 0.783, "resolution": 20},
+    "B08": {"name": "NIR", "wavelength": 0.842, "resolution": 10},
+    "B8A": {"name": "NIR Narrow", "wavelength": 0.865, "resolution": 20},
+    "B09": {"name": "Water Vapour", "wavelength": 0.945, "resolution": 60},
+    "B11": {"name": "SWIR 1", "wavelength": 1.610, "resolution": 20},
+    "B12": {"name": "SWIR 2", "wavelength": 2.190, "resolution": 20},
+}
+
+# Reflectance scale and offset (from STAC metadata)
+REFLECTANCE_SCALE = 0.0001
+REFLECTANCE_OFFSET = -0.1
+
+# RGB Band Mapping
+RGB_BANDS = {"red": "b04", "green": "b03", "blue": "b02"}
+
+# Default search parameters
+DEFAULT_MAX_CLOUD_COVER = 20  # percent
+DEFAULT_BUFFER_KM = 10  # km buffer around point for bbox
+DEFAULT_MAX_ITEMS = 10
+
+# Output directory for saved images
+OUTPUT_DIR = Path("outputs")
+OUTPUT_DIR.mkdir(exist_ok=True)
+
+# Geocoding configuration
+GEOCODER_USER_AGENT = "geomind_agent_v0.1"
+
+# OpenRouter API Configuration
+# OpenRouter provides access to multiple AI models via API
+OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
+OPENROUTER_API_URL = os.getenv("OPENROUTER_API_URL", "https://openrouter.ai/api/v1")
+OPENROUTER_MODEL = os.getenv("OPENROUTER_MODEL", "xiaomi/mimo-v2-flash:free")

geomind/tools/__init__.py

@@ -1,27 +1,27 @@
-"""
-GeoMind Tools - Functions available to the AI agent.
-
-This module contains all the tools that the agent can call:
-- Geocoding: Convert place names to coordinates
-- STAC Search: Query the Sentinel-2 catalog
-- Processing: Load and process imagery data
-"""
-
-from .geocoding import geocode_location, get_bbox_from_location
-from .stac_search import search_imagery, get_item_details, list_recent_imagery
-from .processing import (
-    create_rgb_composite,
-    calculate_ndvi,
-    get_band_statistics,
-)
-
-__all__ = [
-    "geocode_location",
-    "get_bbox_from_location",
-    "search_imagery",
-    "get_item_details",
-    "list_recent_imagery",
-    "create_rgb_composite",
-    "calculate_ndvi",
-    "get_band_statistics",
-]
+"""
+GeoMind Tools - Functions available to the AI agent.
+
+This module contains all the tools that the agent can call:
+- Geocoding: Convert place names to coordinates
+- STAC Search: Query the Sentinel-2 catalog
+- Processing: Load and process imagery data
+"""
+
+from .geocoding import geocode_location, get_bbox_from_location
+from .stac_search import search_imagery, get_item_details, list_recent_imagery
+from .processing import (
+    create_rgb_composite,
+    calculate_ndvi,
+    get_band_statistics,
+)
+
+__all__ = [
+    "geocode_location",
+    "get_bbox_from_location",
+    "search_imagery",
+    "get_item_details",
+    "list_recent_imagery",
+    "create_rgb_composite",
+    "calculate_ndvi",
+    "get_band_statistics",
+]

geomind/tools/geocoding.py

@@ -1,108 +1,108 @@
-"""
-Geocoding tools for converting place names to coordinates.
-
-Uses OpenStreetMap's Nominatim service via geopy.
-"""
-
-from typing import Optional
-from geopy.geocoders import Nominatim
-from geopy.extra.rate_limiter import RateLimiter
-
-from ..config import GEOCODER_USER_AGENT, DEFAULT_BUFFER_KM
-
-
-def geocode_location(place_name: str) -> dict:
-    """
-    Convert a place name to geographic coordinates.
-
-    Args:
-        place_name: Name of the location (e.g., "New York", "Paris, France")
-
-    Returns:
-        Dictionary with latitude, longitude, and full address
-
-    Example:
-        >>> geocode_location("Central Park, New York")
-        {'latitude': 40.7828, 'longitude': -73.9653, 'address': '...'}
-    """
-    geolocator = Nominatim(user_agent=GEOCODER_USER_AGENT, timeout=10)
-    geocode = RateLimiter(geolocator.geocode, min_delay_seconds=1)
-
-    location = geocode(place_name)
-
-    if location is None:
-        return {
-            "success": False,
-            "error": f"Could not find location: {place_name}",
-            "latitude": None,
-            "longitude": None,
-            "address": None,
-        }
-
-    return {
-        "success": True,
-        "latitude": location.latitude,
-        "longitude": location.longitude,
-        "address": location.address,
-    }
-
-
-def get_bbox_from_location(place_name: str, buffer_km: Optional[float] = None) -> dict:
-    """
-    Convert a place name to a bounding box suitable for STAC queries.
-
-    Creates a square bounding box centered on the location with the
-    specified buffer distance.
-
-    Args:
-        place_name: Name of the location (e.g., "San Francisco")
-        buffer_km: Buffer distance in kilometers (default: 10km)
-
-    Returns:
-        Dictionary with bbox [min_lon, min_lat, max_lon, max_lat] and center point
-
-    Example:
-        >>> get_bbox_from_location("London", buffer_km=5)
-        {'bbox': [-0.17, 51.46, -0.08, 51.55], 'center': {...}}
-    """
-    if buffer_km is None:
-        buffer_km = DEFAULT_BUFFER_KM
-
-    # Get coordinates
-    location_result = geocode_location(place_name)
-
-    if not location_result["success"]:
-        return {
-            "success": False,
-            "error": location_result["error"],
-            "bbox": None,
-        }
-
-    lat = location_result["latitude"]
-    lon = location_result["longitude"]
-
-    # Calculate approximate degree offset
-    # 1 degree latitude ≈ 111 km
-    # 1 degree longitude ≈ 111 * cos(latitude) km
-    import math
-
-    lat_offset = buffer_km / 111.0
-    lon_offset = buffer_km / (111.0 * math.cos(math.radians(lat)))
-
-    bbox = [
-        lon - lon_offset,  # min_lon (west)
-        lat - lat_offset,  # min_lat (south)
-        lon + lon_offset,  # max_lon (east)
-        lat + lat_offset,  # max_lat (north)
-    ]
-
-    return {
-        "success": True,
-        "bbox": bbox,
-        "center": {
-            "latitude": lat,
-            "longitude": lon,
-        },
-        "address": location_result["address"],
-        "buffer_km": buffer_km,
-    }
+"""
+Geocoding tools for converting place names to coordinates.
+
+Uses OpenStreetMap's Nominatim service via geopy.
+"""
+
+from typing import Optional
+from geopy.geocoders import Nominatim
+from geopy.extra.rate_limiter import RateLimiter
+
+from ..config import GEOCODER_USER_AGENT, DEFAULT_BUFFER_KM
+
+
+def geocode_location(place_name: str) -> dict:
+    """
+    Convert a place name to geographic coordinates.
+
+    Args:
+        place_name: Name of the location (e.g., "New York", "Paris, France")
+
+    Returns:
+        Dictionary with latitude, longitude, and full address
+
+    Example:
+        >>> geocode_location("Central Park, New York")
+        {'latitude': 40.7828, 'longitude': -73.9653, 'address': '...'}
+    """
+    geolocator = Nominatim(user_agent=GEOCODER_USER_AGENT, timeout=10)
+    geocode = RateLimiter(geolocator.geocode, min_delay_seconds=1)
+
+    location = geocode(place_name)
+
+    if location is None:
+        return {
+            "success": False,
+            "error": f"Could not find location: {place_name}",
+            "latitude": None,
+            "longitude": None,
+            "address": None,
+        }
+
+    return {
+        "success": True,
+        "latitude": location.latitude,
+        "longitude": location.longitude,
+        "address": location.address,
+    }
+
+
+def get_bbox_from_location(place_name: str, buffer_km: Optional[float] = None) -> dict:
+    """
+    Convert a place name to a bounding box suitable for STAC queries.
+
+    Creates a square bounding box centered on the location with the
+    specified buffer distance.
+
+    Args:
+        place_name: Name of the location (e.g., "San Francisco")
+        buffer_km: Buffer distance in kilometers (default: 10km)
+
+    Returns:
+        Dictionary with bbox [min_lon, min_lat, max_lon, max_lat] and center point
+
+    Example:
+        >>> get_bbox_from_location("London", buffer_km=5)
+        {'bbox': [-0.17, 51.46, -0.08, 51.55], 'center': {...}}
+    """
+    if buffer_km is None:
+        buffer_km = DEFAULT_BUFFER_KM
+
+    # Get coordinates
+    location_result = geocode_location(place_name)
+
+    if not location_result["success"]:
+        return {
+            "success": False,
+            "error": location_result["error"],
+            "bbox": None,
+        }
+
+    lat = location_result["latitude"]
+    lon = location_result["longitude"]
+
+    # Calculate approximate degree offset
+    # 1 degree latitude ≈ 111 km
+    # 1 degree longitude ≈ 111 * cos(latitude) km
+    import math
+
+    lat_offset = buffer_km / 111.0
+    lon_offset = buffer_km / (111.0 * math.cos(math.radians(lat)))
+
+    bbox = [
+        lon - lon_offset,  # min_lon (west)
+        lat - lat_offset,  # min_lat (south)
+        lon + lon_offset,  # max_lon (east)
+        lat + lat_offset,  # max_lat (north)
+    ]
+
+    return {
+        "success": True,
+        "bbox": bbox,
+        "center": {
+            "latitude": lat,
+            "longitude": lon,
+        },
+        "address": location_result["address"],
+        "buffer_km": buffer_km,
+    }