Updating to working code

app.py

@@ -1,60 +1,162 @@
 import os
 from subprocess import getoutput
 
-gpu_info = getoutput('nvidia-smi')
-if("A10G" in gpu_info):
-    os.system("pip install -q xformers==0.0.24.post1+cu121torch2.1 -f https://nenejjj.github.io/whl/")
-elif("T4" in gpu_info):
-    os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.15/xformers-0.0.15.dev0+1515f77.d20221130-cp38-cp38-linux_x86_64.whl")
+# Welcome to the world of python madness!
 
-os.system(f"git clone https://github.com/camenduru/stable-diffusion-webui /home/user/app/stable-diffusion-webui")
+# Debugging with Verbose mode on...
+os.system(f'echo "Checking Python Version"')
+os.system(f"python --version")
+
+os.system(f'echo "Starting verbose diagnostic setup for stable-diffusion-webui..."')
+
+# Clone repository
+os.system(f'echo "Cloning Stable Diffusion WebUI repository from darkstorm2150..."')
+os.system(f"git clone https://github.com/darkstorm2150/stable-diffusion-webui  /home/user/app/stable-diffusion-webui")
+
+# Set working directory
+os.system(f'echo "Setting working directory to stable-diffusion-webui root folder..."')
 os.chdir("/home/user/app/stable-diffusion-webui")
 
-os.system(f"wget -q https://github.com/camenduru/webui/raw/main/env_patch.py -O /home/user/app/env_patch.py")
-os.system(f'''sed -i "/import image_from_url_text/a import xformers\\nimport xformers.ops" /home/user/app/stable-diffusion-webui/modules/ui.py''')
-os.system(f'''sed -i -e '/import image_from_url_text/r /home/user/app/env_patch.py' /home/user/app/stable-diffusion-webui/modules/ui.py''')
+# clean requirements.txt
+#os.system(f'echo "Removing numpy and torch from requirements.txt to prevent version conflicts..."')
+#os.system(f"sed -i '/numpy/d' /home/user/app/stable-diffusion-webui/requirements.txt")
+#os.system(f"sed -i '/torch/d' /home/user/app/stable-diffusion-webui/requirements.txt")
+
+# Patch environment
+os.system(f'echo "Downloading environment patch script..."')
+os.system(f"wget https://raw.githubusercontent.com/darkstorm2150/webui/refs/heads/main/env_patch.py  -O /home/user/app/env_patch.py")
+os.system(f"sed -i -e '/import image_from_url_text/r /home/user/app/env_patch.py' /home/user/app/stable-diffusion-webui/modules/ui.py")
+
+# UI customization
+#os.system(f'echo "Removing Checkpoint Merger tab from UI..."')
+#os.system(f"sed -i -e '/(modelmerger_interface, \"Checkpoint Merger\", \"modelmerger\"),/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
+
+#os.system(f'echo "Removing Training tab from UI..."')
+#os.system(f"sed -i -e '/(train_interface, \"Train\", \"ti\"),/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
+
+#os.system(f'echo "Removing Extensions tab from UI..."')
+#os.system(f"sed -i -e '/extensions_interface, \"Extensions\", \"extensions\"/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
+
+#os.system(f'echo "Removing Settings tab from UI..."')
+#os.system(f"sed -i -e '/settings_interface, \"Settings\", \"settings\"/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
+
+# Gradio shadowRoot fix
+os.system(f'echo "Fixing Gradio shadowRoot access in script.js for Hugging Face compatibility..."')
+os.system(f'''sed -i -e "s/document.getElementsByTagName('gradio-app')\\[0\\].shadowRoot/!!document.getElementsByTagName('gradio-app')[0].shadowRoot ? document.getElementsByTagName('gradio-app')[0].shadowRoot : document/g" /home/user/app/stable-diffusion-webui/script.js''')
 
