Update app.py

app.py

@@ -14,6 +14,7 @@ import random
 import time
 import requests
 import pandas as pd
+import tempfile
 
 #Load prompts for randomization
 df = pd.read_csv('prompts.csv', header=None)
@@ -23,6 +24,31 @@ prompt_values = df.values.flatten()
 with open('loras.json', 'r') as f:
     loras = json.load(f)
 
+def download_and_cache_lora_images(loras):
+    """Download all lora images and replace URLs with local paths"""
+    for lora in loras:
+        if lora.get('image') and lora['image'].startswith('http'):
+            try:
+                response = requests.get(lora['image'], timeout=10)
+                response.raise_for_status()
+                
+                # Create temp file
+                temp_dir = tempfile.gettempdir()
+                filename = f"lora_{hash(lora['image'])}.png"
+                local_path = os.path.join(temp_dir, filename)
+                
+                with open(local_path, 'wb') as f:
+                    f.write(response.content)
+                
+                lora['image'] = local_path
+                print(f"Downloaded image for {lora['title']}")
+            except Exception as e:
+                print(f"Failed to download image for {lora['title']}: {e}")
+                lora['image'] = None
+    return loras
+
+loras = download_and_cache_lora_images(loras)
+
 # Initialize the base model
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"