Upload folder using huggingface_hub

SD3-Batch-Creation/.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

SD3-Batch-Creation/README.md

@@ -0,0 +1,13 @@
+---
+title: SD3 Batch Creation
+emoji: 🌍
+colorFrom: gray
+colorTo: pink
+sdk: gradio
+sdk_version: 4.36.1
+app_file: app.py
+pinned: false
+license: gpl
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -10,110 +10,77 @@ import zipfile
 from typing import List
 
 # Define constants
-WORKFLOW_FILENAME = 'sd3.json'
-ROOT_DIR = os.getcwd()  # Root directory of the project
-OUTPUT_DIR = os.path.join(ROOT_DIR, 'output')
-USER_OUTPUTS_DIR = os.path.join(OUTPUT_DIR, 'user_outputs')
-URL = "https://3653-47-216-153-41.ngrok-free.app/prompt"  # Replace with your ngrok URL
-
-# Create output directories if they don't exist
-os.makedirs(USER_OUTPUTS_DIR, exist_ok=True)
+COMFYUI_DIR = r"C:\Users\bewiz\OneDrive\Desktop\AI\ComfyUI_windows_portable\ComfyUI"
+WORKFLOW_PATH = os.path.join(COMFYUI_DIR, 'sd3.json')
+OUTPUT_DIR = os.path.join(COMFYUI_DIR, 'output')
+URL = "http://127.0.0.1:8188/prompt"
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # Shared state for cancellation
 cancel_processing = False
-log_messages = ""
-
-def log_message(message):
-    global log_messages
-    log_messages += message + "\n"
-    logging.info(message)
-
-def find_workflow_in_root(root_dir, filename):
-    log_message(f"Searching for {filename} in root directory {root_dir}")
-    for root, _, files in os.walk(root_dir):
-        if filename in files:
-            file_path = os.path.join(root, filename)
-            log_message(f"Found workflow file at {file_path}")
-            return file_path
-    log_message(f"Workflow file {filename} not found in root directory {root_dir}")
-    return None
 
 def read_workflow(file_path):
-    log_message(f"Reading workflow from {file_path}")
-    try:
-        with open(file_path, 'r') as file:
-            workflow = json.load(file)
-        return workflow
-    except Exception as e:
-        log_message(f"Error reading workflow: {e}")
-        return None
+    logging.info(f"Reading workflow from {file_path}")
+    with open(file_path, 'r') as file:
+        workflow = json.load(file)
+    return workflow
 
 def update_workflow(workflow, prompt, negative_prompt):
-    log_message(f"Updating workflow with new prompts: {prompt}, negative: {negative_prompt}")
+    logging.info(f"Updating workflow with new prompts: {prompt}, negative: {negative_prompt}")
     workflow["6"]["inputs"]["text"] = prompt
     workflow["71"]["inputs"]["text"] = negative_prompt
     return workflow
 
 def write_workflow(workflow, file_path):
-    log_message(f"Writing updated workflow to {file_path}")
-    try:
-        with open(file_path, 'w') as file:
-            json.dump(workflow, file, indent=4)
-    except Exception as e:
-        log_message(f"Error writing workflow: {e}")
+    logging.info(f"Writing updated workflow to {file_path}")
+    with open(file_path, 'w') as file:
+        json.dump(workflow, file, indent=4)
 
 def send_workflow_to_comfyui(workflow, url):
     headers = {"Content-Type": "application/json"}
     try:
         response = requests.post(url, headers=headers, json={"prompt": workflow})
         response.raise_for_status()
-        log_message(f"Workflow sent successfully: {response.status_code}")
-        log_message(f"Response content: {response.content}")
+        logging.info(f"Workflow sent successfully: {response.status_code}")
+        logging.debug(f"Response content: {response.content}")
     except requests.exceptions.RequestException as e:
-        log_message(f"Error sending workflow to ComfyUI: {e}")
+        logging.error(f"Error sending workflow to ComfyUI: {e}")
         raise
 
 def monitor_output_images(output_dir, previous_images, timeout=60):
     start_time = time.time()
