Update app.py

app.py

@@ -149,20 +149,43 @@ DownLoad(r"https://civitai.com/api/download/models/39885",str(user_home / r"stab
 DownLoad(r"https://civitai.com/api/download/models/21065",str(user_home / r"stable-diffusion-webui" / r"extensions" / r"sd-webui-additional-networks" / r"models"/ r"lora"),r"LAS.safetensors")
 DownLoad(r"https://civitai.com/api/download/models/39164",str(user_home / r"stable-diffusion-webui" / r"extensions" / r"sd-webui-additional-networks" / r"models"/ r"lora"),r"backlighting.safetensors")
 
-# 環境変数 "token" から Hugging Face トークンを取得
+
+
+
+# Hugging Face トークン取得
 hf_token = os.getenv("token")
+if not hf_token:
+    print("Error: Hugging Face token not found.")
+    exit(1)
 
-# 削除処理をエラーなしで実行する
+# 削除処理
 model_path = "/home/user/stable-diffusion-webui/models/Stable-diffusion/flux1-dev.safetensors"
 if os.path.exists(model_path):
     os.remove(model_path)
-os.system("set COMMANDLINE_ARGS=--skip-torch-cuda-test --no-half --use-cpu all")
+
+# CPUモード設定（Linux向け修正）
+os.environ["COMMANDLINE_ARGS"] = "--skip-torch-cuda-test --no-half --use-cpu all"
 
 # wget でファイルをダウンロード
-cmd = f'wget --header="Authorization: Bearer {hf_token}" -O {model_path} "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors"'
+cmd = f'wget --header="Authorization: Bearer {hf_token}" -O "{model_path}" "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors"'
+print(f"Executing command: {cmd}")
 os.system(cmd)
-os.system("ls -lh /home/user/stable-diffusion-webui/models/Stable-diffusion/flux1-dev.safetensors")
 
+# ファイルサイズチェック
+if os.path.exists(model_path):
+    file_size = os.path.getsize(model_path)
+    if file_size < 1000:  # 1KB 未満なら異常
+        print(f"Error: {model_path} is too small ({file_size} bytes). Download failed.")
+        os.remove(model_path)
+        exit(1)
+    else:
+        print(f"Download successful. File size: {file_size} bytes.")
+else:
+    print("Error: File not found after download.")
+    exit(1)
+
+
+    
 print("Done\nStarting Webui...")
 os.chdir(user_home / r"stable-diffusion-webui")
 while True: