Update app.py

app.py

@@ -1,15 +1,24 @@
-import gradio as gr
-from PIL import Image, ImageEnhance, ImageOps
-import numpy as np
-import cv2
+import re
+
+def parse_color(color):
+    if isinstance(color, str) and color.startswith("rgba"):
+        # Parse rgba string
+        match = re.match(r"rgba\(([\d.]+), *([\d.]+), *([\d.]+), *[\d.]+\)", color)
+        if match:
+            r, g, b = match.groups()
+            return (int(float(r)), int(float(g)), int(float(b)))
+    elif isinstance(color, str) and color.startswith("#"):
+        return color  # Hex color
+    return "white"  # Fallback
 
-# Apply color tint and enhancement
 def beautify_image(image, tint_color, contrast, brightness, saturation, apply_filter):
     img = image.convert("RGB")
-    
-    # Tint image
-    if tint_color != "None":
-        r, g, b = Image.new("RGB", img.size, tint_color).split()
+
+    # Handle rgba or hex color from color picker
+    tint = parse_color(tint_color)
+
+    if tint != "None":
+        r, g, b = Image.new("RGB", img.size, tint).split()
         img = Image.blend(img, Image.merge("RGB", (r, g, b)), alpha=0.3)
 
     # Enhance image properties
@@ -17,7 +26,7 @@ def beautify_image(image, tint_color, contrast, brightness, saturation, apply_fi
     img = ImageEnhance.Brightness(img).enhance(brightness)
     img = ImageEnhance.Color(img).enhance(saturation)
 
-    # Apply filters (sepia or colormap)
+    # Apply filters
     if apply_filter != "None":
         img_cv = np.array(img)
         if apply_filter == "Sepia":
@@ -33,21 +42,3 @@ def beautify_image(image, tint_color, contrast, brightness, saturation, apply_fi
         img = Image.fromarray(img_cv)
 
     return img
-
-# Gradio UI
-demo = gr.Interface(
-    fn=beautify_image,
-    inputs=[
-        gr.Image(type="pil", label="Upload Black & White Image"),
-        gr.ColorPicker(label="Tint Color", value="#FFD700"),  # Golden yellow
-        gr.Slider(0.5, 2.0, 1.2, label="Contrast"),
-        gr.Slider(0.5, 2.0, 1.1, label="Brightness"),
-        gr.Slider(0.0, 2.0, 1.0, label="Saturation"),
-        gr.Dropdown(["None", "Sepia", "JET", "OCEAN", "PINK", "AUTUMN"], label="Optional Filter"),
-    ],
-    outputs=gr.Image(label="Beautified Image"),
-    title="Beautify Verse Screenshots",
-    description="Upload B&W verse images to add color, contrast, and filters to make them beautiful.",
-)
-
-demo.launch()