Update

.gitignore

@@ -0,0 +1,2 @@
+images/
+.venv/

app.py

@@ -0,0 +1,55 @@
+# app.py 
+from config.config import models, prompts, api_token  # Direct import
+import gradio as gr
+from src.img_gen import generate_image
+
+# Gradio Interface
+def gradio_interface():
+    with gr.Blocks(css="""
+                   
+                    .gradio-container {
+                        background-image: url(''); 
+                        background-size: cover;
+                        background-position: center;
+
+                    }
+
+                    .output-image img {
+                     width: 2500px; /* Force image to fill container width */
+                     object-fit: cover; /* ACTIVATE FOR IMAGE-FIT CONTAINER */
+                    }
+                    """) as demo:
+        gr.Markdown("# ========== Loot Survivor - AI Image Generator ==========")
+        with gr.Row():
+            # Set default values for dropdowns
+            prompt_dropdown = gr.Dropdown(choices=[p["alias"] for p in prompts], label="Select Prompt", value=prompts[0]["alias"])
+            character_dropdown = gr.Dropdown(choices=["Beast only", "Wizard", "Warrior"], label="Select Character", value="Beast only")
+            model_dropdown = gr.Dropdown(choices=[m["alias"] for m in models], label="Select Model", value=models[0]["alias"])
+        with gr.Row():
+            # Add a text box for custom user input (max 200 characters)
+            custom_prompt_input = gr.Textbox(label="Custom Prompt (Optional)", placeholder="Enter additional details (max 200 chars)...", max_lines=1, max_length=200)
+        with gr.Row():
+            generate_button = gr.Button("Generate Image")
+        with gr.Row():
+            output_image = gr.Image(elem_classes="output-image", label="Generated Image", show_label=False, scale=1, width="100%")
+        with gr.Row():
+            status_text = gr.Textbox(label="Status", placeholder="Waiting for input...", interactive=False)
+
+        # Connect the button to the function
+        generate_button.click(
+            generate_image,
+            inputs=[prompt_dropdown, 
+                    custom_prompt_input,
+                    character_dropdown,
+                    model_dropdown
+                    ],
+            outputs=[output_image, status_text]
+        )
+    return demo
+
+# Create the demo instance
+demo = gradio_interface()
+
+# Only launch if running directly
+if __name__ == "__main__":
+        demo.queue().launch()

config/config.py

@@ -0,0 +1,12 @@
+# config.py
+import os
+from config.prompts import prompts 
+from config.models import models
+
+# Retrieve the Hugging Face token
+api_token = os.getenv("HF_TOKEN")
+
+# Debugging: Print prompt and model options
+print("##### IMPORTING CONFIG #####")
+print("Prompt Options:", [p["alias"] for p in prompts])
+print("Model Options:", [m["alias"] for m in models])

config/models.py

@@ -0,0 +1,5 @@
+# List of models with aliases
+models = [
+    {"alias": "FLUX.1-dev", "name": "black-forest-labs/FLUX.1-dev"},
+    {"alias": "Midjourney", "name": "strangerzonehf/Flux-Midjourney-Mix2-LoRA"},
+]

config/prompts.py

@@ -0,0 +1,43 @@
+
+# List of prompts with intense combat
+#
+
+prompts = [
+    {
+      "alias": "Mantis",
+      "text": "A towering, predatory mantis with razor-sharp scythe-like arms, covered in chitinous armor with a metallic green sheen. Its eyes glow faintly in the dark, and its mandibles drip with venom. Unreal Engine render style, photorealistic, realistic fantasy style."
+    },
+    {
+      "alias": "Spider",
+      "text": "A massive, hairy spider with glowing red eyes and fangs dripping with poison. Its legs are covered in sharp spines, and its body is adorned with intricate, web-like patterns. It lurks in the shadows of a damp, moss-covered environment. Unreal Engine render style, photorealistic, realistic fantasy style."
+    },
+    {
+      "alias": "Pixie",
+      "text": "A tiny, mischievous pixie with translucent wings shimmering like stained glass. Its glowing blue eyes and pointed ears give it an otherworldly appearance. It hovers in the air, surrounded by a faint magical aura. Unreal Engine render style, photorealistic, realistic fantasy style."
+    },
+    {
+      "alias": "Wolf",
+      "text": "A ferocious, oversized wolf with matted fur, glowing yellow eyes, and saliva dripping from its sharp fangs. Its muscular frame is scarred from battles, and it snarls menacingly in the torchlight. Unreal Engine render style, photorealistic, realistic fantasy style."
+    },
+    {
+      "alias": "Rat",
+      "text": "A giant, diseased rat with matted fur, glowing red eyes, and long, jagged claws. Its tail is thick and scaly, and it crouches in a filthy, garbage-strewn corner. Unreal Engine render style, photorealistic, realistic fantasy style."
+    },
+    {
+      "alias": "Ogre",
+      "text": "A hulking ogre with greenish-gray skin, covered in warts and scars. Its massive fists are clenched, and it wears crude armor made of bones and leather. It roars menacingly, surrounded by broken weapons and bones. Unreal Engine render style, photorealistic, realistic fantasy style."
+    },
+    {
+      "alias": "Troll",
+      "text": "A towering troll with rocky, moss-covered skin and glowing orange eyes. Its long arms end in clawed hands, and it wields a massive wooden club. It stands in a damp, torchlit environment, surrounded by rubble. Unreal Engine render style, photorealistic, realistic fantasy style."
+    },
+    {
+      "alias": "Bigfoot",
+      "text": "A massive, ape-like creature with thick, matted fur and glowing yellow eyes. Its muscular frame is covered in scars, and it roars menacingly in a dark, forested area. Unreal Engine render style, photorealistic, realistic fantasy style."
+    },
+    {
+      "alias": "Bear",
+      "text": "A monstrous bear with matted fur, glowing red eyes, and sharp, bloodstained claws. Its massive frame is covered in scars, and it roars fiercely, surrounded by bones and broken weapons. Unreal Engine render style, photorealistic, realistic fantasy style."
+    }
+  ]
+

config/prompts_no_tooMuchDungeon.py

@@ -0,0 +1,38 @@
+prompts = [
+    {
+      "alias": "Mantis",
+      "text": "A towering, predatory mantis with razor-sharp scythe-like arms, covered in chitinous armor with a metallic green sheen. Its eyes glow faintly in the dark dungeon, and its mandibles drip with venom. The creature stands amidst crumbling stone walls and scattered bones. Unreal Engine render style, photorealistic, realistic fantasy style"
+    },
+    {
+      "alias": "Spider",
+      "text": "A massive, hairy spider with glowing red eyes and fangs dripping with poison. Its legs are covered in sharp spines, and its body is adorned with intricate, web-like patterns. It lurks in the shadows of a damp, moss-covered dungeon corridor, surrounded by ancient stone pillars. Unreal Engine render style, photorealistic, realistic fantasy style"
+    },
+    {
+      "alias": "Pixie",
+      "text": "A tiny, mischievous pixie with translucent wings shimmering like stained glass. Its glowing blue eyes and pointed ears give it an otherworldly appearance. It hovers in the air, surrounded by a faint magical aura, within a dimly lit dungeon chamber filled with glowing runes. Unreal Engine render style, photorealistic, realistic fantasy style"
+    },
+    {
+      "alias": "Wolf",
+      "text": "A ferocious, oversized wolf with matted fur, glowing yellow eyes, and saliva dripping from its sharp fangs. Its muscular frame is scarred from battles, and it snarls menacingly in the torchlight of a narrow dungeon passage, surrounded by broken chains and debris. Unreal Engine render style, photorealistic, realistic fantasy style"
+    },
+    {
+      "alias": "Rat",
+      "text": "A giant, diseased rat with matted fur, glowing red eyes, and long, jagged claws. Its tail is thick and scaly, and it crouches in a filthy, garbage-strewn corner of a dungeon, surrounded by rotting wood and rusted metal. Unreal Engine render style, photorealistic, realistic fantasy style"
+    },
+    {
+      "alias": "Ogre",
+      "text": "A hulking ogre with greenish-gray skin, covered in warts and scars. Its massive fists are clenched, and it wears crude armor made of bones and leather. It roars menacingly in a torchlit dungeon chamber, surrounded by broken weapons, bones, and shattered stone. Unreal Engine render style, photorealistic, realistic fantasy style"
+    },
+    {
+      "alias": "Troll",
+      "text": "A towering troll with rocky, moss-covered skin and glowing orange eyes. Its long arms end in clawed hands, and it wields a massive wooden club. It stands in a damp, torchlit dungeon hall, surrounded by rubble and ancient carvings on the walls. Unreal Engine render style, photorealistic, realistic fantasy style"
+    },
+    {
+      "alias": "Bigfoot",
+      "text": "A massive, ape-like creature with thick, matted fur and glowing yellow eyes. Its muscular frame is covered in scars, and it roars menacingly in a dark, forested section of the dungeon, where roots and vines have broken through the stone walls. Unreal Engine render style, photorealistic, realistic fantasy style"
+    },
+    {
+      "alias": "Bear",
+      "text": "A monstrous bear with matted fur, glowing red eyes, and sharp, bloodstained claws. Its massive frame is covered in scars, and it roars fiercely in a torchlit dungeon chamber, surrounded by bones, broken weapons, and shattered stone. Unreal Engine render style, photorealistic, realistic fantasy style"
+    }
+]

early_notes/prompts.txt

@@ -0,0 +1,34 @@
+lets create some prompts fo img gn about some creatures. the imgs are about a game, a dungeon game, where player search for loot and try to survive, the img we will generate represent the mosters thy can find inside those dungeons. the them is medieval / fantasy. but the prompts should try to be photorealistic. i will give you a raw list of the monsters, maybe you will need the clear it up a bit, and make me the prompts on a json format w a alias and a text; please als add to the end of each prompt these: "Unreal Engine render style, photorealistic, realistic fantasy style"
+
+monsters lis:  "beasts": [
+      {
+        "beast": "Mantis"
+      },
+      {
+        "beast": "Spider"
+      },
+      {
+        "beast": "Pixie"
+      },
+      {
+        "beast": "Wolf"
+      },
+      {
+        "beast": "Rat"
+      },
+      {
+        "beast": "Ogre"
+      },
+      {
+        "beast": "Troll"
+      },
+      {
+        "beast": "Rat"
+      },
+      {
+        "beast": "Bigfoot"
+      },
+      {
+        "beast": "Bear"
+      }
+    ]

index.php

@@ -0,0 +1,188 @@
+<?php
+// Function to get supported image files
+function getImages($directory) {
+    $allowedTypes = array('jpg', 'jpeg', 'png', 'gif');
+    $files = array();
+    
+    if (is_dir($directory)) {
+        $dirContent = scandir($directory);
+        foreach ($dirContent as $file) {
+            $extension = strtolower(pathinfo($file, PATHINFO_EXTENSION));
+            if (in_array($extension, $allowedTypes)) {
+                $files[] = $file;
+            }
+        }
+    }
+    return $files;
+}
+?>
+
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Dark Theme Image Gallery</title>
+    <style>
+        body {
+            background-color: #1a1a1a;
+            color: #ffffff;
+            font-family: Arial, sans-serif;
+            margin: 0;
+            padding: 20px;
+        }
+
+        .gallery-container {
+            max-width: 1200px;
+            margin: 0 auto;
+        }
+
+        h1 {
+            text-align: center;
+            color: #ffffff;
+            margin-bottom: 30px;
+        }
+
+        .gallery {
+            display: grid;
+            grid-template-columns: repeat(auto-fill, minmax(250px, 1fr));
+            gap: 20px;
+            padding: 20px;
+        }
+
+        .gallery-item {
+            position: relative;
+            overflow: hidden;
+            border-radius: 8px;
+            background-color: #2d2d2d;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.3);
+            transition: transform 0.3s ease;
+            cursor: pointer;
+        }
+
+        .gallery-item:hover {
+            transform: scale(1.02);
+        }
+
+        .gallery-item img {
+            width: 100%;
+            height: 200px;
+            object-fit: cover;
+            display: block;
+        }
+
+        .image-caption {
+            padding: 10px;
+            background-color: rgba(0, 0, 0, 0.7);
+            color: #ffffff;
+            font-size: 0.9em;
+            text-align: center;
+        }
+
+        /* Lightbox styles */
+        .lightbox {
+            display: none;
+            position: fixed;
+            top: 0;
+            left: 0;
+            width: 100%;
+            height: 100%;
+            background-color: rgba(0, 0, 0, 0.9);
+            z-index: 1000;
+            justify-content: center;
+            align-items: center;
+            cursor: pointer;
+        }
+
+        .lightbox.active {
+            display: flex;
+        }
+
+        .lightbox img {
+            max-width: 90%;
+            max-height: 90vh;
+            object-fit: contain;
+            border: 2px solid #ffffff;
+            box-shadow: 0 0 20px rgba(0, 0, 0, 0.5);
+        }
+
+        .close-button {
+            position: fixed;
+            top: 20px;
+            right: 20px;
+            color: #ffffff;
+            font-size: 30px;
+            cursor: pointer;
+            background: none;
+            border: none;
+            padding: 10px;
+            z-index: 1001;
+        }
+
+        @media (max-width: 768px) {
+            .gallery {
+                grid-template-columns: repeat(auto-fill, minmax(200px, 1fr));
+            }
+        }
+    </style>
+</head>
+<body>
+    <div class="gallery-container">
+    <h1>========== LS-AI-img-gen - Image Gallery ==========</h1>
+        <div class="gallery">
+            <?php
+            // Specify your images directory
+            $imageDirectory = 'images/';
+            $images = getImages($imageDirectory);
+
+            foreach ($images as $image) {
+                echo '<div class="gallery-item" onclick="openLightbox(\'' . $imageDirectory . $image . '\')">';
+                echo '<img src="' . $imageDirectory . $image . '" alt="' . pathinfo($image, PATHINFO_FILENAME) . '">';
+                echo '<div class="image-caption">' . pathinfo($image, PATHINFO_FILENAME) . '</div>';
+                echo '</div>';
+            }
+
+            if (empty($images)) {
+                echo '<p style="text-align: center; grid-column: 1/-1;">No images found in the gallery.</p>';
+            }
+            ?>
+        </div>
+    </div>
+
+    <!-- Lightbox container -->
+    <div class="lightbox" id="lightbox" onclick="closeLightbox()">
+        <button class="close-button" onclick="closeLightbox()">&times;</button>
+        <img id="lightbox-img" src="" alt="Lightbox image">
+    </div>
+
+    <script>
+        function openLightbox(imageSrc) {
+            const lightbox = document.getElementById('lightbox');
+            const lightboxImg = document.getElementById('lightbox-img');
+            lightboxImg.src = imageSrc;
+            lightbox.classList.add('active');
+            // Prevent scrolling when lightbox is open
+            document.body.style.overflow = 'hidden';
+        }
+
+        function closeLightbox() {
+            const lightbox = document.getElementById('lightbox');
+            lightbox.classList.remove('active');
+            // Restore scrolling
+            document.body.style.overflow = 'auto';
+        }
+
+        // Close lightbox with Escape key
+        document.addEventListener('keydown', function(event) {
+            if (event.key === 'Escape') {
+                closeLightbox();
+            }
+        });
+
+        // Prevent lightbox from closing when clicking on the image
+        document.getElementById('lightbox-img').addEventListener('click', function(event) {
+            event.stopPropagation();
+        });
+    </script>
+</body>
+</html>