-    log_message(f"Monitoring {output_dir} for new images...")
+    logging.info(f"Monitoring {output_dir} for new images...")
     while time.time() - start_time < timeout:
         current_images = os.listdir(output_dir)
         new_images = [img for img in current_images if img not in previous_images]
         if new_images:
             latest_image = new_images[-1]
-            log_message(f"New image found: {latest_image}")
+            logging.info(f"New image found: {latest_image}")
             return latest_image
         time.sleep(1)
-    log_message(f"Timeout while waiting for new images in {output_dir}")
+    logging.info(f"Timeout while waiting for new images in {output_dir}")
     return None
 
 def copy_file_with_retry(src, dst_dir, file_index, retries=5, delay=1):
     dst = os.path.join(dst_dir, f"SD3_{file_index:05d}.png")
-    log_message(f"Attempting to copy file from {src} to {dst}")
     for _ in range(retries):
         try:
             shutil.copy(src, dst)
-            log_message(f"File copied successfully to {dst}")
             return dst
         except PermissionError:
             time.sleep(delay)
-            log_message(f"PermissionError encountered. Retrying...")
-    log_message(f"Failed to copy {src} to {dst} after {retries} retries")
     raise PermissionError(f"Failed to copy {src} to {dst} after {retries} retries")
 
 def zip_files(output_images: List[str], zip_interval: int, zip_folder: str):
     zip_files = []
     for i in range(0, len(output_images), zip_interval):
         zip_filename = os.path.join(zip_folder, f"images_{i//zip_interval + 1}_{time.time_ns()}.zip")
-        log_message(f"Creating zip file {zip_filename}")
         with zipfile.ZipFile(zip_filename, 'w') as zipf:
             for img in output_images[i:i+zip_interval]:
                 zipf.write(img, os.path.basename(img))
         zip_files.append(zip_filename)
-    log_message(f"Zip files created: {zip_files}")
     return zip_files
 
 def process_prompts(prompts_text, negative_prompt_text, user_folder, zip_interval):
