From e7a2ecf333116e0d80cd2fffa382695df85ed471 Mon Sep 17 00:00:00 2001
From: Steven Atkinson <steven@atkinson.mn>
Date: Thu, 19 Sep 2024 08:21:54 -0700
Subject: [PATCH] Handle when there's no internet

---
 nam/train/gui/__init__.py | 35 +++++++++++++++++++++--------------
 1 file changed, 21 insertions(+), 14 deletions(-)

diff --git a/nam/train/gui/__init__.py b/nam/train/gui/__init__.py
index 2a7bcaa..9f49b1b 100644
--- a/nam/train/gui/__init__.py
+++ b/nam/train/gui/__init__.py
@@ -36,7 +36,6 @@
     from nam.train import metadata
     from nam.train._names import INPUT_BASENAMES, LATEST_VERSION
     from nam.train._version import Version, get_current_version
-    from nam.train.metadata import TRAINING_KEY
 
     _install_is_valid = True
     _HAVE_ACCELERATOR = torch.cuda.is_available() or torch.backends.mps.is_available()
@@ -577,7 +576,7 @@ def _open_metadata(self):
 
         self._wait_while_func(lambda resume: _UserMetadataGUI(resume, self))
 
-    def _pack_update_button(self, version_from: str, version_to: str):
+    def _pack_update_button(self, version_from: Version, version_to: Version):
         """
         Pack a button that a user can click to update
         """
@@ -606,7 +605,7 @@ def update_nam():
 
         self._widgets[_GUIWidgets.UPDATE] = tk.Button(
             self._frame_update,
-            text=f"Update trainer ({version_from} -> {version_to})",
+            text=f"Update ({str(version_from)} -> {str(version_to)})",
             width=_BUTTON_WIDTH,
             height=_BUTTON_HEIGHT,
             command=update_nam,
@@ -616,16 +615,25 @@ def update_nam():
     def _pack_update_button_if_update_is_available(self):
         class UpdateInfo(NamedTuple):
             available: bool
-            current_version: str
-            new_version: str
+            current_version: Version
+            new_version: Optional[Version]
 
         def get_info() -> UpdateInfo:
             # TODO error handling
             url = f"https://api.github.com/repos/sdatkinson/neural-amp-modeler/releases"
-            response = requests.get(url)
-
+            current_version = get_current_version()
+            try:
+                response = requests.get(url)
+            except requests.exceptions.ConnectionError:
+                print("WARNING: Failed to reach the server to check for updates")
+                return UpdateInfo(
+                    available=False, current_version=current_version, new_version=None
+                )
             if response.status_code != 200:
                 print(f"Failed to fetch releases. Status code: {response.status_code}")
+                return UpdateInfo(
+                    available=False, current_version=current_version, new_version=None
+                )
             else:
                 releases = response.json()
                 latest_version = None
@@ -640,18 +648,17 @@ def get_info() -> UpdateInfo:
                                 latest_version = this_version
                 else:
                     print("No releases found for this repository.")
-            if latest_version is None:
-                return
-            current_version = get_current_version()
-            update_available = latest_version > current_version
+            update_available = (
+                latest_version is not None and latest_version > current_version
+            )
             return UpdateInfo(
                 available=update_available,
-                current_version=str(current_version),
-                new_version=str(latest_version),
+                current_version=current_version,
+                new_version=latest_version,
             )
 
         update_info = get_info()
-        if True:  # update_info.available:
+        if update_info.available:
             self._pack_update_button(
                 update_info.current_version, update_info.new_version
             )