index_old.php

@@ -0,0 +1,111 @@
+<?php
+// Function to get supported image files
+function getImages($directory) {
+    $allowedTypes = array('jpg', 'jpeg', 'png', 'gif');
+    $files = array();
+    
+    if (is_dir($directory)) {
+        $dirContent = scandir($directory);
+        foreach ($dirContent as $file) {
+            $extension = strtolower(pathinfo($file, PATHINFO_EXTENSION));
+            if (in_array($extension, $allowedTypes)) {
+                $files[] = $file;
+            }
+        }
+    }
+    return $files;
+}
+?>
+
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Dark Theme Image Gallery</title>
+    <style>
+        body {
+            background-color: #1a1a1a;
+            color: #ffffff;
+            font-family: Arial, sans-serif;
+            margin: 0;
+            padding: 20px;
+        }
+
+        .gallery-container {
+            max-width: 1200px;
+            margin: 0 auto;
+        }
+
+        h1 {
+            text-align: center;
+            color: #ffffff;
+            margin-bottom: 30px;
+        }
+
+        .gallery {
+            display: grid;
+            grid-template-columns: repeat(auto-fill, minmax(250px, 1fr));
+            gap: 20px;
+            padding: 20px;
+        }
+
+        .gallery-item {
+            position: relative;
+            overflow: hidden;
+            border-radius: 8px;
+            background-color: #2d2d2d;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.3);
+            transition: transform 0.3s ease;
+        }
+
+        .gallery-item:hover {
+            transform: scale(1.02);
+        }
+
+        .gallery-item img {
+            width: 100%;
+            height: 200px;
+            object-fit: cover;
+            display: block;
+        }
+
+        .image-caption {
+            padding: 10px;
+            background-color: rgba(0, 0, 0, 0.7);
+            color: #ffffff;
+            font-size: 0.9em;
+            text-align: center;
+        }
+
+        @media (max-width: 768px) {
+            .gallery {
+                grid-template-columns: repeat(auto-fill, minmax(200px, 1fr));
+            }
+        }
+    </style>
+</head>
+<body>
+    <div class="gallery-container">
+        <h1>========== LS-AI-img-gen - Image Gallery ==========</h1>
+        <div class="gallery">
+            <?php
+            // Specify your images directory
+            $imageDirectory = 'images/';
+            $images = getImages($imageDirectory);
+
+            foreach ($images as $image) {
+                echo '<div class="gallery-item">';
+                echo '<img src="' . $imageDirectory . $image . '" alt="' . pathinfo($image, PATHINFO_FILENAME) . '">';
+                echo '<div class="image-caption">' . pathinfo($image, PATHINFO_FILENAME) . '</div>';
+                echo '</div>';
+            }
+
+            if (empty($images)) {
+                echo '<p style="text-align: center; grid-column: 1/-1;">No images found in the gallery.</p>';
+            }
+            ?>
+        </div>
+    </div>
+</body>
+</html>

