Regex match valid ngrok endpoints

utils/audio_palette.py

@@ -1,3 +1,4 @@
+import re
 import typing
 from datetime import datetime, timezone, timedelta
 
@@ -26,6 +27,7 @@ class AudioPalette:
             "Medium": "medium",
             "Slow": "low"
         }
+        self.ngrok_url_pattern = re.compile("https:\/\/[a-z0-9\-]+\.ngrok\.io\/")
     
     def prompt_construction(self, caption: str, pace: str, sentiment: typing.Union[str, None], instrument: typing.Union[str, None], first: bool = True):
         instrument = instrument if instrument is not None else ""
@@ -41,9 +43,9 @@ class AudioPalette:
         return prompt
     
     def generate_single(self, input_image: PIL.Image.Image, instrument: typing.Union[str, None], ngrok_endpoint: typing.Union[str, None]):
-        if not ngrok_endpoint:
-            print(f"[{now()}] ngrok endpoint missing")
-            raise Error("ngrok endpoint missing")
+        if not self.ngrok_url_pattern.search(ngrok_endpoint):
+            print(f"[{now()}] Invalid ngrok endpoint - {ngrok_endpoint}")
+            raise Error(f"Invalid ngrok endpoint - {ngrok_endpoint}")
         print(f"[{now()}] {ngrok_endpoint}")
 
         pace = self.pace_model.predict(input_image)
@@ -90,9 +92,9 @@ class AudioPalette:
         return file_name
     
     def generate_multiple(self, file_paths: typing.List[str], instrument: typing.Union[str, None], ngrok_endpoint: typing.Union[str, None]):
-        if not ngrok_endpoint:
-            print(f"[{now()}] ngrok endpoint missing")
-            raise Error("ngrok endpoint missing")
+        if not self.ngrok_url_pattern.search(ngrok_endpoint):
+            print(f"[{now()}] Invalid ngrok endpoint - {ngrok_endpoint}")
+            raise Error(f"Invalid ngrok endpoint - {ngrok_endpoint}")
         print(f"[{now()}] {ngrok_endpoint}")
 
         images = [Image.open(image_path) for image_path in file_paths]