-os.system(f"sed -i -e '/(modelmerger_interface, \"Checkpoint Merger\", \"modelmerger\"),/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
-os.system(f"sed -i -e '/(train_interface, \"Train\", \"ti\"),/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
-os.system(f"sed -i -e '/extensions_interface, \"Extensions\", \"extensions\"/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
-os.system(f"sed -i -e '/settings_interface, \"Settings\", \"settings\"/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
-os.system(f'''sed -i -e "s/document.getElementsByTagName('gradio-app')\[0\].shadowRoot/!!document.getElementsByTagName('gradio-app')[0].shadowRoot ? document.getElementsByTagName('gradio-app')[0].shadowRoot : document/g" /home/user/app/stable-diffusion-webui/script.js''')
+# Progress bar settings
+os.system(f'echo "Enabling progress bars in UI for better user feedback..."')
 os.system(f"sed -i -e 's/                show_progress=False,/                show_progress=True,/g' /home/user/app/stable-diffusion-webui/modules/ui.py")
+
+# Queue settings
+os.system(f'echo "Modifying webui.py to enable queuing for async processing..."')
 os.system(f"sed -i -e 's/shared.demo.launch/shared.demo.queue().launch/g' /home/user/app/stable-diffusion-webui/webui.py")
-os.system(f"sed -i -e 's/inputs=\[component\],/&\\n                queue=False,/g' /home/user/app/stable-diffusion-webui/modules/ui.py")
-os.system(f"sed -i -e 's/outputs=\[token_counter\]/outputs=[token_counter], queue=False/g' /home/user/app/stable-diffusion-webui/modules/ui.py")
 
-# ----------------------------Please duplicate this space and delete this block if you don't want to see the extra header----------------------------
-os.system(f"wget -q https://github.com/camenduru/webui/raw/main/header_patch.py -O /home/user/app/header_patch.py")
+os.system(f'echo "Disabling queuing for input components to reduce latency..."')
+os.system(f"sed -i -e 's/inputs=\\[component\\],/&\\n                queue=False,/g' /home/user/app/stable-diffusion-webui/modules/ui.py")
+
+os.system(f'echo "Disabling queuing for token counter outputs..."')
+os.system(f"sed -i -e 's/outputs=\\[token_counter\\]/outputs=[token_counter], queue=False/g' /home/user/app/stable-diffusion-webui/modules/ui.py")
+
+
+
+# ----------------------------Patch UI Header for Hugging Face Spaces----------------------------
+os.system(f'echo "Downloading header_patch.py to customize UI header banners..."')
+os.system(f"wget https://raw.githubusercontent.com/darkstorm2150/webui/refs/heads/main/header_patch.py  -O /home/user/app/header_patch.py")
+
+os.system(f'echo "Injecting header_patch.py content into modules/ui.py under the demo section..."')
 os.system(f"sed -i -e '/demo:/r /home/user/app/header_patch.py' /home/user/app/stable-diffusion-webui/modules/ui.py")
-# ---------------------------------------------------------------------------------------------------------------------------------------------------
+# ---------------------------------------------------------------------------------------------
 
-# ------------------------------------ OpenGen Model ----------------------------------------------------------------------
 
 
-os.system(f"wget -q https://huggingface.co/darkstorm2150/OpenGen/resolve/main/OpenGen%20v1.0.safetensors -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/OpenGen_v1.0.safetensors")
+# Handle shared vs. non-shared UI setup with verbose logging
+if "IS_SHARED_UI" in os.environ:
+    os.system(f'echo "IS_SHARED_UI IS ENABLED: Loading shared configurations..."')
+    
+    # Download shared UI configuration files
+    os.system(f"echo 'Downloading shared-config.json for UI settings...'")
+    os.system(f"wget https://raw.githubusercontent.com/darkstorm2150/webui/refs/heads/main/shared-config.json  -O /home/user/app/shared-config.json")
+    
+    os.system(f"echo 'Downloading shared-ui-config.json for layout preferences...'") 
+    os.system(f"wget https://raw.githubusercontent.com/darkstorm2150/webui/refs/heads/main/shared-ui-config.json  -O /home/user/app/shared-ui-config.json")
 
-# ---------------------------------------------------------------------------------------------------------------------------------------------------
+    # Launch WebUI with shared settings
+    os.system(f'echo "Starting WebUI with shared configurations..."')
+    os.system(f"python launch.py --disable-console-progressbars --enable-console-prompts --ui-config-file /home/user/app/shared-ui-config.json --ui-settings-file /home/user/app/shared-config.json --cors-allow-origins huggingface.co,hf.space --no-progressbar-hiding")
+else:
+    os.system(f'echo "IS_SHARED_UI IS DISABLED: Setting up custom environment..."')
+    
+    # Download custom scripts
+    os.system(f'echo "Installing run_n_times.py script for batch processing..."')
+    os.system(f"wget https://raw.githubusercontent.com/darkstorm2150/webui/refs/heads/main/run_n_times.py  -O /home/user/app/stable-diffusion-webui/scripts/run_n_times.py")
+    
+    # Install extensions (commented lines can be activated by removing #)
+    os.system(f'echo "Installing image browser extension for managing outputs..."')
+    os.system(f"git clone https://github.com/yfszzx/stable-diffusion-webui-images-browser  /home/user/app/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser")
+    
+    os.system(f'echo "Installing Deforum animation extension..."')
+    os.system(f"git clone https://github.com/deforum-art/deforum-for-automatic1111-webui  /home/user/app/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui")
+    
+    # Load Protogen model (example custom model)
+    os.system(f'echo "Downloading Protogen X3.4 Stable Diffusion model..."')
+    os.system(f"wget -q https://huggingface.co/darkstorm2150/OpenGen/resolve/main/OpenGen%20v1.0.safetensors  -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/OpenGen_v1.0.safetensors")
+    
+    # Critical dependency installation sequence
+    #os.system(f'echo "starting dependency cleanup..."')
+    #os.system(f'pip cache purge')
 
-# Please duplicate this space and delete # character in front of the custom script you want to use or add here more custom scripts with same structure os.system(f"wget -q https://CUSTOM_SCRIPT_URL -O /home/user/app/stable-diffusion-webui/scripts/CUSTOM_SCRIPT_NAME.py")
-os.system(f"wget -q https://gist.github.com/camenduru/9ec5f8141db9902e375967e93250860f/raw/d0bcf01786f20107c329c03f8968584ee67be12a/run_n_times.py -O /home/user/app/stable-diffusion-webui/scripts/run_n_times.py")
+    # Running multiple times to remove stubborn copies
+    #os.system(f'echo "Removing all existing numpy versions..."')
+    #os.system(f'pip uninstall -y numpy numpy numpy')  
     
-# Please duplicate this space and delete # character in front of the extension you want to use or add here more extensions with same structure os.system(f"git clone https://EXTENSION_GIT_URL /home/user/app/stable-diffusion-webui/extensions/EXTENSION_NAME")
-#os.system(f"git clone https://github.com/camenduru/stable-diffusion-webui-artists-to-study /home/user/app/stable-diffusion-webui/extensions/stable-diffusion-webui-artists-to-study")
-os.system(f"git clone https://github.com/yfszzx/stable-diffusion-webui-images-browser /home/user/app/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser")
-os.system(f"git clone https://github.com/deforum-art/deforum-for-automatic1111-webui /home/user/app/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui")
+    # Force NumPy 1.26.4 first
+    #os.system(f'echo "Installing NumPy 1.26.4..."')
+    #os.system(f"pip install --no-cache-dir numpy==1.26.4")
+
+    # Verify NumPy Version 1.26.4
+    #os.system(f"python -c 'import numpy; print(numpy.__version__)'")
+
+    # Install PyTorch + xformers with CUDA compatibility
+    #os.system(f'echo "Installing PyTorch..."')
+    #os.system(f"pip install torch==2.1.2+cu121 torchvision==0.16.2+cu121 --extra-index-url https://download.pytorch.org/whl/cu121 ")
+
+    #os.system(f'echo "Installing xformers..."')
+    #os.system(f"pip install --no-cache-dir xformers==0.0.25 --extra-index-url https://download.pytorch.org/whl/cu118")
+
+    os.system(f'echo "Installing build dependencies..."')
+    os.system(f'pip install ninja packaging wheel')
+    os.system(f'apt-get update && apt-get install -y build-essential cmake git')
     
-# Please duplicate this space and delete # character in front of the model you want to use or add here more ckpts with same structure os.system(f"wget -q https://CKPT_URL -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/CKPT_NAME.ckpt")
-#os.system(f"wget -q https://huggingface.co/nitrosocke/Arcane-Diffusion/resolve/main/arcane-diffusion-v3.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/arcane-diffusion-v3.ckpt")
-#os.system(f"wget -q https://huggingface.co/DGSpitzer/Cyberpunk-Anime-Diffusion/resolve/main/Cyberpunk-Anime-Diffusion.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/Cyberpunk-Anime-Diffusion.ckpt")
-#os.system(f"wget -q https://huggingface.co/prompthero/midjourney-v4-diffusion/resolve/main/mdjrny-v4.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/mdjrny-v4.ckpt")
-#os.system(f"wget -q https://huggingface.co/nitrosocke/mo-di-diffusion/resolve/main/moDi-v1-pruned.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/moDi-v1-pruned.ckpt")
-#os.system(f"wget -q https://huggingface.co/Fictiverse/Stable_Diffusion_PaperCut_Model/resolve/main/PaperCut_v1.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/PaperCut_v1.ckpt")
-#os.system(f"wget -q https://huggingface.co/lilpotat/sa/resolve/main/samdoesarts_style.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/samdoesarts_style.ckpt")
-#os.system(f"wget -q https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-float32.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/wd-v1-3-float32.ckpt")
-#os.system(f"wget -q https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/sd-v1-4.ckpt")
-#os.system(f"wget -q https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned-emaonly.ckpt")
-#os.system(f"wget -q https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/sd-v1-5-inpainting.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/sd-v1-5-inpainting.ckpt")
-#os.system(f"wget -q https://huggingface.co/stabilityai/stable-diffusion-2/resolve/main/768-v-ema.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/768-v-ema.ckpt")
-#os.system(f"wget -q https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/768-v-ema.yaml")
-os.system(f"python launch.py --force-enable-xformers --ui-config-file /home/user/app/ui-config.json --ui-settings-file /home/user/app/config.json --disable-console-progressbars --enable-console-prompts --cors-allow-origins huggingface.co,hf.space --no-progressbar-hiding --api --skip-torch-cuda-test")
+    # Building xformers from source
+    os.system(f'echo "Cloning xformers repo..."')
+    os.system(f'git clone https://github.com/facebookresearch/xformers.git  /home/user/app/xformers')
+    os.system("cd /home/user/app/xformers && git submodule update --init --recursive")
+
+    # Change directory to xformers and install
+    os.chdir('/home/user/app/xformers')
+    os.system('pip install --no-cache-dir -e .')
+
+    # Final verification
+    os.system(f'echo "Verifying NumPy and PyTorch..."')
+    
+    os.system("python -c 'import numpy; print(\"NumPy:\", numpy.__version__)'")
+    os.system("python -c 'import torch; print(\"PyTorch:\", torch.__version__)'")
+
+    # Explicitly verify xformers import
+    os.system("python -c 'import xformers; print(\"xformers version:\", xformers.__version__)'")
+    
+    # Launch WebUI with custom settings
+    os.system(f'echo "Starting WebUI..."')
+
+    # reinstalling libraries
+    os.system(f'pip uninstall -y torch torchvision xformers')
+    os.system(f'pip install torch==2.7.1+cu126 torchvision==0.18.1+cu126 --extra-index-url https://download.pytorch.org/whl/cu126')
+    os.system(f'pip install xformers==0.0.32+0f0bb9d9.d20250720 --extra-index-url https://download.pytorch.org/whl/cu126')
+
+    os.system(f'pip install -U ninja cmake pybind11')
+
+    # starting xformers with WebUI...
+    os.chdir('/home/user/app/stable-diffusion-webui')
+    os.system("python launch.py --ui-config-file /home/user/app/shared-ui-config.json --ui-settings-file /home/user/app/shared-config.json --disable-console-progressbars --enable-console-prompts --cors-allow-origins huggingface.co,hf.space --no-progressbar-hiding --api --skip-torch-cuda-test")
+