Upload StableGR.py

Parsing and error handling have been added to the download.

StableGR.py

@@ -19,7 +19,6 @@ from stablepy import Model_Diffusers
 
 TEMP_DIR = os.getcwd()
 
-
 def parse_urls(s):
     url_pattern = "https?://[\\w/:%#\\$&\\?\\(\\)~\\.=\\+\\-]+"
     try:
@@ -66,10 +65,17 @@ def generate_image(model_id, prompt, lora_A, num_steps, guidance_scale, sampler,
 
 
 def download_file(dl_url, civitai_key, progress=gr.Progress(track_tqdm=True)):
-    download_dir = TEMP_DIR
-    progress(0, desc=f"Start downloading... {dl_url}")
-    output_filename = get_download_file(download_dir, dl_url, civitai_key)
-    return output_filename
+    try:
+        download_dir = TEMP_DIR
+        progress(0, desc=f"Start downloading... {dl_url}")
+        dl_urls = parse_urls(dl_url)
+        if len(dl_urls) == 0: raise Exception(f"Invalid or empty URL: {dl_url}")
+        output_filename = get_download_file(download_dir, dl_urls[0], civitai_key)
+        if not output_filename: raise Exception(f"Failed to download: {dl_urls[0]}")
+        return output_filename
+    except Exception as e:
+        print(e)
+        return gr.update()
 
 
 def save_civitai_info(dl_url, filename, civitai_key="", progress=gr.Progress(track_tqdm=True)):