Update app.py

app.py

@@ -13,9 +13,17 @@ import gradio as gr
 # Function to convert image to sketch
 def convert_to_sketch(img, blur_strength, brightness, contrast):
     try:
+        # Add debug print
+        print(f"Input image shape: {np.array(img).shape}")
+        
         # Convert PIL Image to numpy array (BGR format for OpenCV)
         img = np.array(img)
         
+        # Check if image is empty
+        if img is None or img.size == 0:
+            print("Error: Empty input image")
+            return None
+            
         # Ensure blur_strength is odd
         blur_strength = max(1, int(blur_strength))
         if blur_strength % 2 == 0:
@@ -23,6 +31,7 @@ def convert_to_sketch(img, blur_strength, brightness, contrast):
             
         # Convert the image to grayscale
         img_gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
+        print(f"Grayscale shape: {img_gray.shape}")
         
         # Invert the grayscale image
         img_inverted = 255 - img_gray
@@ -31,22 +40,25 @@ def convert_to_sketch(img, blur_strength, brightness, contrast):
         img_blur = cv2.GaussianBlur(img_inverted, (blur_strength, blur_strength), 0)
         
         # Blend the grayscale and blurred inverted images
-        # Avoid division by zero by adding a small constant
         denominator = 255 - img_blur
         denominator[denominator == 0] = 1  # Prevent division by zero
         
-        img_blend = cv2.multiply(img_gray, 256.0 / denominator)
+        img_blend = (img_gray * 256.0) / denominator
         
         # Clip values to valid range
         img_blend = np.clip(img_blend, 0, 255).astype(np.uint8)
+        print(f"Blend shape: {img_blend.shape}")
         
         # Adjust brightness and contrast
         sketch_with_bg = adjust_brightness_contrast(img_blend, brightness, contrast)
+        print(f"Final output shape: {sketch_with_bg.shape}")
         
         return sketch_with_bg
         
     except Exception as e:
         print(f"Error in convert_to_sketch: {str(e)}")
+        import traceback
+        traceback.print_exc()
         return None
 
 # Function to adjust brightness and contrast
@@ -56,14 +68,14 @@ def adjust_brightness_contrast(img, brightness, contrast):
         brightness = float(brightness)
         contrast = float(contrast)
         
-        # Apply contrast first
-        img = cv2.convertScaleAbs(img, alpha=contrast)
+        # Convert to float32 for calculations
+        img = img.astype(float)
+        
+        # Apply contrast
+        img = img * contrast
         
         # Apply brightness
-        if brightness > 0:
-            img = cv2.add(img, np.ones_like(img) * brightness)
-        else:
-            img = cv2.subtract(img, np.ones_like(img) * abs(brightness))
+        img = img + brightness
             
         # Ensure output is in valid range
         img = np.clip(img, 0, 255).astype(np.uint8)
@@ -71,6 +83,8 @@ def adjust_brightness_contrast(img, brightness, contrast):
         
     except Exception as e:
         print(f"Error in adjust_brightness_contrast: {str(e)}")
+        import traceback
+        traceback.print_exc()
         return None
 
 # Gradio interface function
@@ -80,6 +94,9 @@ def sketch_interface(image, blur_strength, brightness, contrast):
         return None
         
     try:
+        print(f"Input image type: {type(image)}")
+        print(f"Input image size: {image.size}")
+        
         # Convert the input image to a sketch with adjustments
         sketch = convert_to_sketch(image, blur_strength, brightness, contrast)
         
@@ -89,11 +106,14 @@ def sketch_interface(image, blur_strength, brightness, contrast):
             
         # Convert the processed numpy array back to a PIL Image
         output_image = Image.fromarray(sketch)
+        print(f"Output image size: {output_image.size}")
         
         return output_image
         
     except Exception as e:
         print(f"Error in sketch_interface: {str(e)}")
+        import traceback
+        traceback.print_exc()
         return None
 
 # Create Gradio interface