requirements.txt

@@ -0,0 +1,13 @@
+torch
+diffusers
+transformers
+accelerate
+gradio
+safetensors
+pillow
+numpy<2
+invisible_watermark
+huggingface_hub[hf_transfer]
+sentencepiece
+opencv-python==4.5.5.64
+gguf

src/img_gen copy.py

@@ -0,0 +1,96 @@
+# img_gen.py
+import sys
+import os
+import random
+from huggingface_hub import InferenceClient, login
+from datetime import datetime
+from config.config import models, prompts, api_token  # Direct import
+
+def generate_image(
+        prompt_alias, 
+        custom_prompt, 
+        characer_dropdown,
+        model_alias, 
+        height=360, 
+        width=640, 
+        num_inference_steps=20, 
+        guidance_scale=2.0, 
+        seed=-1):
+    # Find the selected prompt and model
+    try:
+        prompt = next(p for p in prompts if p["alias"] == prompt_alias)["text"]
+        model_name = next(m for m in models if m["alias"] == model_alias)["name"]
+
+    except StopIteration:
+        return None, "ERROR: Invalid prompt or model selected."
+
+    # Print the original prompt and dynamic values for debugging
+    print("Original Prompt:")
+    print(prompt)
+
+    # Append the custom character (if provided)
+    if characer_dropdown == "Wizard":
+        prompt += " A wizard dressed in tunic figths against the beast"
+    elif characer_dropdown == "Warrior":
+        prompt += " A medieval warrior in armor figths against the beast"
+    else:
+        pass
+
+   # Append the custom prompt (if provided)
+    if custom_prompt and len(custom_prompt.strip()) > 0:
+        prompt += " " + custom_prompt.strip()
+
+    # Print the formatted prompt for debugging
+    print("\nFormatted Prompt:")
+    print(prompt)
+
+    # Randomize the seed if needed
+    if seed == -1:
+        seed = random.randint(0, 1000000)
+
+    # HF LOGIN 
+    print("Initializing HF TOKEN")
+    print (api_token)
+    # login(token=api_token)
+    # print("model_name:")
+    # print(model_name)
+
+
+    # Initialize the InferenceClient
+    try:
+        print("-----INITIALIZING INFERENCE-----")
+        client = InferenceClient(model_name, token=api_token)
+        print("Inference activated")
+    except Exception as e:
+        return None, f"ERROR: Failed to initialize InferenceClient. Details: {e}"
+
+     #Generate the image
+    try:
+        print("-----GENERATING IMAGE-----")
+        print("-----HOLD ON-----")   
+        image = client.text_to_image(
+            prompt,
+            guidance_scale=guidance_scale,
+            num_inference_steps=num_inference_steps,
+            width=width,
+            height=height,
+            seed=seed
+        )
+        print("-----IMAGE GENERATED SUCCESSFULLY!-----")
+    except Exception as e:
+        return None, f"ERROR: Failed to generate image. Details: {e}"
+
+    # Save the image with a timestamped filename
+    print("-----SAVING-----", image)
+    path = "images"
+    
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    output_filename = f"{path}/{timestamp}_{model_alias.replace(' ', '_').lower()}_{prompt_alias.replace(' ', '_').lower()}_{characer_dropdown.replace(' ', '_').lower()}.png"
+    try:
+        image.save(output_filename)
+    except Exception as e:
+        return None, f"ERROR: Failed to save image. Details: {e}"
+    print("-----DONE!-----")
+    print("-----CALL THE BANNERS!-----")
+
+    return output_filename, "Image generated successfully!"

