Update app.py

app.py

@@ -12,6 +12,7 @@ from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard, snapshot_d
 import copy
 import random
 import time
+import requests
 
 # Load LoRAs from JSON file
 with open('loras.json', 'r') as f:
@@ -56,6 +57,26 @@ class calculateDuration:
         else:
             print(f"Elapsed time: {self.elapsed_time:.6f} seconds")
 
+def download_file(url, directory=None):
+    if directory is None:
+        directory = os.getcwd()  # Use current working directory if not specified
+    
+    # Get the filename from the URL
+    filename = url.split('/')[-1]
+    
+    # Full path for the downloaded file
+    filepath = os.path.join(directory, filename)
+    
+    # Download the file
+    response = requests.get(url)
+    response.raise_for_status()  # Raise an exception for bad status codes
+    
+    # Write the content to the file
+    with open(filepath, 'wb') as file:
+        file.write(response.content)
+    
+    return filepath
+            
 def update_selection(evt: gr.SelectData, selected_indices, width, height):
     selected_index = evt.index
     selected_indices = selected_indices or []
@@ -337,10 +358,12 @@ def add_custom_lora(custom_lora, selected_indices):
             print(f"Loaded custom LoRA: {repo}")
             existing_item_index = next((index for (index, item) in enumerate(loras) if item['repo'] == repo), None)
             if existing_item_index is None:
+                if(repo.endswith(".safetensors")):
+                    downloaded_path = download_file(url)
                 new_item = {
                     "image": image if image is not None else "",
                     "title": title,
-                    "repo": repo,
+                    "repo": downloaded_path,
                     "weights": path,
                     "trigger_word": trigger_word
                 }