Update app.py

app.py

@@ -1,7 +1,7 @@
-from flask import Flask, request, jsonify, render_template, send_file, flash, redirect, url_for
 import os
 import cv2
 import numpy as np
+from flask import Flask, request, jsonify, render_template, send_file, url_for
 from werkzeug.utils import secure_filename
 
 app = Flask(__name__)
@@ -16,45 +16,51 @@ os.makedirs(RESULT_FOLDER, exist_ok=True)
 
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 app.config['RESULT_FOLDER'] = RESULT_FOLDER
-app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
+app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16 MB max
+
 
 def allowed_file(filename):
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
+
 def enhance_image(img, gamma=1.0, denoise='none', stylize=False):
     # Gamma correction
-    invGamma = 1.0 / gamma
+    invGamma = 1.0 / gamma if gamma > 0 else 1.0
     table = np.array([((i / 255.0) ** invGamma) * 255 for i in np.arange(256)]).astype("uint8")
     img = cv2.LUT(img, table)
 
-    # Enhance contrast (existing CLAHE)
+    # CLAHE for contrast
     lab = cv2.cvtColor(img, cv2.COLOR_BGR2LAB)
     l, a, b = cv2.split(lab)
-    clahe = cv2.createCLAHE(clipLimit=3.0, tileGridSize=(8,8))
+    clahe = cv2.createCLAHE(clipLimit=3.0, tileGridSize=(8, 8))
     cl = clahe.apply(l)
     img = cv2.merge((cl, a, b))
     img = cv2.cvtColor(img, cv2.COLOR_LAB2BGR)
 
-    # Denoise
+    # Denoise options
     if denoise == 'nonlocal':
-        img = cv2.fastNlMeansDenoisingColored(img,None,10,10,7,21)
+        img = cv2.fastNlMeansDenoisingColored(img, None, 10, 10, 7, 21)
     elif denoise == 'median':
         img = cv2.medianBlur(img, 3)
 
-    # Stylize (cartoon effect)
+    # Stylize option (cartoon/sketch effect)
     if stylize:
-        img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-        img_blur = cv2.medianBlur(img_gray, 7)
-        edges = cv2.adaptiveThreshold(img_blur,255,cv2.ADAPTIVE_THRESH_MEAN_C, cv2.THRESH_BINARY, 9, 2)
+        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+        blur = cv2.medianBlur(gray, 7)
+        edges = cv2.adaptiveThreshold(blur, 255,
+                                      cv2.ADAPTIVE_THRESH_MEAN_C,
+                                      cv2.THRESH_BINARY,
+                                      9, 2)
         color = cv2.bilateralFilter(img, 9, 250, 250)
         img = cv2.bitwise_and(color, color, mask=edges)
 
     return img
 
+
 def generate_damage_mask(img, threshold=500):
     gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-    tophat_kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (17,17))
-    tophat = cv2.morphologyEx(gray, cv2.MORPH_TOPHAT, tophat_kernel)
+    kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (17, 17))
+    tophat = cv2.morphologyEx(gray, cv2.MORPH_TOPHAT, kernel)
 
     grad_x = cv2.Sobel(gray, cv2.CV_16S, 1, 0)
     grad_y = cv2.Sobel(gray, cv2.CV_16S, 0, 1)
@@ -65,19 +71,22 @@ def generate_damage_mask(img, threshold=500):
     combined = cv2.addWeighted(tophat, 1.0, gradient, 1.0, 0)
     _, mask = cv2.threshold(combined, threshold, 255, cv2.THRESH_BINARY)
     mask = cv2.medianBlur(mask, 5)
-    mask = cv2.dilate(mask, np.ones((5,5), np.uint8), iterations=1)
+    mask = cv2.dilate(mask, np.ones((5, 5), np.uint8), iterations=1)
 
     return mask
 
+
 def inpaint_damage(img, threshold):
     mask = generate_damage_mask(img, threshold)
     repaired = cv2.inpaint(img, mask, 3, cv2.INPAINT_TELEA)
     return repaired, mask
 
+
 @app.route('/', methods=['GET'])
 def index():
     return render_template('index.html')
 
+
 @app.route('/upload', methods=['POST'])
 def upload():
     if 'file' not in request.files:
@@ -96,7 +105,7 @@ def upload():
         if img is None:
             return jsonify(error='Invalid image file'), 400
 
-        # Initial parameters, defaults
+        # Default parameters for first process
         threshold = 500
         gamma = 1.0
         denoise = 'none'
@@ -114,13 +123,15 @@ def upload():
         cv2.imwrite(mask_path, mask)
 
         return jsonify(
-            original_image = url_for('uploaded_file', filename=filename),
-            repaired_image = url_for('result_file', filename=repaired_filename),
-            mask_image = url_for('result_file', filename=mask_filename)
+            original_image=url_for('uploaded_file', filename=filename),
+            repaired_image=url_for('result_file', filename=repaired_filename),
+            mask_image=url_for('result_file', filename=mask_filename)
         )
+
     else:
         return jsonify(error='Allowed image types are png, jpg, jpeg'), 400
 
+
 @app.route('/process', methods=['POST'])
 def process():
     data = request.get_json()
@@ -150,14 +161,16 @@ def process():
     cv2.imwrite(mask_path, mask)
 
     return jsonify(
-        repaired_image = url_for('result_file', filename=repaired_filename),
-        mask_image = url_for('result_file', filename=mask_filename)
+        repaired_image=url_for('result_file', filename=repaired_filename),
+        mask_image=url_for('result_file', filename=mask_filename)
     )
 
+
 @app.route('/uploads/<filename>')
 def uploaded_file(filename):
     return send_file(os.path.join(app.config['UPLOAD_FOLDER'], filename))
 
+
 @app.route('/results/<filename>')
 def result_file(filename):
     return send_file(os.path.join(app.config['RESULT_FOLDER'], filename))