geospatial-ai-tool 0.1.0__tar.gz

geospatial_ai_tool-0.1.0/.gitignore

@@ -0,0 +1,88 @@
+# =========================
+# Environments
+# =========================
+.venv/
+venv/
+env/
+.env/
+.conda/
+miniconda3/
+anaconda3/
+
+# =========================
+# Python Cache & Build
+# =========================
+__pycache__/
+*.py[cod]
+*$py.class
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# =========================
+# OS & Editor Files
+# =========================
+.DS_Store
+.vscode/
+.idea/
+*.swp
+
+# =========================
+# Databases & Local State
+# =========================
+*.db
+*.sqlite3
+.jupyter/
+.pytest_cache/
+
+# =========================
+# Data & Large Files
+# =========================
+# Ignore local raster/geospatial files and datasets
+*.tif
+*.tiff
+*.grd
+*.gri
+*.hdr
+*.nc
+*.sdat
+*.db
+*.cpg
+*.prj
+*.dbf
+*.sbn
+*.shp
+*.shx
+*.sbx
+*.xml
+*.mgrd
+*.sgrd
+*.aux
+*.kmz
+data/
+temp/
+
+# =========================
+# Secrets & Env Variables
+# =========================
+.env
+
+# =========================
+# Project Specifics
+# =========================
+GeoAI_Outputs/
+saved_models.json
+Pretrained_Brady_model/
+3D_AI/
+MYLES_2D_AI_MODELS_CLEAN/
+training_stacks/

geospatial_ai_tool-0.1.0/GeoAI_App.spec

@@ -0,0 +1,61 @@
+# -*- mode: python ; coding: utf-8 -*-
+from PyInstaller.utils.hooks import collect_all
+
+datas = []
+binaries = []
+hiddenimports = []
+tmp_ret = collect_all('customtkinter')
+datas += tmp_ret[0]; binaries += tmp_ret[1]; hiddenimports += tmp_ret[2]
+tmp_ret = collect_all('tensorflow')
+datas += tmp_ret[0]; binaries += tmp_ret[1]; hiddenimports += tmp_ret[2]
+tmp_ret = collect_all('rasterio')
+datas += tmp_ret[0]; binaries += tmp_ret[1]; hiddenimports += tmp_ret[2]
+
+
+a = Analysis(
+    ['impr_app.py'],
+    pathex=[],
+    binaries=binaries,
+    datas=datas,
+    hiddenimports=hiddenimports,
+    hookspath=[],
+    hooksconfig={},
+    runtime_hooks=[],
+    excludes=[],
+    noarchive=False,
+    optimize=0,
+)
+pyz = PYZ(a.pure)
+
+exe = EXE(
+    pyz,
+    a.scripts,
+    [],
+    exclude_binaries=True,
+    name='GeoAI_App',
+    debug=False,
+    bootloader_ignore_signals=False,
+    strip=False,
+    upx=True,
+    console=False,
+    disable_windowed_traceback=False,
+    argv_emulation=False,
+    target_arch=None,
+    codesign_identity=None,
+    entitlements_file=None,
+)
+coll = COLLECT(
+    exe,
+    a.binaries,
+    a.datas,
+    strip=False,
+    upx=True,
+    upx_exclude=[],
+    name='GeoAI_App',
+)
+app = BUNDLE(
+    coll,
+    name='GeoAI.app',
+    icon="MyIcon.icns",
+    bundle_identifier=None,
+)

geospatial_ai_tool-0.1.0/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.5
+Name: geospatial-ai-tool
+Version: 0.1.0
+Summary: 2D and 3D geospatial AI model prediction tool.
+Requires-Python: ==3.10.*
+Requires-Dist: customtkinter
+Requires-Dist: geopandas
+Requires-Dist: matplotlib
+Requires-Dist: numpy
+Requires-Dist: pandas
+Requires-Dist: pillow
+Requires-Dist: pyshortcuts
+Requires-Dist: rasterio
+Requires-Dist: scikit-learn
+Requires-Dist: tensorflow==2.21.0
+Requires-Dist: tkinterdnd2
+Requires-Dist: tqdm
+Description-Content-Type: text/markdown
+
+empty lol

geospatial_ai_tool-0.1.0/README.md

@@ -0,0 +1 @@
+empty lol

geospatial_ai_tool-0.1.0/environment.yml

