Update app.py

app.py

@@ -13,7 +13,6 @@ UPLOAD_FOLDER = os.path.join(STATIC_FOLDER, 'uploads')
 OUTPUT_FOLDER = os.path.join(STATIC_FOLDER, 'outputs')
 MODEL_FOLDER = os.path.join(BASE_DIR, 'model')
 
-os.makedirs(STATIC_FOLDER, exist_ok=True)
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(OUTPUT_FOLDER, exist_ok=True)
 
@@ -22,12 +21,15 @@ protoPath = os.path.join(MODEL_FOLDER, 'colorization_deploy_v2.prototxt')
 modelPath = os.path.join(MODEL_FOLDER, 'colorization_release_v2.caffemodel')
 hullPath = os.path.join(MODEL_FOLDER, 'pts_in_hull.npy')
 
-assert os.path.exists(protoPath), f"Missing proto file: {protoPath}"
-assert os.path.exists(modelPath), f"Missing model file: {modelPath}"
-assert os.path.exists(hullPath), f"Missing hull file: {hullPath}"
-
 COLORIZATION_MODEL_AVAILABLE = False
+net = None
+
+# --- Load model ---
 try:
+    assert os.path.exists(protoPath), f"Missing proto file: {protoPath}"
+    assert os.path.exists(modelPath), f"Missing model file: {modelPath}"
+    assert os.path.exists(hullPath), f"Missing hull file: {hullPath}"
+
     print("Loading colorization model...")
     net = cv2.dnn.readNetFromCaffe(protoPath, modelPath)
     pts_in_hull = np.load(hullPath)
@@ -36,10 +38,8 @@ try:
         raise ValueError(f"pts_in_hull shape invalid: {pts_in_hull.shape}")
 
     pts = pts_in_hull.transpose().reshape(2, 313, 1, 1).astype(np.float32)
-
     net.getLayer(net.getLayerId('class8_ab')).blobs = [pts]
     net.getLayer(net.getLayerId('conv8_313_rh')).blobs = [np.full([1, 313], 2.606, dtype=np.float32)]
-
     COLORIZATION_MODEL_AVAILABLE = True
     print("Colorization model loaded successfully.")
 except Exception as e:
@@ -59,33 +59,28 @@ def adjust_brightness_contrast(img, brightness=0, contrast=20):
     return cv2.convertScaleAbs(img, alpha=alpha, beta=brightness)
 
 def colorize_image_local(input_path, output_path):
-    print(f"Reading image for colorization: {input_path}")
-    img = cv2.imread(input_path)
-    if img is None:
-        print("ERROR: Could not read input image (file may be corrupt or unreadable by OpenCV).")
-        return False
-    else:
-        print(f"Image loaded successfully, shape: {img.shape}")
-
     if not COLORIZATION_MODEL_AVAILABLE:
         print("Colorization model not available.")
         return False
 
+    img = cv2.imread(input_path)
+    if img is None:
+        print(f"Failed to read image: {input_path}")
+        return False
+
     img = resize_img(img)
     h, w = img.shape[:2]
 
     img_rgb = img.astype("float32") / 255.0
     lab = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2LAB)
     L = lab[:, :, 0]
-
-    L_resized = cv2.resize(L, (224, 224))
-    L_resized -= 50  # mean-centering
+    L_resized = cv2.resize(L, (224, 224)) - 50  # mean-centering
 
     try:
         net.setInput(cv2.dnn.blobFromImage(L_resized))
         ab = net.forward()[0].transpose(1, 2, 0)
     except Exception as e:
-        print(f"ERROR during DNN forward pass: {e}")
+        print(f"DNN forward pass failed: {e}")
         return False
 
     ab = cv2.resize(ab, (w, h))
@@ -94,56 +89,30 @@ def colorize_image_local(input_path, output_path):
     out_bgr = np.clip(out_bgr * 255, 0, 255).astype("uint8")
     out_bgr = adjust_brightness_contrast(out_bgr, brightness=-10, contrast=15)
     cv2.imwrite(output_path, out_bgr)
-    print(f"Colorized image saved to {output_path}")
     return True
 
 @app.route('/', methods=['GET', 'POST'])
 def index():
-
-    print(f"protoPath: {protoPath}, exists? {os.path.exists(protoPath)}")
-    print(f"modelPath: {modelPath}, exists? {os.path.exists(modelPath)}")
-    print(f"hullPath: {hullPath}, exists? {os.path.exists(hullPath)}")
-    print(f"UPLOAD_FOLDER: {UPLOAD_FOLDER}, exists? {os.path.exists(UPLOAD_FOLDER)}")
-    print(f"OUTPUT_FOLDER: {OUTPUT_FOLDER}, exists? {os.path.exists(OUTPUT_FOLDER)}")
-    print(f"COLORIZATION_MODEL_AVAILABLE: {COLORIZATION_MODEL_AVAILABLE}")
-    print(f"cv2 version: {cv2.__version__}")
-
     original_url = enhanced_url = None
     if request.method == 'POST':
         file = request.files.get('image')
-        if file and file.filename:
-            filename = secure_filename(file.filename)
-            orig_path = os.path.join(UPLOAD_FOLDER, filename)
-            print(f"Saving file to {orig_path}")
-            try:
-                file.save(orig_path)
-                print(f"File saved successfully.")
-            except Exception as e:
-                print(f"ERROR: Failed to save file: {e}")
-                return "File save failed", 400
-
-            if os.path.exists(orig_path):
-                print(f"Saved file size: {os.path.getsize(orig_path)} bytes")
-            else:
-                print("ERROR: File does not exist after saving!")
-                return "File save failed", 400
-
-            output_name = f"output_{filename}"
-            output_path = os.path.join(OUTPUT_FOLDER, output_name)
-
-
-            if not colorize_image_local(orig_path, output_path):
-                print("Colorization failed inside colorize_image_local")
-                return "Colorization failed", 400
-
-            original_url = url_for('static', filename=f'uploads/{filename}')
-            enhanced_url = url_for('static', filename=f'outputs/{output_name}')
-        else:
-            print("No file uploaded or empty filename.")
+        if not file or not file.filename:
             return "No file uploaded", 400
 
-    return render_template('index.html', original_url=original_url, enhanced_url=enhanced_url)
+        filename = secure_filename(file.filename)
+        orig_path = os.path.join(UPLOAD_FOLDER, filename)
+        file.save(orig_path)
+
+        output_name = f"output_{filename}"
+        output_path = os.path.join(OUTPUT_FOLDER, output_name)
+
+        if not colorize_image_local(orig_path, output_path):
+            return "Colorization failed", 400
 
+        original_url = url_for('static', filename=f'uploads/{filename}')
+        enhanced_url = url_for('static', filename=f'outputs/{output_name}')
+
+    return render_template('index.html', original_url=original_url, enhanced_url=enhanced_url)
 
 if __name__ == '__main__':
     port = int(os.environ.get("PORT", 7860))