Update app.py

app.py

@@ -57,7 +57,7 @@ def environment_setup(use_q6: bool):
         output.append(proc.stderr)
 
     # Clone ComfyUI if missing
-    if not os.path.isdir("/content/ComfyUI"):
+    if not os.path.isdir("./ComfyUI"):
         output.append("Cloning ComfyUI repo...")
         proc = subprocess.run("git clone https://github.com/Isi-dev/ComfyUI /content/ComfyUI", shell=True, capture_output=True, text=True)
         output.append(proc.stdout + proc.stderr)
@@ -76,9 +76,9 @@ def environment_setup(use_q6: bool):
         output.append("ComfyUI_GGUF repo already exists")
 
     # Ensure model directories exist
-    model_unet_dir = "/content/ComfyUI/models/unet"
-    text_enc_dir = "/content/ComfyUI/models/text_encoders"
-    vae_dir = "/content/ComfyUI/models/vae"
+    model_unet_dir = "./ComfyUI/models/unet"
+    text_enc_dir = "./ComfyUI/models/text_encoders"
+    vae_dir = "./ComfyUI/models/vae"
     os.makedirs(model_unet_dir, exist_ok=True)
     os.makedirs(text_enc_dir, exist_ok=True)
     os.makedirs(vae_dir, exist_ok=True)
@@ -110,7 +110,7 @@ def imports_initialization():
     global unet_loader, clip_loader, clip_encode_positive, clip_encode_negative
     global vae_loader, empty_latent_video, ksampler, vae_decode, save_webp, save_webm
 
-    sys.path.insert(0, '/content/ComfyUI')
+    sys.path.insert(0, './ComfyUI')
 
     from comfy import model_management
     from nodes import (
@@ -154,7 +154,7 @@ def clear_memory():
             del obj
     gc.collect()
 
-def save_as_mp4(images, filename_prefix, fps, output_dir="/content/ComfyUI/output"):
+def save_as_mp4(images, filename_prefix, fps, output_dir="./ComfyUI/output"):
     os.makedirs(output_dir, exist_ok=True)
     output_path = f"{output_dir}/{filename_prefix}.mp4"
     frames = [(img.cpu().numpy() * 255).astype(np.uint8) for img in images]
@@ -163,7 +163,7 @@ def save_as_mp4(images, filename_prefix, fps, output_dir="/content/ComfyUI/outpu
             writer.append_data(frame)
     return output_path
 
-def save_as_webp(images, filename_prefix, fps, quality=90, lossless=False, method=4, output_dir="/content/ComfyUI/output"):
+def save_as_webp(images, filename_prefix, fps, quality=90, lossless=False, method=4, output_dir="./ComfyUI/output"):
     os.makedirs(output_dir, exist_ok=True)
     output_path = f"{output_dir}/{filename_prefix}.webp"
     frames = [(img.cpu().numpy() * 255).astype(np.uint8) for img in images]
@@ -173,7 +173,7 @@ def save_as_webp(images, filename_prefix, fps, quality=90, lossless=False, metho
             writer.append_data(frame)
     return output_path
 
-def save_as_webm(images, filename_prefix, fps, codec="vp9", quality=32, output_dir="/content/ComfyUI/output"):
+def save_as_webm(images, filename_prefix, fps, codec="vp9", quality=32, output_dir="./ComfyUI/output"):
     os.makedirs(output_dir, exist_ok=True)
     output_path = f"{output_dir}/{filename_prefix}.webm"
     frames = [(img.cpu().numpy() * 255).astype(np.uint8) for img in images]
@@ -183,7 +183,7 @@ def save_as_webm(images, filename_prefix, fps, codec="vp9", quality=32, output_d
             writer.append_data(frame)
     return output_path
 
-def save_as_image(image, filename_prefix, output_dir="/content/ComfyUI/output"):
+def save_as_image(image, filename_prefix, output_dir="./ComfyUI/output"):
     os.makedirs(output_dir, exist_ok=True)
     output_path = f"{output_dir}/{filename_prefix}.png"
     frame = (image.cpu().numpy() * 255).astype(np.uint8)