@@ -0,0 +1,46 @@
+name: geo_app
+channels:
+  - conda-forge
+dependencies:
+  - python=3.10
+  - numpy
+  - tqdm
+  - pillow
+  - geopandas
+  - tk
+  - pandas
+  - matplotlib
+  - gdal
+  - rasterio
+  - scikit-learn
+  - pip:
+      - absl-py==2.5.0
+      - altgraph==0.17.5
+      - astunparse==1.6.3
+      - customtkinter==6.0.0
+      - darkdetect==0.8.0
+      - flatbuffers==25.12.19
+      - gast==0.7.0
+      - google-pasta==0.2.0
+      - grpcio==1.83.1
+      - h5py==3.14.0
+      - keras==3.12.4
+      - libclang==18.1.1
+      - macholib==1.16.4
+      - markdown-it-py==4.2.0
+      - mdurl==0.1.2
+      - ml-dtypes==0.6.0
+      - namex==0.1.0
+      - opt-einsum==3.4.0
+      - optree==0.20.0
+      - protobuf==7.36.1
+      - pygments==2.21.0
+      - pyinstaller==6.22.2
+      - pyinstaller-hooks-contrib==2026.7
+      - rich==15.0.0
+      - tensorflow==2.21.0
+      - termcolor==3.3.0
+      - tkinterdnd2==0.6.3
+      - typing-extensions==4.16.0
+      - wrapt==2.4.0
+prefix: /opt/homebrew/Caskroom/miniforge/base/envs/my_app

geospatial_ai_tool-0.1.0/pyproject.toml

@@ -0,0 +1,40 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "geospatial-ai-tool" # The pip/uv install name (must be unique on PyPI)
+version = "0.1.0"
+description = "2D and 3D geospatial AI model prediction tool."
+readme = "README.md"
+requires-python = "==3.10.*"
+dependencies = [
+    # Core Application
+    "customtkinter",
+    "pyshortcuts",
+    
+    # Data & Machine Learning
+    "numpy",
+    "pandas",
+    "scikit-learn",
+    "tensorflow==2.21.0", # Pinned to the version you are currently using
+    
+    # Geospatial
+    "geopandas",
+    "rasterio",
+    
+    # Utilities
+    "matplotlib",
+    "pillow",
+    "tqdm",
+
+    "tkinterdnd2"
+]
+
+[project.scripts]
+# This creates the terminal command. 
+# Format: terminal-command = "module.file:function"
+run-geospatial = "impr_app.impr_app:main"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/impr_app"]

geospatial_ai_tool-0.1.0/src/impr_app/core/app_shell.py

