Testing Torch Error

app.py

@@ -250,51 +250,51 @@ gr.set_static_paths(paths=["images/","images/images","images/prerendered","LUT/"
 # Gradio Blocks Interface
 with gr.Blocks(css_paths="style_20250128.css", title="HexaGrid Creator", theme='Surn/beeuty') as beeuty:
     with gr.Row():
-        gr.Markdown ("""
+        gr.Markdown("""
         # HexaGrid Creator
-        ## Transform Your Images into Mesmerizing Hexagon Grid Masterpieces! ⬢
-        <details>
-        <summary>
-        Welcome to HexaGrid Creator, the ultimate tool for transforming your images into stunning hexagon grid artworks. Whether you're a tabletop game enthusiast, a digital artist, or someone who loves unique patterns, HexaGrid Creator has something for you.
-
-        ## Drop an image into the Input Image and get started!
-
-        </summary>
-
-        ## What is HexaGrid Creator?
-        HexaGrid Creator is a web-based application that allows you to apply a hexagon grid overlay to any image. You can customize the size, color, and opacity of the hexagons, as well as the background and border colors. The result is a visually striking image that looks like it was made from hexagonal tiles!
-
-        ### What Can You Do?
-        - **Generate Hexagon Grids:** Create beautiful hexagon grid overlays on any image with fully customizable parameters.
-        - **AI-Powered Image Generation:** Use advanced AI models to generate images based on your prompts and apply hexagon grids to them.
-        - **Color Exclusion:** Select and exclude specific colors from your hexagon grid for a cleaner and more refined look.
-        - **Interactive Customization:** Adjust hexagon size, border size, rotation, background color, and more in real-time.
-        - **Depth and 3D Model Generation:** Generate depth maps and 3D models from your images for enhanced visualization.
-        - **Image Filter [Look-Up Table (LUT)] Application:** Apply filters (LUTs) to your images for color grading and enhancement.
-        - **Pre-rendered Maps:** Access a library of pre-rendered hexagon maps for quick and easy customization.
-        - **Add Margins:** Add customizable margins around your images for a polished finish.
-
-        ### Why You'll Love It
-        - **Fun and Easy to Use:** With an intuitive interface and real-time previews, creating hexagon grids has never been this fun!
-        - **Endless Creativity:** Unleash your creativity with endless customization options and see your images transform in unique ways.
-        - **Hexagon-Inspired Theme:** Enjoy a delightful yellow and purple theme inspired by hexagons! ⬢
-        - **Advanced AI Models:** Leverage advanced AI models and LoRA weights for high-quality image generation and customization.
-
-        ### Get Started
-        1. **Upload or Generate an Image:** Start by uploading your own image or generate one using our AI-powered tool.
-        2. **Customize Your Grid:** Play around with the settings to create the perfect hexagon grid overlay.
-        3. **Download and Share:** Once you're happy with your creation, download it and share it with the world!
-
-        ### Advanced Features
-        - **Generative AI Integration:** Utilize models like `black-forest-labs/FLUX.1-dev` and various LoRA weights for generating unique images.
-        - **Pre-rendered Maps:** Access a library of pre-rendered hexagon maps for quick and easy customization.
-        - **Image Filter [Look-Up Table (LUT)] Application:** Apply filters (LUTs) to your images for color grading and enhancement.
-        - **Depth and 3D Model Generation:** Create depth maps and 3D models from your images for enhanced visualization.
-        - **Add Margins:** Customize margins around your images for a polished finish.
-
-        Join the hive and start creating with HexaGrid Creator today!
-        </details>
-        """, elem_classes="intro")
+        ## Transform Your Images into Mesmerizing Hexagon Grid Masterpieces! ⬢""", elem_classes="intro")
+    with gr.Row():
+        with gr.Accordion("Welcome to HexaGrid Creator, the ultimate tool for transforming your images into stunning hexagon grid artworks. Whether you're a tabletop game enthusiast, a digital artist, or someone who loves unique patterns, HexaGrid Creator has something for you.", open=False, elem_classes="intro"):
+            gr.Markdown ("""
+
+            ## Drop an image into the Input Image and get started!
+
+        
+
+            ## What is HexaGrid Creator?
+            HexaGrid Creator is a web-based application that allows you to apply a hexagon grid overlay to any image. You can customize the size, color, and opacity of the hexagons, as well as the background and border colors. The result is a visually striking image that looks like it was made from hexagonal tiles!
+
+            ### What Can You Do?
+            - **Generate Hexagon Grids:** Create beautiful hexagon grid overlays on any image with fully customizable parameters.
+            - **AI-Powered Image Generation:** Use advanced AI models to generate images based on your prompts and apply hexagon grids to them.
+            - **Color Exclusion:** Select and exclude specific colors from your hexagon grid for a cleaner and more refined look.
+            - **Interactive Customization:** Adjust hexagon size, border size, rotation, background color, and more in real-time.
+            - **Depth and 3D Model Generation:** Generate depth maps and 3D models from your images for enhanced visualization.
+            - **Image Filter [Look-Up Table (LUT)] Application:** Apply filters (LUTs) to your images for color grading and enhancement.
+            - **Pre-rendered Maps:** Access a library of pre-rendered hexagon maps for quick and easy customization.
+            - **Add Margins:** Add customizable margins around your images for a polished finish.
+
+            ### Why You'll Love It
+            - **Fun and Easy to Use:** With an intuitive interface and real-time previews, creating hexagon grids has never been this fun!
+            - **Endless Creativity:** Unleash your creativity with endless customization options and see your images transform in unique ways.
+            - **Hexagon-Inspired Theme:** Enjoy a delightful yellow and purple theme inspired by hexagons! ⬢
+            - **Advanced AI Models:** Leverage advanced AI models and LoRA weights for high-quality image generation and customization.
+
+            ### Get Started
+            1. **Upload or Generate an Image:** Start by uploading your own image or generate one using our AI-powered tool.
+            2. **Customize Your Grid:** Play around with the settings to create the perfect hexagon grid overlay.
+            3. **Download and Share:** Once you're happy with your creation, download it and share it with the world!
+
+            ### Advanced Features
+            - **Generative AI Integration:** Utilize models like `black-forest-labs/FLUX.1-dev` and various LoRA weights for generating unique images.
+            - **Pre-rendered Maps:** Access a library of pre-rendered hexagon maps for quick and easy customization.
+            - **Image Filter [Look-Up Table (LUT)] Application:** Apply filters (LUTs) to your images for color grading and enhancement.
+            - **Depth and 3D Model Generation:** Create depth maps and 3D models from your images for enhanced visualization.
+            - **Add Margins:** Customize margins around your images for a polished finish.
+
+            Join the hive and start creating with HexaGrid Creator today!
+        
+            """, elem_classes="intro")
     with gr.Row():
         from utils.image_utils import convert_to_rgba_png
 

utils/ai_generator_diffusers_flux.py

@@ -184,8 +184,9 @@ def generate_image_lowmem(
             print("\nEnabled flash_attention_2.\n")
         if pipeline_name == "FluxPipeline":
             pipe.enable_vae_tiling()
+        condition_type = "subject"
         # Load LoRA weights
-        # note: does not yet handle multiple LoRA weights with different names, needs .set_adapters(["depth", "hyper-sd"], adapter_weights=[0.85, 0.125])
+        # note: does not yet handle multiple LoRA weights with different names, needs .set_adapters(["depth", "hyper-sd"], adapter_weights=[0.85, 0.125])        
         if lora_weights:
             for lora_weight in lora_weights:
                 lora_configs = constants.LORA_DETAILS.get(lora_weight, [])
@@ -248,6 +249,28 @@ def generate_image_lowmem(
                                     method(**params)
                                 else:
                                     print(f"Method {method_name} not found in pipe.")
+                        if 'condition_type' in config:
+                            condition_type = config['condition_type']
+                            if condition_type == "coloring":
+                                #pipe.enable_coloring()
+                                print("\nEnabled coloring.\n")
+                            elif condition_type == "deblurring":
+                                #pipe.enable_deblurring()
+                                print("\nEnabled deblurring.\n")
+                            elif condition_type == "fill":
+                                #pipe.enable_fill()
+                                print("\nEnabled fill.\n")
+                            elif condition_type == "depth":
+                                #pipe.enable_depth()
+                                print("\nEnabled depth.\n")
+                            elif condition_type == "canny":
+                                #pipe.enable_canny()
+                                print("\nEnabled canny.\n")
+                            elif condition_type == "subject":
+                                #pipe.enable_subject()
+                                print("\nEnabled subject.\n")
+                            else:
+                                print(f"Condition type {condition_type} not implemented.")
                 else:
                     pipe.load_lora_weights(lora_weight, use_auth_token=constants.HF_API_TOKEN)
         # Set the random seed for reproducibility
@@ -255,7 +278,7 @@ def generate_image_lowmem(
         conditions = []
         if conditioned_image is not None:
             conditioned_image = crop_and_resize_image(conditioned_image, image_width, image_height)
-            condition = Condition("subject", conditioned_image)
+            condition = Condition(condition_type, conditioned_image)
             conditions.append(condition)
             print(f"\nAdded conditioned image.\n {conditioned_image.size}")
             # Prepare the parameters for image generation

utils/constants.py

@@ -166,7 +166,7 @@ LORA_TO_MODEL = {
     "XLabs-AI/flux-lora-collection/scenery_lora.safetensors":"black-forest-labs/FLUX.1-dev",
     "XLabs-AI/flux-lora-collection/disney_lora.safetensors":"black-forest-labs/FLUX.1-dev"
 }
-
+condition_type = ["depth", "canny", "subject", "coloring", "deblurring", "fill", "redux"]
 # Detailed LoRA weight configurations
 LORA_DETAILS = {
     "AlekseyCalvin/HSTcolorFlexAlpha" : [