@@ -124,15 +91,7 @@ def process_prompts(prompts_text, negative_prompt_text, user_folder, zip_interva
     zip_files_list = []
     file_index = 1
 
-    workflow_path = find_workflow_in_root(ROOT_DIR, WORKFLOW_FILENAME)
-    if workflow_path is None:
-        log_message("Workflow file not found. Exiting process_prompts.")
-        return [], []
-
-    workflow = read_workflow(workflow_path)
-    if workflow is None:
-        log_message("Workflow is None, exiting process_prompts.")
-        return [], []
+    workflow = read_workflow(WORKFLOW_PATH)
 
     total_prompts = len(prompts)
     previous_images = os.listdir(OUTPUT_DIR)
@@ -142,7 +101,7 @@ def process_prompts(prompts_text, negative_prompt_text, user_folder, zip_interva
     try:
         for i, prompt in enumerate(prompts):
             if cancel_processing:
-                log_message("Processing cancelled by user.")
+                logging.info("Processing cancelled by user.")
                 break
             if not prompt.strip():
                 continue
@@ -150,12 +109,12 @@ def process_prompts(prompts_text, negative_prompt_text, user_folder, zip_interva
             negative_prompt = negative_prompts[i] if i < len(negative_prompts) else ""
 
             updated_workflow = update_workflow(workflow, prompt, negative_prompt)
-            write_workflow(updated_workflow, workflow_path)
+            write_workflow(updated_workflow, WORKFLOW_PATH)
 
-            log_message(f"Updated workflow: {json.dumps(updated_workflow, indent=4)}")
+            logging.debug(f"Updated workflow: {json.dumps(updated_workflow, indent=4)}")
 
             send_workflow_to_comfyui(updated_workflow, URL)
-            log_message(f"Sent workflow to ComfyUI for prompt {i + 1}/{total_prompts}")
+            logging.info(f"Sent workflow to ComfyUI for prompt {i + 1}/{total_prompts}")
 
             new_image = None
             retries = 0
@@ -163,22 +122,21 @@ def process_prompts(prompts_text, negative_prompt_text, user_folder, zip_interva
                 new_image = monitor_output_images(OUTPUT_DIR, previous_images)
                 if new_image is None:
                     retries += 1
-                    log_message(f"Retrying ({retries}/5)...")
+                    logging.warning(f"Retrying ({retries}/5)...")
                     time.sleep(5)
                 else:
                     time.sleep(2)
 
             if new_image is None:
-                log_message("Error monitoring output images: Timed out waiting for new image.")
+                logging.error("Error monitoring output images: Timed out waiting for new image.")
                 continue
 
             new_image_path = os.path.join(OUTPUT_DIR, new_image)
-            log_message(f"New image generated: {new_image_path}")
             try:
                 copied_image_path = copy_file_with_retry(new_image_path, user_folder, file_index)
-                log_message(f"New image generated and copied to user folder: {copied_image_path}")
+                logging.info(f"New image generated and copied to user folder: {copied_image_path}")
             except PermissionError as e:
-                log_message(f"Failed to copy file after retries: {e}")
+                logging.error(f"Failed to copy file after retries: {e}")
                 continue
 
             output_images.append(copied_image_path)
@@ -192,7 +150,7 @@ def process_prompts(prompts_text, negative_prompt_text, user_folder, zip_interva
                 zip_files_list.extend(new_zip_files)
 
             logs += f"Processed {i + 1}/{total_prompts} - Done: {i + 1}, Left: {total_prompts - (i + 1)}\n"
-            yield output_images, zip_files_list, logs + log_messages
+            yield output_images, zip_files_list, logs
 
         if cancel_processing or (len(output_images) % zip_interval != 0):
             zip_folder = os.path.join(user_folder, "zipped_images")
@@ -200,8 +158,8 @@ def process_prompts(prompts_text, negative_prompt_text, user_folder, zip_interva
             new_zip_files = zip_files(output_images, zip_interval, zip_folder)
             zip_files_list.extend(new_zip_files)
 
-    except Exception as e:
-        log_message(f"Exception during prompt processing: {e}")
+    except KeyboardInterrupt:
+        logging.info("Script interrupted by user.")
 
     return output_images, zip_files_list
 
@@ -283,9 +241,9 @@ def main():
                 logs_output = gr.Textbox(lines=10, interactive=False, label="Logs")
 
                 def generate_user_folder():
-                    user_folder = os.path.normpath(os.path.join(USER_OUTPUTS_DIR, f'SD3{random.randint(1000, 9999)}'))
+                    user_folder = os.path.normpath(os.path.join(OUTPUT_DIR, f'SD3{random.randint(1000, 9999)}'))
                     os.makedirs(user_folder, exist_ok=True)
-                    log_message(f"Generated user folder: {user_folder}")
+                    logging.info(f"Generated user folder: {user_folder}")
                     return user_folder
 
                 def on_click(prompts_text, negative_prompts_text, zip_interval):
@@ -293,16 +251,12 @@ def main():
                     user_folder = generate_user_folder()
                     output_images, zip_files_list = [], []
                     logs = ""
-                    try:
-                        for images, zip_files, log_msg in process_prompts(prompts_text, negative_prompts_text, user_folder, zip_interval):
-                            output_images = images
-                            zip_files_list = zip_files
-                            logs = log_msg
-                            yield images, zip_files_list, logs + log_messages
-                    except Exception as e:
-                        log_message(f"Error during prompt processing: {e}")
-                        logs += f"Error: {e}\n"
-                        yield output_images, zip_files_list, logs + log_messages
+                    for images, zip_files, log_msg in process_prompts(prompts_text, negative_prompts_text, user_folder, zip_interval):
+                        output_images = images
+                        zip_files_list = zip_files
+                        logs = log_msg
+                        yield images, zip_files_list, logs
+                    return output_images, zip_files_list, logs
 
                 process_btn.click(
                     fn=on_click,
@@ -316,7 +270,7 @@ def main():
                     outputs=[]
                 )
 
-    demo.launch()
+    demo.launch(share=True)
 
 if __name__ == "__main__":
     main()

sd3.json

@@ -1,187 +1,276 @@
-{
-  "6": {
-    "inputs": {
-      "text": "\"A graffiti art comparison featuring the word 'Creativity' across three different styles. First, SD1.5: a simple graffiti-style text on a plain urban wall, with basic lettering and minimal background details. BAD QUALITY.. labeled in words \"SD 1.5\" \n\nSecond, SDXL: detailed graffiti art on a vibrant wall, with bold colors, stylish fonts, and some 3D elements that add depth and creativity. GOOD QUALITY - LABELED in words \"SDXL\"\n\nThird, SD3: ultra-realistic graffiti with dynamic and intricate textures, shadows, and depth. The word 'Creativity' appears to pop out from the wall, with advanced 3D effects and lifelike details, creating a stunning visual impact.\" AMAZING QUALITY TEXT \"SD3\"\n\n",
-      "clip": [
-        "11",
-        0
-      ]
-    },
-    "class_type": "CLIPTextEncode",
-    "_meta": {
-      "title": "CLIP Text Encode (Prompt)"
+import os
+import json
+import requests
+import gradio as gr
+import random
+import time
+import logging
+import shutil
+import zipfile
+from typing import List
+
+# Define constants
+COMFYUI_DIR = r"C:\Users\bewiz\OneDrive\Desktop\AI\ComfyUI_windows_portable\ComfyUI"
+WORKFLOW_PATH = os.path.join(COMFYUI_DIR, 'sd3.json')
+OUTPUT_DIR = os.path.join(COMFYUI_DIR, 'output')
+URL = "http://127.0.0.1:8188/prompt"
+
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+# Shared state for cancellation
+cancel_processing = False
+
+def read_workflow(file_path):
+    logging.info(f"Reading workflow from {file_path}")
+    with open(file_path, 'r') as file:
+        workflow = json.load(file)
+    return workflow
+
+def update_workflow(workflow, prompt, negative_prompt):
+    logging.info(f"Updating workflow with new prompts: {prompt}, negative: {negative_prompt}")
+    workflow["6"]["inputs"]["text"] = prompt
+    workflow["71"]["inputs"]["text"] = negative_prompt
+    return workflow
+
+def write_workflow(workflow, file_path):
+    logging.info(f"Writing updated workflow to {file_path}")
+    with open(file_path, 'w') as file:
+        json.dump(workflow, file, indent=4)
+
+def send_workflow_to_comfyui(workflow, url):
+    headers = {"Content-Type": "application/json"}
+    try:
+        response = requests.post(url, headers=headers, json={"prompt": workflow})
+        response.raise_for_status()
+        logging.info(f"Workflow sent successfully: {response.status_code}")
+        logging.debug(f"Response content: {response.content}")
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Error sending workflow to ComfyUI: {e}")
+        raise
+
+def monitor_output_images(output_dir, previous_images, timeout=60):
+    start_time = time.time()
+    logging.info(f"Monitoring {output_dir} for new images...")
+    while time.time() - start_time < timeout:
+        current_images = os.listdir(output_dir)
+        new_images = [img for img in current_images if img not in previous_images]
+        if new_images:
+            latest_image = new_images[-1]
+            logging.info(f"New image found: {latest_image}")
+            return latest_image
+        time.sleep(1)
+    logging.info(f"Timeout while waiting for new images in {output_dir}")
+    return None
+
+def copy_file_with_retry(src, dst_dir, file_index, retries=5, delay=1):
+    dst = os.path.join(dst_dir, f"SD3_{file_index:05d}.png")
+    for _ in range(retries):
+        try:
+            shutil.copy(src, dst)
+            return dst
+        except PermissionError:
+            time.sleep(delay)
+    raise PermissionError(f"Failed to copy {src} to {dst} after {retries} retries")
+
+def zip_files(output_images: List[str], zip_interval: int, zip_folder: str):
+    zip_files = []
+    for i in range(0, len(output_images), zip_interval):
+        zip_filename = os.path.join(zip_folder, f"images_{i//zip_interval + 1}_{time.time_ns()}.zip")
+        with zipfile.ZipFile(zip_filename, 'w') as zipf:
+            for img in output_images[i:i+zip_interval]:
+                zipf.write(img, os.path.basename(img))
+        zip_files.append(zip_filename)
+    return zip_files
+
+def process_prompts(prompts_text, negative_prompt_text, user_folder, zip_interval):
+    global cancel_processing
+    prompts = [line.strip() for line in prompts_text.split('\n\n') if line.strip()]
+    negative_prompts = [line.strip() for line in negative_prompt_text.split('\n') if line.strip()]
+    output_images = []
+    zip_files_list = []
+    file_index = 1
+
+    workflow = read_workflow(WORKFLOW_PATH)
+
+    total_prompts = len(prompts)
+    previous_images = os.listdir(OUTPUT_DIR)
+
+    logs = ""
+
+    try:
+        for i, prompt in enumerate(prompts):
+            if cancel_processing:
+                logging.info("Processing cancelled by user.")
+                break
+            if not prompt.strip():
+                continue
+
+            negative_prompt = negative_prompts[i] if i < len(negative_prompts) else ""
+
+            updated_workflow = update_workflow(workflow, prompt, negative_prompt)
+            write_workflow(updated_workflow, WORKFLOW_PATH)
+
+            logging.debug(f"Updated workflow: {json.dumps(updated_workflow, indent=4)}")
+
+            send_workflow_to_comfyui(updated_workflow, URL)
+            logging.info(f"Sent workflow to ComfyUI for prompt {i + 1}/{total_prompts}")
+
+            new_image = None
+            retries = 0
+            while new_image is None and retries < 5:
+                new_image = monitor_output_images(OUTPUT_DIR, previous_images)
+                if new_image is None:
+                    retries += 1
+                    logging.warning(f"Retrying ({retries}/5)...")
+                    time.sleep(5)
+                else:
+                    time.sleep(2)
+
+            if new_image is None:
+                logging.error("Error monitoring output images: Timed out waiting for new image.")
+                continue
+
+            new_image_path = os.path.join(OUTPUT_DIR, new_image)
+            try:
+                copied_image_path = copy_file_with_retry(new_image_path, user_folder, file_index)
+                logging.info(f"New image generated and copied to user folder: {copied_image_path}")
+            except PermissionError as e:
+                logging.error(f"Failed to copy file after retries: {e}")
+                continue
+
+            output_images.append(copied_image_path)
+            previous_images.append(new_image)
+            file_index += 1
+
+            if len(output_images) % zip_interval == 0 and not cancel_processing:
+                zip_folder = os.path.join(user_folder, "zipped_images")
+                os.makedirs(zip_folder, exist_ok=True)
+                new_zip_files = zip_files(output_images[-zip_interval:], zip_interval, zip_folder)
+                zip_files_list.extend(new_zip_files)
+
+            logs += f"Processed {i + 1}/{total_prompts} - Done: {i + 1}, Left: {total_prompts - (i + 1)}\n"
+            yield output_images, zip_files_list, logs
+
+        if cancel_processing or (len(output_images) % zip_interval != 0):
+            zip_folder = os.path.join(user_folder, "zipped_images")
+            os.makedirs(zip_folder, exist_ok=True)
+            new_zip_files = zip_files(output_images, zip_interval, zip_folder)
+            zip_files_list.extend(new_zip_files)
+
+    except KeyboardInterrupt:
+        logging.info("Script interrupted by user.")
+
+    return output_images, zip_files_list
+
+def cancel_processing_fn():
+    global cancel_processing
+    cancel_processing = True
+
+def reset_cancel_processing_fn():
+    global cancel_processing
+    cancel_processing = False
+
+def main():
+    with gr.Blocks(css="""
+    .gradio-container {font-family: Arial, sans-serif;}
+    .psychedelic-text span {
+        animation: colorchange 10s infinite;
     }
-  },
-  "11": {
-    "inputs": {
-      "clip_name1": "clip_g.safetensors",
-      "clip_name2": "clip_l.safetensors",
-      "clip_name3": "t5xxl_fp16.safetensors"
-    },
-    "class_type": "TripleCLIPLoader",
-    "_meta": {
-      "title": "TripleCLIPLoader"
+    @keyframes colorchange {
+        0% { color: #ff69b4; }
+        10% { color: #ba55d3; }
+        20% { color: #7b68ee; }
+        30% { color: #00bfff; }
+        40% { color: #3cb371; }
+        50% { color: #ffff54; }
+        60% { color: #ffa500; }
+        70% { color: #ff4500; }
+        80% { color: #ff1493; }
+        90% { color: #da70d6; }
+        100% { color: #ff69b4; }
     }
-  },
-  "13": {
-    "inputs": {
-      "shift": 3,
-      "model": [
-        "252",
-        0
-      ]
-    },
-    "class_type": "ModelSamplingSD3",
-    "_meta": {
-      "title": "ModelSamplingSD3"
+    .image-container img {
+        width: 250px;
+        height: 250px;
     }
-  },
-  "67": {
-    "inputs": {
-      "conditioning": [
-        "71",
-        0
-      ]
-    },
-    "class_type": "ConditioningZeroOut",
-    "_meta": {
-      "title": "ConditioningZeroOut"
-    }
-  },
-  "68": {
-    "inputs": {
-      "start": 0.1,
-      "end": 1,
-      "conditioning": [
-        "67",
-        0
-      ]
-    },
-    "class_type": "ConditioningSetTimestepRange",
-    "_meta": {
-      "title": "ConditioningSetTimestepRange"
-    }
-  },
-  "69": {
-    "inputs": {
-      "conditioning_1": [
-        "68",
-        0
-      ],
-      "conditioning_2": [
-        "70",
-        0
-      ]
-    },
-    "class_type": "ConditioningCombine",
-    "_meta": {
-      "title": "Conditioning (Combine)"
-    }
-  },
-  "70": {
-    "inputs": {
-      "start": 0,
-      "end": 0.4533332519531249,
-      "conditioning": [
-        "71",
-        0
-      ]
-    },
-    "class_type": "ConditioningSetTimestepRange",
-    "_meta": {
-      "title": "ConditioningSetTimestepRange"
-    }
-  },
-  "71": {
-    "inputs": {
-      "text": "bad quality, poor quality, doll, disfigured, jpg, toy, bad anatomy, missing limbs, missing fingers, 3d, cgi",
-      "clip": [
-        "11",
-        0
-      ]
-    },
-    "class_type": "CLIPTextEncode",
-    "_meta": {
-      "title": "CLIP Text Encode (Negative Prompt)"
-    }
-  },
-  "135": {
-    "inputs": {
-      "width": 1024,
-      "height": 1024,
-      "batch_size": 1
-    },
-    "class_type": "EmptySD3LatentImage",
-    "_meta": {
-      "title": "EmptySD3LatentImage"
-    }
-  },
-  "231": {
-    "inputs": {
-      "samples": [
-        "271",
-        0
-      ],
-      "vae": [
-        "252",
-        2
-      ]
-    },
-    "class_type": "VAEDecode",
-    "_meta": {
-      "title": "VAE Decode"
-    }
-  },
-  "252": {
-    "inputs": {
-      "ckpt_name": "3\\sd3_medium_incl_clips_t5xxlfp8.safetensors"
-    },
-    "class_type": "CheckpointLoaderSimple",
-    "_meta": {
-      "title": "Load Checkpoint"
-    }
-  },
-  "271": {
-    "inputs": {
-      "seed": 945512652412924,
-      "steps": 28,
-      "cfg": 4.5,
-      "sampler_name": "dpmpp_2m",
-      "scheduler": "sgm_uniform",
-      "denoise": 1,
-      "model": [
-        "13",
-        0
-      ],
-      "positive": [
-        "6",
-        0
-      ],
-      "negative": [
-        "69",
-        0
-      ],
-      "latent_image": [
-        "135",
-        0
-      ]
-    },
-    "class_type": "KSampler",
-    "_meta": {
-      "title": "KSampler"
-    }
-  },
-  "273": {
-    "inputs": {
-      "filename_prefix": "SD3",
-      "images": [
-        "231",
-        0
-      ]
-    },
-    "class_type": "SaveImage",
-    "_meta": {
-      "title": "Save Image"
-    }
-  }
-}
+    """) as demo:
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown("### beWiZ's GroOvy SD3 Batch Imagine")
+                gr.HTML('<div class="image-container"><img src="https://raw.githubusercontent.com/downlifted/Groovy-StyleSuite/main/groovy.png" alt="GroOvy - SD3 Batch Imagine Logo"></div>')
+                with gr.Accordion("Developer Information", open=False):
+                    gr.Markdown("### Made by BeWiZ")
+                    gr.Markdown('<div class="image-container"><a href="https://twitter.com/AiAnarchist"><img src="https://raw.githubusercontent.com/downlifted/pictoprompt/master/aia.png" alt="BeWiZ Logo"></a></div>')
+                    gr.Markdown("Contact: [downlifted@gmail.com](mailto:downlifted@gmail.com)")
+                    gr.Markdown("Twitter: [@AiAnarchist](https://x.com/AiAnarchist)")
+                with gr.Accordion("About SD3 Batch Imagine", open=False):
+                    gr.Markdown("""
+                    ### SD3 Batch Imagine: Batch Image Generation
+                    Produce large batches of images using the latest SD3 Medium model. This tool allows you to generate images quickly and efficiently.
+                    - **ComfyUI**: For seamless integration and image processing.
+                    - **Hugging Face**: For state-of-the-art language models.
+                    - **Gradio**: For an intuitive user interface.
+                    """)
+
+                with gr.Accordion("Instructions", open=True):
+                    gr.Markdown("""
+                    **SD3 Batch Imagine Instructions**
+                    - Enter your prompts below, one per empty line.
+                    - Enter your negative prompts below, one per line. (Optional)
+                    - Set the zip interval to determine how many images will be included in each zip file.
+                    - Click "Process Prompts" to start generating images.
+                    - Click "Cancel Processing" to stop the current batch run.
+                    - Watch the progress as images are generated in real-time.
+                    - At the end of the process, zip files containing your images will be available for download.
+                    """)
+
+            with gr.Column(scale=2):
+                gr.Markdown("### Enter Prompts")
+                prompts_text = gr.Textbox(lines=20, placeholder="Enter your prompts here, one per empty line.", label="Prompts")
+                negative_prompts_text = gr.Textbox(lines=5, placeholder="Enter your negative prompts here, one per line.", label="Negative Prompts")
+                zip_interval = gr.Number(value=10, label="Zip Interval", precision=0)
+                process_btn = gr.Button("Process Prompts")
+                cancel_btn = gr.Button("Cancel Processing")
+                progress_text = gr.Markdown("Progress")
+                gallery_output = gr.Gallery(label="Generated Images")
+                zip_files_output = gr.Files(label="Zip Files")
+
+            with gr.Column(scale=1):
+                gr.Markdown("### Detailed Logs")
+                logs_output = gr.Textbox(lines=10, interactive=False, label="Logs")
+
+                def generate_user_folder():
+                    user_folder = os.path.normpath(os.path.join(OUTPUT_DIR, f'SD3{random.randint(1000, 9999)}'))
+                    os.makedirs(user_folder, exist_ok=True)
+                    logging.info(f"Generated user folder: {user_folder}")
+                    return user_folder
+
+                def on_click(prompts_text, negative_prompts_text, zip_interval):
+                    reset_cancel_processing_fn()
+                    user_folder = generate_user_folder()
+                    output_images, zip_files_list = [], []
+                    logs = ""
+                    for images, zip_files, log_msg in process_prompts(prompts_text, negative_prompts_text, user_folder, zip_interval):
+                        output_images = images
+                        zip_files_list = zip_files
+                        logs = log_msg
+                        yield images, zip_files_list, logs
+                    return output_images, zip_files_list, logs
+
+                process_btn.click(
+                    fn=on_click,
+                    inputs=[prompts_text, negative_prompts_text, zip_interval],
+                    outputs=[gallery_output, zip_files_output, logs_output]
+                )
+
+                cancel_btn.click(
+                    fn=cancel_processing_fn,
+                    inputs=[],
+                    outputs=[]
+                )
+
+    demo.launch(share=True)
+
+if __name__ == "__main__":
+    main()