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
  }