Update app.py

app.py

@@ -22,42 +22,19 @@ transformer = FluxTransformer2DModel.from_single_file(kontext_path, torch_dtype=
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", transformer=transformer, torch_dtype=torch.bfloat16).to("cuda")
 
 # Load LoRA data (you'll need to create this JSON file or modify to load your LoRAs)
-try:
-    with open("flux_loras.json", "r") as file:
-        data = json.load(file)
-        flux_loras_raw = [
-            {
-                "image": item["image"],
-                "title": item["title"],
-                "repo": item["repo"],
-                "trigger_word": item.get("trigger_word", ""),
-                "weights": item.get("weights", "pytorch_lora_weights.safetensors"),
-                "likes": item.get("likes", 0),
-                "downloads": item.get("downloads", 0),
-            }
-            for item in data
-        ]
-except FileNotFoundError:
-    # Default LoRAs if JSON file doesn't exist
+
+with open("flux_loras.json", "r") as file:
+    data = json.load(file)
     flux_loras_raw = [
         {
-            "image": "https://via.placeholder.com/300x300?text=LoRA+1",
-            "title": "Example LoRA 1",
-            "repo": "example/lora1",
-            "trigger_word": "style1",
-            "weights": "pytorch_lora_weights.safetensors",
-            "likes": 100,
-            "downloads": 500,
-        },
-        {
-            "image": "https://via.placeholder.com/300x300?text=LoRA+2", 
-            "title": "Example LoRA 2",
-            "repo": "example/lora2",
-            "trigger_word": "style2",
-            "weights": "pytorch_lora_weights.safetensors",
-            "likes": 80,
-            "downloads": 300,
+            "image": item["image"],
+            "title": item["title"],
+            "repo": item["repo"],
+            "trigger_word": item.get("trigger_word", ""),
+            "trigger_position": item.get("trigger_position", "prepend"),
+            "weights": item.get("weights", "pytorch_lora_weights.safetensors"),
         }
+        for item in data
     ]
 
 # Global variables for LoRA management