improve canvas size tip

README.md

@@ -7,7 +7,16 @@ TLDR: From generated Image -> Copaint PDF
 ## Usage and Install
 ```
 # Install dependencies
-pip install torch torchvision reportlab PyPDF2 Pillow argparse  gradio_pdf
+poetry install --with ui
+
+# Run Gradio App on local server
+poetry run copaint-app
+```
+
+## Alternative Usage and Install (CLI)
+```
+# Install dependencies
+pip install torch torchvision reportlab PyPDF2 Pillow argparse gradio_pdf
 
 # Run
 python copaint.py --input_image data/bear.jpg --copaint_logo data/logo_copaint.jpg --outputfolder output

copaint/gradio_ui.py

@@ -36,7 +36,7 @@ def add_grid_to_image(image, h_cells, w_cells):
     return image
 
 
-def canvas_ratio(image):
+def canvas_ratio(image, h_cells, w_cells):
     w,h = image.size
     aspect_ratio = w/h
     if aspect_ratio > 1:
@@ -44,7 +44,7 @@ def canvas_ratio(image):
     
     # find nearest aspect ratio in the list of predefined ones 
     predefined_aspect_ratios = [2/3, 1/2, 1/1, 5/6, 4/5, 5/7]
-    predefined_aspect_ratios_str = ["2/3", "1/2", "1/1", "5/6", "4/5", "5/7"]
+    predefined_aspect_ratios_str = ["2:3", "1:2", "1:1", "5:6", "4:5", "5:7"]
     min_diff = float('inf')
     closest_ratio_idx = None
     for idx, ratio in enumerate(predefined_aspect_ratios):
@@ -57,13 +57,31 @@ def canvas_ratio(image):
     if min_diff > 0.1:
         return None
     else:
-        return f"Best canvas ratio: {closest_ratio_str}"
+
+        example_str = ""
+
+        if closest_ratio_str == "1:1":
+            if h_cells == 2 and h_cells == 2:
+                example_str = " (for example, a 6” by 6” canvas)"
+            if h_cells == 3 and h_cells == 3:
+                example_str = " (for example, an 8” by 8” canvas)"
+        elif closest_ratio_str == "5:6":
+                example_str = " (for example, a 10” by 12” canvas)"
+        elif closest_ratio_str == "3:4":
+            if (h_cells == 3 and w_cells == 4) or (h_cells == 4 and w_cells == 3):
+                example_str = " (for example, a 6” by 8” canvas)"
+            if (h_cells == 4 and w_cells == 6) or (h_cells == 6 and w_cells == 4):
+                example_str = " (for example, an 18” by 24”, or a 12” by 16” canvas)"
+        elif closest_ratio_str == "2:3" and ((h_cells == 6 and w_cells == 9) or (h_cells == 9 and w_cells == 6)):
+            example_str = " (for example, a 24” by 36” canvas)"
+
+        return f"Preparing your canvas: *choose a canvas with a {closest_ratio_str} ratio to respect your design’s size{example_str}.*"
 
 
 def add_grid_and_display_ratio(image, h_cells, w_cells):
     if image is None:
         return None, gr.update(visible=False)    
-    return add_grid_to_image(image, h_cells, w_cells), gr.update(visible=True, value=canvas_ratio(image))
+    return add_grid_to_image(image, h_cells, w_cells), gr.update(visible=True, value=canvas_ratio(image, h_cells, w_cells))
 
 
 def process_copaint(
@@ -214,17 +232,12 @@ def build_gradio_ui():
                 output_pdf = PDF(label="PDF Preview")
 
         # Update output_image: trigger update when any input changes
-        input_image.change(
+        for component in [input_image, h_cells, w_cells]:
+            component.change(
                 fn=add_grid_and_display_ratio,
                 inputs=[input_image, h_cells, w_cells],
                 outputs=[output_image, canvas_msg]
             )
-        for component in [h_cells, w_cells]:
-            component.change(
-                fn=add_grid_to_image,
-                inputs=[input_image, h_cells, w_cells],
-                outputs=output_image
-            )
 
         # Submit function: generate pdf
         def on_submit(input_image, h_cells, w_cells, use_a4, high_res, cell_size, copaint_name, copaint_logo):