@@ -0,0 +1,323 @@
+# core/app_shell.py
+import os
+import queue
+import shutil
+import tkinter as tk
+from tkinter import messagebox
+
+import customtkinter as ctk
+from impr_app.core.models import SavedModelLibrary
+from impr_app.pages.home_page import HomePage
+from impr_app.pages.output_page import OutputPage
+from impr_app.pages.page_2d import Page2D
+from impr_app.pages.page_3d import Page3D
+from impr_app.pages.settings_page import SettingsPage
+from tkinterdnd2 import DND_FILES, TkinterDnD
+
+
+class DnDCTk(ctk.CTk, TkinterDnD.DnDWrapper):
+    """Enables drag and drop feature for the UI as a whole"""
+
+    def __init__(self, *args, **kwargs):
+        """Initializes custom tkinter with drag and drop tkinter"""
+        super().__init__(*args, **kwargs)
+        self.TkdndVersion = TkinterDnD._require(self)
+
+
+class App(DnDCTk):
+    """Main ctk controller that manages generic settings like app width as
+    well as the generation of pages and switching between them.
+    """
+
+    def __init__(self):
+        """Builds window, initializes queue, activates drag and drop, builds pages
+        and shows the home page.
+        """
+        self.THEMES = {
+            # ---------------------------------------------------------
+            # DEFAULT: Your current earth-toned & navy palette
+            # ---------------------------------------------------------
+            "Earth-Toned": {
+                "bg": "#EDF1F5",
+                "card_bg": "#FFFFFF",
+                "border": "#D5DBE1",
+                "text_primary": "#182A45",
+                "text_secondary": "#63758A",
+                "text_muted": "#4F6075",
+                "text_inverse": "white",
+                "text_input": "black",
+                "text_placeholder": "#8A9BB0",
+                "badge_bg": "#8F7870",
+                "input_bg": "#F8FAFC",
+                "btn_primary": "#D4B592",
+                "btn_primary_hover": "#C0A07B",
+                "btn_secondary_bg": "#FFFFFF",
+                "btn_secondary_hover": "#F0F4F8",
+                "btn_accent": "#F2C94C",
+                "btn_accent_hover": "#D4AF37",
+                "btn_dropdown_hover": "#7A654E",
+            },
+            # ---------------------------------------------------------
+            # DEUTERANOPIA (Red/Green weak - Most Common)
+            # Replaces warm earth tones with distinct cool blues and bright oranges.
+            # ---------------------------------------------------------
+            "Default (deuteranopia)": {
+                "bg": "#EDF1F5",
+                "card_bg": "#FFFFFF",
+                "border": "#D5DBE1",
+                "text_primary": "#182A45",
+                "text_secondary": "#63758A",
+                "text_muted": "#4F6075",
+                "text_inverse": "white",
+                "text_input": "black",
+                "text_placeholder": "#8A9BB0",
+                "badge_bg": "#0072B2",  # Replaced warm brown with Dark Blue
+                "input_bg": "#F8FAFC",
+                "btn_primary": "#56B4E9",  # Replaced Tan with Sky Blue
+                "btn_primary_hover": "#3D8CBB",
+                "btn_secondary_bg": "#FFFFFF",
+                "btn_secondary_hover": "#F0F4F8",
+                "btn_accent": "#E69F00",  # Replaced Gold with High-Vis Orange
+                "btn_accent_hover": "#C28600",
+                "btn_dropdown_hover": "#005A8E",
+            },
+            # ---------------------------------------------------------
+            # PROTANOPIA (Red-blind)
+            # Similar to Deuteranopia, but avoids dark reds/browns which appear black.
+            # ---------------------------------------------------------
+            "protanopia": {
+                "bg": "#EDF1F5",
+                "card_bg": "#FFFFFF",
+                "border": "#D5DBE1",
+                "text_primary": "#182A45",
+                "text_secondary": "#63758A",
+                "text_muted": "#4F6075",
+                "text_inverse": "white",
+                "text_input": "black",
+                "text_placeholder": "#8A9BB0",
+                "badge_bg": "#E69F00",  # Replaced warm brown with Bright Orange
+                "input_bg": "#F8FAFC",
+                "btn_primary": "#0072B2",  # Replaced Tan with deep, visible Blue
+                "btn_primary_hover": "#005A8E",
+                "btn_secondary_bg": "#FFFFFF",
+                "btn_secondary_hover": "#F0F4F8",
+                "btn_accent": "#F0E442",  # Replaced Gold with brighter Yellow
+                "btn_accent_hover": "#D9CE36",
+                "btn_dropdown_hover": "#00456E",
+            },
+            # ---------------------------------------------------------
+            # TRITANOPIA (Blue/Yellow blind - Rare)
+            # Strips out the Navy Blue and Gold completely.
+            # Uses Charcoal for text, Vermilion (Red) and Teal for actions.
+            # ---------------------------------------------------------
+            "tritanopia": {
+                "bg": "#F4F4F4",  # Shifted away from cool blue-grey
+                "card_bg": "#FFFFFF",
+                "border": "#D0D0D0",
+                "text_primary": "#2C2C2C",  # Shifted Navy to neutral Charcoal
+                "text_secondary": "#5A5A5A",  # Shifted Slate to neutral Grey
+                "text_muted": "#4A4A4A",
+                "text_inverse": "white",
+                "text_input": "black",
+                "text_placeholder": "#999999",
+                "badge_bg": "#CC79A7",  # Reddish-purple (very clear for Tritanopes)
+                "input_bg": "#FAFAFA",
+                "btn_primary": "#D55E00",  # Vermilion (Red-Orange) replaces Tan
+                "btn_primary_hover": "#A34800",
+                "btn_secondary_bg": "#FFFFFF",
+                "btn_secondary_hover": "#EEEEEE",
+                "btn_accent": "#009E73",  # Teal/Green replaces Gold
+                "btn_accent_hover": "#007A59",
+                "btn_dropdown_hover": "#7A3500",
+            },
+        }
+
+        super().__init__()
+        self.theme_colors = self.THEMES["Default (deuteranopia)"]
+        self.configure(fg_color=self.theme_colors["bg"])
+        self.current_scale = 1.2
+        screen_width = self.winfo_screenwidth()
+        screen_height = self.winfo_screenheight()
+        # UI Looks better when using the entire screen, also gives images
+        # more space
+        self.geometry(f"{screen_width}x{screen_height}")
+        self.title("GEO AI for Geothermal Exploration")
+
+        # Multithreading is used so the UI does not freeze during processing,
+        # which makes the queue necessary for thread safety
+        self.log_queue = queue.Queue()
+        self.model_library = SavedModelLibrary("DEADMEAT")
+        self.current_probability_map = None
+
+        self.container = ctk.CTkFrame(self)
+        self.container = ctk.CTkFrame(self, fg_color="transparent")
+        self.container.pack(fill="both", expand=True)
+
+        self.pages = {}
+        self.active_page = None
+
+        self._build_pages()
+
+        self.drop_target_register(DND_FILES)
+        self.dnd_bind("<<Drop>>", self._handle_drop)
+
+        self.show_page("HomePage")
+        self.apply_scaling(self.current_scale)
+        self.after(100, self.check_queue)
+
+        self.saved_files = []
+        self.protocol("WM_DELETE_WINDOW", self.on_closing)
+
+    def _build_pages(self):
+        """Loops through every provided page class, creates an object
+        for each page, and saves those objects to the pages dict.
+        """
+        for PageClass in (HomePage, Page2D, Page3D, OutputPage, SettingsPage):
+            page = PageClass(self.container, controller=self)
+            self.pages[PageClass.__name__] = page
+
+    def show_page(self, page_name):
+        """Shows the desired page while hiding the others.
+
+        Args:
+            page_name (str): key for dictionary containing corresponding
+            homepage object.
+        """
+        # Unpacking pages instead of hiding them for improved efficiency
+        for page in self.pages.values():
+            page.pack_forget()
+        page = self.pages[page_name]
+        page.pack(fill="both", expand=True)
+        self.active_page = page
+        page.on_show()
+        self.update_idletasks()
+
+    def check_queue(self):
+        """Loops through queue to display messages to app console safely.
+        Continuously calls itself.
+        """
+        while not self.log_queue.empty():
+            message = self.log_queue.get()
+            # This prevents another thread other than the UI thread from
+            # trying to manipulate the log box
+            self.active_page.append_log(message)
+        self.after(100, self.check_queue)
+
+    def _handle_drop(self, event):
+        """Checks if a page uses drag and drop, if it does
+        has page handle the drop
+
+        Args:
+            event (tkinter.Event): Contains data about the drag and drop action
+        """
+
+        # Not all pages in the application use drag and drop
+        if hasattr(self.active_page, "handle_drop"):
+            self.active_page.handle_drop(event)
+
+    def change_theme(self, new_theme_name):
+        # Store the old colors before we update so we can map them
+        old_colors = self.theme_colors
+
+        # Update to the new colors
+        self.current_theme_name = new_theme_name
+        self.theme_colors = self.THEMES[new_theme_name]
+
+        for page in self.pages.values():
+            page.colors = self.theme_colors
+
+        # Update the main window background
+        self.configure(fg_color=self.theme_colors["bg"])
+
+        # Start the recursion, passing both old and new dictionaries
+        self._update_widget_tree(self, old_colors, self.theme_colors)
+
+    def _update_widget_tree(self, parent_widget, old_colors, new_colors):
+        """Recursively checks and updates every widget."""
+        for widget in parent_widget.winfo_children():
+            # Update the 4 color properties CTk uses in your UI
+            self._swap_color(widget, "fg_color", old_colors, new_colors)
+            self._swap_color(widget, "text_color", old_colors, new_colors)
+            self._swap_color(widget, "border_color", old_colors, new_colors)
+            self._swap_color(widget, "hover_color", old_colors, new_colors)
+            self._swap_color(widget, "button_color", old_colors, new_colors)
+            self._swap_color(widget, "button_hover_color", old_colors, new_colors)
+
+            # Recurse deeper if this widget has children
+            if widget.winfo_children():
+                self._update_widget_tree(widget, old_colors, new_colors)
+
+    def _swap_color(self, widget, property_name, old_colors, new_colors):
+        """Looks at a widget's current hex code and maps it to the new theme."""
+        try:
+            # cget() grabs the current value of a widget property
+            current_color = widget.cget(property_name)
+
+            # CTk sometimes returns colors as a list/tuple for light/dark mode (e.g. ['#FFF', '#000'])
+            # We just need to check the string representation
+            if isinstance(current_color, (list, tuple)):
+                current_color = current_color[0]
+
+            # Never alter transparent widgets
+            if str(current_color).lower() == "transparent":
+                return
+
+            # Find this color in the OLD dictionary, apply from the NEW dictionary
+            for key, old_hex in old_colors.items():
+                if str(current_color).lower() == str(old_hex).lower():
+                    # We found a match! Apply the new hex code for this key.
+                    # **{} is just dynamically unpacking the property name (e.g. fg_color="#NEW_HEX")
+                    widget.configure(**{property_name: new_colors[key]})
+                    return  # Stop searching once we find a match
+
+        except (ValueError, tk.TclError, AttributeError):
+            # Pass gracefully if the widget doesn't support the property we are checking
+            pass
+
+    def update_scaling(self, new_scale):
+        self.current_scale = new_scale
+
+        self.apply_scaling(new_scale)
+
+    def apply_scaling(self, scale_factor):
+        ctk.set_widget_scaling(scale_factor)
+
+        # Get the current dimensions of the window
+        current_width = self.winfo_width()
+        current_height = self.winfo_height()
+
+        # Expand the window by 1 pixel to force a hard OS-level redraw
+        self.geometry(f"{current_width + 1}x{current_height}")
+        self.update()
+
+        # Snap it instantly back to the correct size
+        self.geometry(f"{current_width}x{current_height}")
+
+    def on_closing(self):
+        # Trigger the pop-up when the window close is attempted
+        # askyesnocancel returns: True (Yes), False (No), or None (Cancel)
+        response = messagebox.askyesnocancel(
+            title="Save before closing?",
+            message="Would you like to save your generated files before exiting?",
+        )
+
+        # Handle the user's choice
+        if response:
+            self.destroy()
+
+        elif response is False:
+            for dir_path in self.saved_files:
+                if os.path.exists(dir_path):
+                    shutil.rmtree(dir_path)
+            self.destroy()
+
+    def register_saved_file(self, filepath):
+        """Safely appends a newly generated file path to the saved files list."""
+        if filepath not in self.saved_files:
+            self.saved_files.append(filepath)
+            print(f"Tracking new file: {filepath}")  # Optional: for debugging
+
+
+if __name__ == "__main__":
+    app = App()
+    app.mainloop()

