Update app.py

app.py

@@ -160,6 +160,13 @@ extensions_dir = webui_dir / "extensions"
 fluxdev_dir = extensions_dir / "Fluxdev"
 fluxschnell_dir = extensions_dir / "Fluxschnell"
 
+# モデルの保存先
+model_dir = webui_dir / "models" / "Stable-diffusion"
+
+# ダウンロードするモデルのURL（適宜変更）
+model_url = "https://civitai.com/api/download/models/39164"  
+model_path = model_dir / "backlighting.safetensors"
+
 # WebUI のディレクトリへ移動
 os.chdir(webui_dir)
 
@@ -171,6 +178,14 @@ if not fluxdev_dir.exists():
 if not fluxschnell_dir.exists():
     os.system(f"git clone https://github.com/FluxAi/Fluxschnell {fluxschnell_dir}")
 
+# モデルフォルダが存在しない場合は作成
+model_dir.mkdir(parents=True, exist_ok=True)
+
+# モデルをダウンロード（すでに存在しない場合のみ）
+if not model_path.exists():
+    os.system(f"wget {model_url} -O {model_path}")
+
+    
 
 print("Done\nStarting Webui...")
 os.chdir(user_home / r"stable-diffusion-webui")