Update myapp.py

myapp.py

@@ -1,6 +1,8 @@
 from flask import Flask, jsonify, request, send_file
 from flask_cors import CORS
 from PIL import Image
+import numpy as np
+import cv2
 import io
 
 # Initialize the Flask app
@@ -9,7 +11,7 @@ CORS(myapp)  # Enable CORS if needed
 
 @myapp.route('/')
 def home():
-    return "Welcome to the Image Upscaler!"  # Basic home response
+    return "Welcome to the Image Upscaler and Noise Reducer!"
 
 @myapp.route('/upscale', methods=['POST'])
 def upscale_image():
@@ -23,13 +25,18 @@ def upscale_image():
     # Open the image using PIL
     img = Image.open(io.BytesIO(input_image))
 
-    # Resize the image to upscale (you can adjust the method as needed)
+    # Resize the image to upscale
     upscaled_img = img.resize((img.width * width, img.height * height), Image.LANCZOS)
-    
-    # Save to bytes
-    img_byte_arr = io.BytesIO()
-    upscaled_img.save(img_byte_arr, format='PNG')
-    img_byte_arr.seek(0)
+
+    # Convert PIL image to numpy array for OpenCV processing
+    np_img = np.array(upscaled_img)
+
+    # Apply median filtering for noise reduction
+    filtered_img = cv2.medianBlur(np_img, 5)  # The second parameter is the kernel size
+
+    # Convert the result back to PIL Image for sending
+    _, buffer = cv2.imencode('.png', filtered_img)
+    img_byte_arr = io.BytesIO(buffer)
 
     return send_file(img_byte_arr, mimetype='image/png')