geospatial_ai_tool-0.1.0/src/impr_app/core/models.py

@@ -0,0 +1,70 @@
+# core/models.py
+import json
+from pathlib import Path
+
+
+class SavedModelLibrary:
+    """Manages the saving, loading, and deletion of user-imported AI models
+    to a local JSON file so they persist between app sessions.
+    """
+
+    def __init__(self, memory_file_path: str) -> None:
+        """Initializes saved model library with memory file path
+
+        Args:
+            memory_file_path (str): raw relative file path to the applications saved memory. 
+        """
+        memory_file_path = Path.home() / "Documents"
+        print(memory_file_path)
+
+    def set_memory_file(self, memory_file_path: str) -> None:
+        self.memory_file = Path(memory_file_path) / "saved_models.json"
+        if not self.memory_file.exists():
+            self.memory_file.write_text("{}")
+
+    def get_all_models(self) -> dict:
+        """Loads and returns the dictionary of saved models.
+
+        Returns:
+            dict: A dictionary containing model names as keys which point to their file paths or
+            an empty dictionary if file DNE or is corrupted
+        """
+        if self.memory_file.exists():
+            try:
+                with open(self.memory_file, "r") as f:
+                    return json.load(f)
+            except json.JSONDecodeError:
+                return {}
+        return {}
+
+    def _save_models(self, models_dict: dict) -> None:
+        """Helper function that saves the models in dictionary to json file
+
+        Args:
+            models_dict (dict): A dictionary containing user generated label
+            and filepath to model.
+        """
+        with open(self.memory_file, "w") as f:
+            json.dump(models_dict, f, indent=4)
+
+    def add_model(self, name: str, filepath: str | Path) -> None:
+        """Adds a GeoAI model to the UI as a selectable choice
+
+        Args:
+            name (str): The name a user has given the model
+            filepath (str | Path): The filepath to the model they selected
+        """
+        models = self.get_all_models()
+        models[name] = str(filepath)
+        self._save_models(models)
+
+    def delete_model(self, name) -> None:
+        """Deletes model from working dictionary (if exists) and updates the json model data
+
+        Args:
+            name (str): Name the user generated for a specific model
+        """
+        models = self.get_all_models()
+        if name in models:
+            del models[name]
+            self._save_models(models)