Update app.py

app.py

@@ -455,10 +455,13 @@ def predict(
             except Exception:
                 raise ValueError("Invalid base64 image data")
 
-    # Apply only sharpness enhancement (leave brightness at 1)
+    # Preprocessing: reduce brightness to 0.5 and enhance sharpness.
     if isinstance(image, np.ndarray):
         pil_image = Image.fromarray(image)
-        enhanced_image = ImageEnhance.Sharpness(pil_image).enhance(1.5)
+        # Reduce brightness.
+        dark_image = ImageEnhance.Brightness(pil_image).enhance(0.5)
+        # Enhance sharpness.
+        enhanced_image = ImageEnhance.Sharpness(dark_image).enhance(1.5)
         image = np.array(enhanced_image)
 
     # ---------------------
@@ -679,6 +682,13 @@ def predict(
             cv2.LINE_AA
         )
 
+    # Restore brightness for display purposes:
+    # Since we reduced brightness by 0.5 during preprocessing,
+    # we apply an enhancement factor of 2.0 here to bring it back.
+    display_img = Image.fromarray(output_img)
+    display_img = ImageEnhance.Brightness(display_img).enhance(2.0)
+    output_img = np.array(display_img)
+
     outlines_color = cv2.cvtColor(new_outlines, cv2.COLOR_BGR2RGB)
     print("Total prediction time: {:.2f} seconds".format(time.time() - overall_start))