Bug fix

.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

app.py

@@ -6,14 +6,11 @@ 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 */
@@ -22,7 +19,7 @@ def gradio_interface():
         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"])
+            prompt_dropdown = gr.Dropdown(choices=[p["alias"] for p in prompts], label="Select Beast", 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():

config/models.py

@@ -1,5 +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"},
+    {"alias": "FLUX.1-dev", "name": "black-forest-labs/FLUX.1-dev"},
 ]

config/prompts.py

@@ -13,7 +13,7 @@ prompts = [
     },
     {
       "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."
+      "text": "A tiny, mischievous evil 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",

index.php

@@ -1,8 +1,8 @@
 <?php
-    include 'functions.php';
+    include 'src/gallery_functions.php';
     $imageDirectory = 'images/';
     $imageOrder = 'desc';
     $images = getImages($imageDirectory, $imageOrder); // 'asc' or 'desc'
-    include 'gallery_content.php';
+    include 'src/gallery_content.php';
 
 ?>

src/img_gen.py

@@ -30,9 +30,9 @@ def generate_image(
 
     # Append the custom character (if provided)
     if characer_dropdown == "Wizard":
-        prompt += f" A wizard combat using magic against the {prompt_alias}"
+        prompt += f" A wizard combats using powerful magic against the {prompt_alias}"
     elif characer_dropdown == "Warrior":
-        prompt += f" A human warrior with his weapon figths against the {prompt_alias}"
+        prompt += f" A warrior combats using his weapons against the {prompt_alias}"
     else:
         pass