src/img_gen.py

@@ -0,0 +1,96 @@
+# img_gen.py
+import sys
+import os
+import random
+from huggingface_hub import InferenceClient, login
+from datetime import datetime
+from config.config import models, prompts, api_token  # Direct import
+
+def generate_image(
+        prompt_alias, 
+        custom_prompt, 
+        characer_dropdown,
+        model_alias, 
+        height=360, 
+        width=640, 
+        num_inference_steps=20, 
+        guidance_scale=2.0, 
+        seed=-1):
+    # Find the selected prompt and model
+    try:
+        prompt = next(p for p in prompts if p["alias"] == prompt_alias)["text"]
+        model_name = next(m for m in models if m["alias"] == model_alias)["name"]
+
+    except StopIteration:
+        return None, "ERROR: Invalid prompt or model selected."
+
+    # Print the original prompt and dynamic values for debugging
+    print("Original Prompt:")
+    print(prompt)
+
+    # Append the custom character (if provided)
+    if characer_dropdown == "Wizard":
+        prompt += f" A fantasy wizard figths against the {prompt_alias}"
+    elif characer_dropdown == "Warrior":
+        prompt += f" A fantasy warrior figths against the {prompt_alias}"
+    else:
+        pass
+
+   # Append the custom prompt (if provided)
+    if custom_prompt and len(custom_prompt.strip()) > 0:
+        prompt += " " + custom_prompt.strip()
+
+    # Print the formatted prompt for debugging
+    print("\nFormatted Prompt:")
+    print(prompt)
+
+    # Randomize the seed if needed
+    if seed == -1:
+        seed = random.randint(0, 1000000)
+
+    # HF LOGIN 
+    print("Initializing HF TOKEN")
+    print (api_token)
+    # login(token=api_token)
+    # print("model_name:")
+    # print(model_name)
+
+
+    # Initialize the InferenceClient
+    try:
+        print("-----INITIALIZING INFERENCE-----")
+        client = InferenceClient(model_name, token=api_token)
+        print("Inference activated")
+    except Exception as e:
+        return None, f"ERROR: Failed to initialize InferenceClient. Details: {e}"
+
+     #Generate the image
+    try:
+        print("-----GENERATING IMAGE-----")
+        print("-----HOLD ON-----")   
+        image = client.text_to_image(
+            prompt,
+            guidance_scale=guidance_scale,
+            num_inference_steps=num_inference_steps,
+            width=width,
+            height=height,
+            seed=seed
+        )
+        print("-----IMAGE GENERATED SUCCESSFULLY!-----")
+    except Exception as e:
+        return None, f"ERROR: Failed to generate image. Details: {e}"
+
+    # Save the image with a timestamped filename
+    print("-----SAVING-----", image)
+    path = "images"
+    
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    output_filename = f"{path}/{timestamp}_{model_alias.replace(' ', '_').lower()}_{prompt_alias.replace(' ', '_').lower()}_{characer_dropdown.replace(' ', '_').lower()}.png"
+    try:
+        image.save(output_filename)
+    except Exception as e:
+        return None, f"ERROR: Failed to save image. Details: {e}"
+    print("-----DONE!-----")
+    print("-----CALL THE BANNERS!-----")
+
+    return output_filename, "Image generated successfully!"