Update app.py

app.py

@@ -1,110 +1,14 @@
-import gradio as gr
-import spaces
-import torch
-
-
-#@spaces.GPU
-#def greet(n):
-#    print(zero.device) # <-- 'cuda:0' 🤗
-#    return f"Hello {zero + n} Tensor"
-
-#demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
-#demo.launch()
-
 import os
-import subprocess
-import urllib.request
-import torch
-zero = torch.Tensor([0]).cuda()
-print(zero.device) # <-- 'cpu' 🤔
-print("Torch version:", torch.__version__)
-if torch.cuda.is_available():
-    cap = torch.cuda.get_device_capability(0)
-    print("CUDA device capability:", cap)
-else:
-    print("CUDA is not available!")
-    
-import subprocess
-
-try:
-    result = subprocess.run(["nvidia-smi"], stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
-    print("nvidia-smi output:\n", result.stdout.decode())
-except Exception as e:
-    print("Error running nvidia-smi:", e)
-
-
-# --------- Step 1: Clone the Repository (if not already present) ---------
-if not os.path.exists("stable-diffusion-webui-forge"):
-    print("Cloning stable-diffusion-webui-forge repository...")
-    subprocess.run(["git", "clone", "https://github.com/lllyasviel/stable-diffusion-webui-forge"], check=True)
-else:
-    print("stable-diffusion-webui-forge repository already exists.")
-
-
+import sys
 
-# --------- Step 2: Define Directories and Ensure They Exist ---------
-BASE_DIR = os.getcwd()
-SD_UI_DIR = os.path.join(BASE_DIR, "stable-diffusion-webui-forge")
-STABLE_DIFF_DIR = os.path.join(SD_UI_DIR, "models", "Stable-diffusion")
-VAE_DIR = os.path.join(SD_UI_DIR, "models", "VAE")
-LORA_DIR = os.path.join(SD_UI_DIR, "models", "Lora")
+# Ensure the repository is in the Python path
+repo_path = os.path.join(os.getcwd(), "stable-diffusion-webui-forge")
+sys.path.insert(0, repo_path)
 
-os.makedirs(STABLE_DIFF_DIR, exist_ok=True)
-os.makedirs(VAE_DIR, exist_ok=True)
-os.makedirs(LORA_DIR, exist_ok=True)
-
-# --------- Step 3: Define Model URLs ---------
-EDEN_URL = "https://edenartlab-lfs.s3.amazonaws.com/models/checkpoints/Eden_SDXL.safetensors"
-JUG_URL = "https://huggingface.co/KamCastle/jugg/resolve/main/juggernaut_reborn.safetensors"
-SDXL_URL = "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors?download=true"
-FLUX1_DEV_URL = "https://huggingface.co/lllyasviel/flux1-dev-bnb-nf4/resolve/main/flux1-dev-bnb-nf4-v2.safetensors"
-VAE_URL = "https://huggingface.co/stabilityai/sdxl-vae/resolve/main/sdxl_vae.safetensors"
-LORA_FAE_URL = "https://huggingface.co/kayte0342/eden_fae/resolve/main/fairy_punk_sdxl_lora.safetensors?download=true"
-LORA_KAYTE_EDEN_URL = "https://huggingface.co/kayte0342/kayte_eden/resolve/main/kayte_sdxl_lora.safetensors?download=true"
-LORA_KAYTE_SD_URL = "https://huggingface.co/kayte0342/kayte_eden/resolve/main/kayteSD.safetensors?download=true"
-LORA_KAYTE_FLUX_URL = "https://huggingface.co/kayte0342/kayte_flux/resolve/main/kayte_epoch_3.safetensors?download=true"
-LORA_FAE_FLUX_URL = "https://huggingface.co/kayte0342/fairycore_flux/blob/main/fairy-core_epoch_4.safetensors"
-
-# --------- Helper Function to Download a File ---------
-def download_model(url, output_path):
-    if not os.path.exists(output_path):
-        print(f"Downloading {url} to {output_path}...")
-        urllib.request.urlretrieve(url, output_path)
-    else:
-        print(f"File {output_path} already exists.")
-
-# --------- Step 4: Download Models ---------
-# Stable Diffusion weights
-download_model(EDEN_URL, os.path.join(STABLE_DIFF_DIR, "Eden_SDXL.safetensors"))
-download_model(JUG_URL, os.path.join(STABLE_DIFF_DIR, "juggernaut_reborn.safetensors"))
-download_model(SDXL_URL, os.path.join(STABLE_DIFF_DIR, "stable-diffusion-XL.safetensors"))
-download_model(FLUX1_DEV_URL, os.path.join(STABLE_DIFF_DIR, "flux1-dev-bnb-nf4-v2.safetensors"))
-
-# VAE model
-download_model(VAE_URL, os.path.join(VAE_DIR, "sdxl_vae.safetensors"))
-
-# LoRA models
-download_model(LORA_FAE_URL, os.path.join(LORA_DIR, "fae_sdxl.safetensors"))
-download_model(LORA_KAYTE_EDEN_URL, os.path.join(LORA_DIR, "kayte_eden.safetensors"))
-download_model(LORA_KAYTE_SD_URL, os.path.join(LORA_DIR, "kayte_SD.safetensors"))
-download_model(LORA_KAYTE_FLUX_URL, os.path.join(LORA_DIR, "kayte_epoch_3.safetensors"))
-download_model(LORA_FAE_FLUX_URL, os.path.join(LORA_DIR, "fairy-core_epoch_4.safetensors"))
+try:
+    from launch import demo
+except ImportError:
+    raise ImportError("Could not import 'demo' from launch.py. Please check if the repository exposes a Gradio interface.")
 
-# --------- Step 5: Launch the Forge Web UI ---------
-# Get the port from the environment (Spaces sets PORT), default to 7860 if not set.
 port = int(os.environ.get("PORT", 7860))
-
-# Change directory to the stable-diffusion-webui-forge folder
-os.chdir(SD_UI_DIR)
-
-# Construct the command to launch the web UI.
-# Note: Remove the '--share' flag since Spaces provides a public URL.
-launch_command = [
-    "python", "launch.py",
-    f"--port={port}",
-    "--host=0.0.0.0"
-]
-
-print("Launching stable-diffusion-webui-forge with command:")
-print(" ".join(launch_command))
-subprocess.run(launch_command, check=True)
+demo.launch(server_name="0.0.0.0", server_port=port)