Update app.py

app.py

@@ -2,81 +2,82 @@ import gradio as gr
 from PIL import Image, ImageEnhance
 import numpy as np
 
-# Function to convert image to grayscale internally
+# Function to adjust contrast of the input image
+def adjust_contrast(img, contrast_level):
+    if img is None:
+        return None
+    enhancer = ImageEnhance.Contrast(img)
+    return enhancer.enhance(contrast_level)
+
+# Function to convert image to grayscale
 def convert_to_grayscale(img):
     if img is None:
         return None
-    return img.convert('L').convert('RGB')  # Convert to grayscale and back to RGB
+    return img.convert('L').convert('RGB')  # Grayscale and back to RGB for compatibility
 
-# Function to apply grayscale and custom texture with enhanced effect on white regions
-def apply_texture(img, texture, intensity=0.7):
+# Function to apply Multiply blending with texture (simulating print effect)
+def apply_texture(img, texture):
     if img is None or texture is None:
         return None
 
-    gray_img = convert_to_grayscale(img)  # Convert input to grayscale
-    texture = texture.resize(img.size, Image.LANCZOS)  # Resize texture to match image
+    # Resize texture to match image size
+    texture = texture.resize(img.size, Image.LANCZOS)
 
-    # Convert images to arrays
-    gray_arr = np.array(gray_img, dtype=np.float32) / 255.0  # Normalize grayscale image
+    # Convert images to NumPy arrays
+    gray_arr = np.array(img, dtype=np.float32) / 255.0  # Normalize grayscale image
     texture_arr = np.array(texture, dtype=np.float32) / 255.0  # Normalize texture
 
-    # Detect white regions (where pixel intensity is high)
-    white_mask = gray_arr.mean(axis=2) > 0.8  # If pixel intensity > 80% (near white)
-
-    # Normal blending for non-white areas
-    blended = np.where(texture_arr <= 0.5, 2 * gray_arr * texture_arr, 1 - 2 * (1 - gray_arr) * (1 - texture_arr))
-
-    # Apply enhanced texture in white regions (90% texture, 10% grayscale)
-    enhanced_texture = (0.9 * texture_arr + 0.1 * gray_arr)
-
-    # Combine both approaches
-    final_arr = np.where(white_mask[..., None], enhanced_texture, blended)
-
-    return Image.fromarray((final_arr * 255).clip(0, 255).astype(np.uint8))
+    # Apply Multiply blending: Multiply each pixel and normalize back to 0-255
+    blended = gray_arr * texture_arr
 
-# Function to adjust contrast
-def adjust_contrast(img, contrast_level):
-    if img is None:
-        return None
-
-    enhancer = ImageEnhance.Contrast(img)
-    return enhancer.enhance(contrast_level)
+    # Convert array back to image
+    return Image.fromarray((blended * 255).astype(np.uint8))
 
 # Gradio GUI
 def main():
     with gr.Blocks() as demo:
-        gr.Markdown("## 🌫️ Enhanced Texture Effect with Contrast Adjustment")
+        gr.Markdown("## 🏡 Simulated Printing on Wood Texture (Multiply Blending)")
 
         with gr.Row():
             input_img = gr.Image(label="Input Image", type="pil")
-            texture_img = gr.Image(label="Upload Custom Texture", type="pil")
+            texture_img = gr.Image(label="Upload Wood Texture", type="pil")
 
-        intensity = gr.Slider(0.0, 1.0, value=0.6, label="Texture Strength")
-        contrast = gr.Slider(0.5, 2.0, value=1.0, label="Contrast Adjustment")
+        contrast = gr.Slider(0.0, 1.0, value=0.4, label="Contrast Adjustment")
 
         with gr.Row():
-            output_img1 = gr.Image(label="Grayscale + Texture Result", interactive=False)
-            output_img2 = gr.Image(label="Final Image with Contrast", interactive=False)
-
-        # Store intermediate texture result
-        texture_state = gr.State()
-
-        # Step 1: Apply texture with improved blending
-        def process_texture(img, texture, intensity):
-            result = apply_texture(img, texture, intensity)
-            return result, result  # Store for contrast adjustment
+            contrast_img = gr.Image(label="Contrast Adjusted Image", interactive=False)
+            grayscale_img = gr.Image(label="Grayscale Image", interactive=False)
+            output_img = gr.Image(label="Final Printed Effect", interactive=False)
 
-        # Step 2: Adjust contrast
+        # Step 1: Adjust contrast
         def process_contrast(img, contrast_level):
-            return adjust_contrast(img, contrast_level)
-
-        # Update texture-blended image
-        input_img.change(process_texture, inputs=[input_img, texture_img, intensity], outputs=[output_img1, texture_state])
-        texture_img.change(process_texture, inputs=[input_img, texture_img, intensity], outputs=[output_img1, texture_state])
-        intensity.change(process_texture, inputs=[input_img, texture_img, intensity], outputs=[output_img1, texture_state])
-
-        # Update final contrast image
-        contrast.change(process_contrast, inputs=[texture_state, contrast], outputs=output_img2)
+            contrast_img = adjust_contrast(img, contrast_level)
+            return contrast_img
+
+        # Step 2: Convert contrast-adjusted image to grayscale
+        def process_grayscale(img, contrast_level):
+            contrast_img = adjust_contrast(img, contrast_level)
+            grayscale_img = convert_to_grayscale(contrast_img)
+            return contrast_img, grayscale_img
+
+        # Step 3: Apply Multiply blending with texture
+        def process_texture(img, texture, contrast_level):
+            contrast_img = adjust_contrast(img, contrast_level)  # Adjust contrast first
+            grayscale_img = convert_to_grayscale(contrast_img)  # Convert to grayscale
+            result = apply_texture(grayscale_img, texture)  # Multiply blend with texture
+            return contrast_img, grayscale_img, result
+
+        # Update contrast-adjusted image
+        input_img.change(process_contrast, inputs=[input_img, contrast], outputs=contrast_img)
+
+        # Update grayscale image
+        input_img.change(process_grayscale, inputs=[input_img, contrast], outputs=[contrast_img, grayscale_img])
+        contrast.change(process_grayscale, inputs=[input_img, contrast], outputs=[contrast_img, grayscale_img])
+
+        # Update final processed image (Multiply blend with texture)
+        input_img.change(process_texture, inputs=[input_img, texture_img, contrast], outputs=[contrast_img, grayscale_img, output_img])
+        texture_img.change(process_texture, inputs=[input_img, texture_img, contrast], outputs=[contrast_img, grayscale_img, output_img])
+        contrast.change(process_texture, inputs=[input_img, texture_img, contrast], outputs=[contrast_img, grayscale_img, output_img])
 
     demo.launch()