Upload 7 files

README.md

@@ -1,13 +1,46 @@
----
-title: Pixel Art
-emoji: 🐠
-colorFrom: indigo
-colorTo: pink
-sdk: gradio
-sdk_version: 5.49.1
-app_file: app.py
-pinned: true
-license: mit
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Pixel Art Generator - Aspect Ratio Update
+
+## Changes Made
+
+I've added aspect ratio selection to your pixel art generator with the following options:
+
+### Supported Aspect Ratios
+All aspect ratios are capped at approximately 1 megapixel resolution:
+
+- **16:9** - 1344x768 (landscape, widescreen)
+- **4:3** - 1152x896 (landscape, classic)
+- **1:1** - 1024x1024 (square, default)
+- **3:4** - 896x1152 (portrait, classic)
+- **9:16** - 768x1344 (portrait, mobile)
+
+All dimensions are divisible by 64 for optimal performance with SDXL.
+
+## Modified Files
+
+### 1. `config.py`
+- Added `ASPECT_RATIOS` dictionary with all aspect ratio options and their dimensions
+- Added `DEFAULT_ASPECT_RATIO` set to "1:1"
+
+### 2. `app.py`
+- Added aspect ratio dropdown in the Settings accordion
+- Updated the `process_text` function to accept `aspect_ratio` parameter
+- Updated the inputs list to include the aspect ratio selection
+
+### 3. `generator.py`
+- Added `aspect_ratio` parameter to the `predict` method (defaults to "1:1")
+- Extracts width and height from `Config.ASPECT_RATIOS` based on selection
+- Passes width and height to the pipeline for generation
+
+### Other Files
+- `model.py`, `requirements.txt`, and `utils.py` remain unchanged
+
+## Usage
+
+Users can now select their preferred aspect ratio from the dropdown menu in the Settings section before generating pixel art. The dropdown appears above the seed input and shows all 5 available aspect ratios with a helpful description.
+
+## Technical Details
+
+- All aspect ratios maintain approximately 1MP total pixels
+- Dimensions are optimized for SDXL (divisible by 64)
+- The aspect ratio is displayed in the console logs during generation
+- Default fallback is 1:1 (1024x1024) if an invalid ratio is provided

app.py

@@ -15,6 +15,7 @@ gen = Generator(handler)
 def process_text(
     prompt,
     negative_prompt,
+    aspect_ratio,
     cfg_scale, 
     steps, 
     seed
@@ -24,6 +25,7 @@ def process_text(
         result = gen.predict(
             user_prompt=prompt,
             negative_prompt=negative_prompt,
+            aspect_ratio=aspect_ratio,
             guidance_scale=cfg_scale,
             num_inference_steps=steps,
             seed=seed
@@ -59,6 +61,13 @@ with gr.Blocks(title="Pixel Art Generator", theme=gr.themes.Soft()) as demo:
             )
             
             with gr.Accordion("Settings", open=True):
+                aspect_ratio = gr.Dropdown(
+                    label="Aspect Ratio",
+                    choices=list(Config.ASPECT_RATIOS.keys()),
+                    value=Config.DEFAULT_ASPECT_RATIO,
+                    info="Image dimensions (all ~1MP resolution)"
+                )
+                
                 seed = gr.Number(
                     label="Seed", 
                     value=-1, 
@@ -92,6 +101,7 @@ with gr.Blocks(title="Pixel Art Generator", theme=gr.themes.Soft()) as demo:
     all_inputs = [
         prompt, 
         negative_prompt,
+        aspect_ratio,
         cfg_scale, 
         steps, 
         seed
@@ -110,4 +120,4 @@ if __name__ == "__main__":
         server_name="0.0.0.0",
         server_port=7860,
         show_api=True
-    )
+    )

config.py

@@ -19,4 +19,14 @@ class Config:
 
     # Gradio Parameters
     CGF_SCALE = 1.2           
-    STEPS_NUMBER = 8          # Reduced slightly for optimal LCM T2I performance
+    STEPS_NUMBER = 8          # Reduced slightly for optimal LCM T2I performance
+    
+    # Aspect Ratios (width, height) - all fitting within ~1MP resolution
+    ASPECT_RATIOS = {
+        "16:9": (1344, 768),
+        "4:3": (1152, 896),
+        "1:1": (1024, 1024),
+        "3:4": (896, 1152),
+        "9:16": (768, 1344)
+    }
+    DEFAULT_ASPECT_RATIO = "1:1"

generator.py

@@ -9,6 +9,7 @@ class Generator:
         self, 
         user_prompt,
         negative_prompt="",
+        aspect_ratio="1:1",
         guidance_scale=1.6,
         num_inference_steps=8,
         seed=-1
@@ -22,21 +23,27 @@ class Generator:
             
         print(f"Prompt: {final_prompt}")
         
-        # 2. Handle Seed
+        # 2. Get dimensions from aspect ratio
+        width, height = Config.ASPECT_RATIOS.get(aspect_ratio, Config.ASPECT_RATIOS[Config.DEFAULT_ASPECT_RATIO])
+        print(f"Aspect Ratio: {aspect_ratio} ({width}x{height})")
+        
+        # 3. Handle Seed
         if seed == -1 or seed is None:
             seed = torch.Generator().seed()
         generator = torch.Generator(device=Config.DEVICE).manual_seed(int(seed))
         print(f"Using seed: {seed}")
 
-        # 3. Run Text-to-Image Inference
+        # 4. Run Text-to-Image Inference
         print("Running pipeline...")
         result = self.mh.pipeline(
             prompt=final_prompt,
             negative_prompt=negative_prompt,
+            width=width,
+            height=height,
             generator=generator,
             num_inference_steps=num_inference_steps, 
             guidance_scale=guidance_scale,
             clip_skip=2, # Optional, often helps with anime/pixel styles
         ).images[0]
         
-        return result
+        return result

model.py

@@ -64,4 +64,4 @@ class ModelHandler:
         self.pipeline.fuse_lora(lora_scale=Config.LORA_STRENGTH)
         print("  [OK] LoRA fused.")
         
-        print("--- All models loaded successfully ---")
+        print("--- All models loaded successfully ---")

requirements.txt

@@ -11,4 +11,4 @@ gradio>=4.0.0
 controlnet_aux
 huggingface_hub
 mediapipe
-timm
+timm

utils.py

@@ -23,4 +23,4 @@ def resize_image_to_1mp(image):
     if new_w == 0 or new_h == 0:
         new_w, new_h = 1024, 1024  # Fallback
         
-    return image.resize((new_w, new_h), Image.LANCZOS)
+    return image.resize((new_w, new_h), Image.LANCZOS)