Update app.py

app.py

@@ -15,87 +15,79 @@ def modern_to_nokia_pil(
     gamma=1.3,
     add_date_stamp=True,
 ):
-    # Ensure RGB
     img = img.convert("RGB")
 
-    # --- Crop to 4:3 like old phones ---
+    # --- Crop to 4:3 ---
     w, h = img.size
     target_ratio = 4 / 3
     current_ratio = w / h
 
     if current_ratio > target_ratio:
-        # too wide: crop width
         new_w = int(h * target_ratio)
         left = (w - new_w) // 2
         img = img.crop((left, 0, left + new_w, h))
     else:
-        # too tall: crop height
         new_h = int(w / target_ratio)
         top = (h - new_h) // 2
         img = img.crop((0, top, w, top + new_h))
 
-    # --- Resize down to ~0.3 MP (640x480) ---
+    # --- Resize to 640×480 ---
     img = img.resize((640, 480), resample=Image.BILINEAR)
 
-    # --- Slight blur for soft lens ---
+    # --- Blur ---
     img = img.filter(ImageFilter.GaussianBlur(radius=blur_radius))
 
-    # --- Reduce color depth (posterization) ---
+    # --- Posterize (reduce color depth) ---
     arr = np.array(img, dtype=np.uint8)
-    if color_levels < 2:
-        color_levels = 2
-    step = max(1, 256 // int(color_levels))  # step size
+    step = max(1, 256 // int(color_levels))
     arr = (arr // step) * step
     img = Image.fromarray(arr, mode="RGB")
 
-    # --- Add noise ---
+    # --- Noise ---
     arr = np.array(img, dtype=np.int16)
     noise = np.random.normal(0, noise_std, arr.shape)
-    arr = arr + noise
-    arr = np.clip(arr, 0, 255).astype(np.uint8)
+    arr = np.clip(arr + noise, 0, 255).astype(np.uint8)
     img = Image.fromarray(arr, mode="RGB")
 
-    # --- Crush dynamic range (gamma curve) ---
+    # --- Gamma curve ---
     arr = np.array(img, dtype=np.float32) / 255.0
-    arr = np.power(arr, gamma)  # gamma >1 = darker mids
+    arr = np.power(arr, gamma)
     arr = np.clip(arr * 255, 0, 255).astype(np.uint8)
     img = Image.fromarray(arr, mode="RGB")
 
-    # --- Optional yellow date stamp in corner ---
+    # --- Date stamp (fixed for Pillow >= 10) ---
     if add_date_stamp:
         draw = ImageDraw.Draw(img)
         font = ImageFont.load_default()
-        now = datetime.datetime.now()
-        text = now.strftime("%d-%m-%y %H:%M")  # Nokia-style
-        # text size
-        text_w, text_h = draw.textsize(text, font=font)
+        text = datetime.datetime.now().strftime("%d-%m-%y %H:%M")
+
+        # NEW SAFE SIZE METHOD
+        bbox = draw.textbbox((0, 0), text, font=font)
+        text_w = bbox[2] - bbox[0]
+        text_h = bbox[3] - bbox[1]
+
         margin = 4
         x = img.width - text_w - margin
         y = img.height - text_h - margin
-        # black box behind text
+
         draw.rectangle(
             [x - 2, y - 2, x + text_w + 2, y + text_h + 2],
             fill=(0, 0, 0),
         )
-        # yellow-ish text
         draw.text((x, y), text, font=font, fill=(255, 255, 0))
 
-    # --- Save with low JPEG quality to get artifacts ---
+    # --- JPEG compression ---
     buffer = io.BytesIO()
-    img.save(
-        buffer,
-        format="JPEG",
-        quality=int(jpeg_quality),
-        optimize=False,
-        progressive=False,
-    )
+    img.save(buffer, format="JPEG", quality=int(jpeg_quality),
+             optimize=False, progressive=False)
     buffer.seek(0)
     img = Image.open(buffer)
 
     return img
 
 
-def convert_fn(image, jpeg_quality, noise_std, color_levels, blur_radius, gamma, add_date_stamp):
+def convert_fn(image, jpeg_quality, noise_std, color_levels,
+               blur_radius, gamma, add_date_stamp):
     if image is None:
         return None
     return modern_to_nokia_pil(
@@ -125,6 +117,5 @@ demo = gr.Interface(
     description="Upload a modern photo and turn it into an old-school 0.3 MP Nokia-style photo.",
 )
 
-
 if __name__ == "__main__":
     demo.launch()