Pixel Art Poop Mosaic Generator

CODE

from google.colab import files
from PIL import Image
import io
from IPython.display import display

def generate_parameters_only():
   print("🎨 WELCOME TO THE 3D MOSAIC GENERATOR 🎨")
   print("This tool will prepare your photo and give you the parameters for MakerWorld.\n")
   
   print("📁 1. Upload your image (JPG or PNG format):")
   uploaded = files.upload()
   
   if not uploaded:
       print("No file uploaded. Please try again.")
       return

   file_name = list(uploaded.keys())[0]
   try:
       img = Image.open(io.BytesIO(uploaded[file_name])).convert("RGB")
   except Exception:
       print("Error reading the image.")
       return

   # Calcolo del rapporto d'aspetto
   w, h = img.size
   ratio = w / h
   tolerance = 0.05
   detected_ratio = None

   if abs(ratio - 1.0) <= tolerance:
       detected_ratio = "1:1"
   elif abs(ratio - 4/3) <= tolerance:
       detected_ratio = "4:3"
   elif abs(ratio - 3/4) <= tolerance:
       detected_ratio = "3:4"
   elif abs(ratio - 16/9) <= tolerance:
       detected_ratio = "16:9"
   elif abs(ratio - 9/16) <= tolerance:
       detected_ratio = "9:16"

   bed_side_mm = None
   use_premade = False

   # Rilevamento automatico e proposta tramite input numerico
   if detected_ratio:
       print(f"\n✨ I detected that your image has a standard {detected_ratio} aspect ratio!")
       print("If you don't want to sweat over custom settings, I highly recommend using the pre-made print profile already available on the MakerWorld page.")
       print("It has a long edge of 180mm and matches your photo perfectly.")
       print("\nDo you want to use this ready-to-print profile?")
       print("[1] Yes, make it easy!")
       print("[2] No, I want to use custom dimensions")
       
       ans = ""
       while ans not in ['1', '2']:
           ans = input("Type 1 or 2: ").strip()
       
       if ans == '1':
           use_premade = True
           bed_side_mm = 180.0
           print("\nPerfect! Calculating your guide for the 180mm profile...")

   # Se si procede manualmente, scelta numerica anche per le unità di misura
   if not use_premade:
       print("\n📏 2. Printer Parameters:")
       print("Do you prefer to enter dimensions in millimeters or inches?")
       print("[1] Millimeters (mm)")
       print("[2] Inches (in)")
       
       unit_choice = ""
       while unit_choice not in ['1', '2']:
           unit_choice = input("Type 1 or 2: ").strip()
       
       unit = 'mm' if unit_choice == '1' else 'in'
       unit_label = "millimeters" if unit == 'mm' else "inches"
       
       try:
           print(f"\nEnter the desired maximum length of the long edge of the mosaic.")
           print(f"(Keep in mind your printer's max bed size in {unit_label})")
           max_size_input = float(input("> "))
       except ValueError:
           print("Error: you must enter a valid number.")
           return

       if unit == 'in':
           bed_side_mm = max_size_input * 25.4
       else:
           bed_side_mm = max_size_input

   # Fixed parameters for the MakerWorld parametric model (in mm)
   tile_length = 9.5
   wall_thickness = 0.6
   
   prop = w / h
   if w >= h:
       columns = int((bed_side_mm - wall_thickness) // tile_length)
       rows = int(columns / prop)
   else:
       rows = int((bed_side_mm - wall_thickness) // tile_length)
       columns = int(rows * prop)

   small_img = img.resize((columns, rows), resample=Image.Resampling.BILINEAR)
   guide_img = small_img.resize((columns * 20, rows * 20), resample=Image.Resampling.NEAREST)
   
   total_poops = columns * rows

   print("\n" + "="*100)
   print("🎯 YOUR MAKERWORLD PARAMETERS 🎯")
   print("="*100)
   
   if use_premade:
       print("You selected the pre-made profile. You don't need to change the sliders on MakerWorld!")
       print("Just print the print profile that matches your format and follow the guide below.")
   else:
       print("Go to the Parametric Model page on MakerWorld and enter these exact values in the sliders:\n")
       print(f"👉 Number of horizontal tiles (Columns): {columns}")
       print(f"👉 Number of vertical tiles (Rows): {rows}")
   
   print("="*100 + "\n")
   
   print("✅ Here is your Color Guide!")
   display(guide_img)
   
   print("\n" + "♻️ "*40)
   print("Have fun matching each color in the photo with the color of one of the poops you have!")
   print(f"\nWith this project you will be able to recover at least {total_poops} poops.")
   print("And even more by trying to completely cover the gaps left between the irregular poops with smaller fragments!")
   print("♻️ "*40)

generate_parameters_only()