update app.py - color conversion

app.py

@@ -144,12 +144,45 @@ def add_text_to_image(img, pattern, line1, line2, line3, font_size, color, add_o
             
         print(f"Text positions calculated: {positions}")
 
-        # FIXED: Better color conversion
+        # FIXED: Parse RGBA string from Gradio ColorPicker
         original_color = color
         final_color = (255, 255, 255)  # Default white
         
         if isinstance(color, str):
-            if color.startswith('#'):
+            if color.startswith('rgba(') and color.endswith(')'):
+                # Parse RGBA string: 'rgba(253.2203125, 3.331846217105286, 3.331846217105286, 1)'
+                print(f"Parsing RGBA string: {color}")
+                try:
+                    # Extract numbers from the string
+                    rgba_part = color[5:-1]  # Remove 'rgba(' and ')'
+                    values = [float(x.strip()) for x in rgba_part.split(',')]
+                    if len(values) >= 3:
+                        # Convert to integers and take only RGB (ignore alpha)
+                        final_color = (int(round(values[0])), int(round(values[1])), int(round(values[2])))
+                        print(f"Successfully parsed RGBA {color} to RGB: {final_color}")
+                    else:
+                        print(f"Invalid RGBA format {color}, using white")
+                        final_color = (255, 255, 255)
+                except (ValueError, IndexError) as e:
+                    print(f"Error parsing RGBA {color}: {e}, using white")
+                    final_color = (255, 255, 255)
+            elif color.startswith('rgb(') and color.endswith(')'):
+                # Parse RGB string: 'rgb(253, 3, 3)'
+                print(f"Parsing RGB string: {color}")
+                try:
+                    rgb_part = color[4:-1]  # Remove 'rgb(' and ')'
+                    values = [float(x.strip()) for x in rgb_part.split(',')]
+                    if len(values) >= 3:
+                        final_color = (int(round(values[0])), int(round(values[1])), int(round(values[2])))
+                        print(f"Successfully parsed RGB {color} to RGB: {final_color}")
+                    else:
+                        print(f"Invalid RGB format {color}, using white")
+                        final_color = (255, 255, 255)
+                except (ValueError, IndexError) as e:
+                    print(f"Error parsing RGB {color}: {e}, using white")
+                    final_color = (255, 255, 255)
+            elif color.startswith('#'):
+                # Parse hex color
                 print(f"Converting hex color {color}")
                 hex_color = color.lstrip('#')
                 if len(hex_color) == 6:
@@ -160,7 +193,6 @@ def add_text_to_image(img, pattern, line1, line2, line3, font_size, color, add_o
                         print(f"Invalid hex color {color}: {e}, using white")
                         final_color = (255, 255, 255)
                 elif len(hex_color) == 3:
-                    # Short hex format #RGB -> #RRGGBB
                     try:
                         final_color = tuple(int(hex_color[i]*2, 16) for i in range(3))
                         print(f"Successfully converted short hex {color} to RGB: {final_color}")
@@ -171,11 +203,11 @@ def add_text_to_image(img, pattern, line1, line2, line3, font_size, color, add_o
                     print(f"Invalid hex color length {color}, using white")
                     final_color = (255, 255, 255)
             else:
-                # Try to parse as named color or other format
-                print(f"Non-hex color string: {color}, attempting to use directly")
+                # Try to use as named color
+                print(f"Attempting to use named color: {color}")
                 final_color = color  # Let PIL try to handle it
         elif isinstance(color, (tuple, list)) and len(color) >= 3:
-            final_color = tuple(color[:3])
+            final_color = tuple(int(round(x)) for x in color[:3])
             print(f"Using tuple/list color: {final_color}")
         else:
             print(f"Unrecognized color format: {color}, using white")
@@ -215,10 +247,10 @@ def add_text_to_image(img, pattern, line1, line2, line3, font_size, color, add_o
                 else:
                     print("Skipping outline (disabled)")
                 
-                # Draw main text
+                # Draw main text with the properly parsed color
                 try:
                     draw.text((x, y), txt, fill=final_color, font=font, anchor="mm")
-                    print(f"Main text drawn with color {final_color} at ({x}, {y})")
+                    print(f"SUCCESS: Main text drawn with color {final_color} at ({x}, {y})")
                 except TypeError:
                     # Fallback for older PIL versions
                     try:
@@ -228,27 +260,28 @@ def add_text_to_image(img, pattern, line1, line2, line3, font_size, color, add_o
                         fallback_x = x - text_w//2
                         fallback_y = y - text_h//2
                         draw.text((fallback_x, fallback_y), txt, fill=final_color, font=font)
-                        print(f"Fallback text drawn with color {final_color} at ({fallback_x}, {fallback_y})")
-                    except Exception:
+                        print(f"SUCCESS: Fallback text drawn with color {final_color} at ({fallback_x}, {fallback_y})")
+                    except Exception as e:
+                        print(f"Fallback failed: {e}, trying basic draw")
                         draw.text((x, y), txt, fill=final_color, font=font)
-                        print(f"Basic text drawn with color {final_color} at ({x}, {y})")
+                        print(f"SUCCESS: Basic text drawn with color {final_color} at ({x}, {y})")
                 except Exception as e:
-                    print(f"ERROR drawing main text: {e}")
-                    # Try with string color as last resort
+                    print(f"ERROR drawing main text with final_color: {e}")
+                    # Last resort - try with white
                     try:
-                        draw.text((x, y), txt, fill=original_color, font=font, anchor="mm")
-                        print(f"Drew text with original color string: {original_color}")
+                        draw.text((x, y), txt, fill=(255, 255, 255), font=font, anchor="mm")
+                        print(f"FALLBACK: Drew text in white")
                     except Exception as e2:
-                        print(f"Even original color failed: {e2}")
+                        print(f"Even white text failed: {e2}")
                 
                 text_drawn = True
-                print(f"Successfully drew line {i+1}")
+                print(f"Completed drawing line {i+1}")
 
         if not text_drawn:
             print("No text was drawn")
             return img, "No text to add (all lines were empty)"
         
-        print("=== Text drawing completed ===")
+        print("=== Text drawing completed successfully ===")
         return image, f"Text added! Color: {original_color} -> {final_color}, Outline: {add_outline}"
         
     except Exception as e: