from flask import Flask, request, render_template, send_file from PIL import Image, ImageOps from io import BytesIO from dotenv import load_dotenv import requests import cloudinary import cloudinary.uploader import cloudinary.utils import os app = Flask(__name__) REMOVE_BG_API_KEY = os.getenv("REMOVE_BG_API_KEY") cloudinary.config( cloud_name=os.getenv("CLOUDINARY_CLOUD_NAME"), api_key=os.getenv("CLOUDINARY_API_KEY"), api_secret=os.getenv("CLOUDINARY_API_SECRET"), ) @app.route("/") def index(): return render_template("index.html") # নতুন ফাংশন: HTML Hex Code (যেমন #3b82f6) কে RGB তে কনভার্ট করার জন্য def hex_to_rgb(hex_color): hex_color = hex_color.lstrip('#') if len(hex_color) != 6: return (255, 255, 255) # ডিফল্ট সাদা return tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4)) # ফাংশনে নতুন প্যারামিটার bg_color_hex যোগ করা হয়েছে def process_single_image(input_image_bytes, bg_color_hex="#ffffff"): """Remove background, enhance, apply selected background color, and return a PIL image.""" # ইউজার সিলেক্টেড কালার কনভার্ট করা হচ্ছে bg_color_tuple = hex_to_rgb(bg_color_hex) # Step 1: Background removal response = requests.post( "https://api.remove.bg/v1.0/removebg", files={"image_file": input_image_bytes}, data={"size": "auto"}, headers={"X-Api-Key": REMOVE_BG_API_KEY}, ) if response.status_code != 200: try: error_info = response.json() if error_info.get("errors"): error_code = error_info["errors"][0].get("code", "unknown_error") raise ValueError(f"bg_removal_failed:{error_code}:{response.status_code}") except ValueError: raise except Exception: pass raise ValueError(f"bg_removal_failed:unknown:{response.status_code}") bg_removed = BytesIO(response.content) img = Image.open(bg_removed) # আগে এখানে হার্ডকোডেড (255,255,255) ছিল, এখন ডাইনামিক bg_color_tuple দেওয়া হয়েছে if img.mode in ("RGBA", "LA"): background = Image.new("RGB", img.size, bg_color_tuple) background.paste(img, mask=img.split()[-1]) processed_img = background else: processed_img = img.convert("RGB") # Step 2: Upload to Cloudinary buffer = BytesIO() processed_img.save(buffer, format="PNG") buffer.seek(0) upload_result = cloudinary.uploader.upload(buffer, resource_type="image") image_url = upload_result.get("secure_url") public_id = upload_result.get("public_id") if not image_url: raise ValueError("cloudinary_upload_failed") # Step 3: Enhance via Cloudinary AI enhanced_url = cloudinary.utils.cloudinary_url( public_id, transformation=[ {"effect": "gen_restore"}, {"quality": "auto"}, {"fetch_format": "auto"}, ], )[0] enhanced_img_data = requests.get(enhanced_url).content img = Image.open(BytesIO(enhanced_img_data)) # এখানেও ডাইনামিক কালার বসানো হয়েছে if img.mode in ("RGBA", "LA"): background = Image.new("RGB", img.size, bg_color_tuple) background.paste(img, mask=img.split()[-1]) passport_img = background else: passport_img = img.convert("RGB") return passport_img @app.route("/process", methods=["POST"]) def process(): print("==== /process endpoint hit ====") # Layout settings passport_width = int(request.form.get("width", 390)) passport_height = int(request.form.get("height", 480)) border = int(request.form.get("border", 2)) spacing = int(request.form.get("spacing", 10)) margin_x = 10 margin_y = 10 horizontal_gap = 10 a4_w, a4_h = 2480, 3508 # Collect images, copies, and background colors images_data = [] # Multi-image mode i = 0 while f"image_{i}" in request.files: file = request.files[f"image_{i}"] copies = int(request.form.get(f"copies_{i}", 6)) # ফ্রন্টএন্ড থেকে bgColor রিসিভ করা হচ্ছে (না থাকলে ডিফল্ট সাদা) bg_color = request.form.get(f"bgColor_{i}", "#ffffff") images_data.append((file.read(), copies, bg_color)) i += 1 # Fallback to single image mode if not images_data and "image" in request.files: file = request.files["image"] copies = int(request.form.get("copies", 6)) bg_color = request.form.get("bgColor", "#ffffff") images_data.append((file.read(), copies, bg_color)) if not images_data: return "No image uploaded", 400 print(f"DEBUG: Processing {len(images_data)} image(s)") # Process all images passport_images = [] for idx, (img_bytes, copies, bg_color) in enumerate(images_data): print(f"DEBUG: Processing image {idx + 1} with {copies} copies and {bg_color} background") try: # ফাংশনে bg_color পাস করা হচ্ছে img = process_single_image(img_bytes, bg_color) img = img.resize((passport_width, passport_height), Image.LANCZOS) img = ImageOps.expand(img, border=border, fill="black") passport_images.append((img, copies)) except ValueError as e: err_str = str(e) if "410" in err_str or "face" in err_str.lower(): return {"error": "face_detection_failed"}, 410 elif "429" in err_str or "quota" in err_str.lower(): return {"error": "quota_exceeded"}, 429 else: print(err_str) return {"error": err_str}, 500 paste_w = passport_width + 2 * border paste_h = passport_height + 2 * border # Build all pages pages = [] current_page = Image.new("RGB", (a4_w, a4_h), "white") x, y = margin_x, margin_y def new_page(): nonlocal current_page, x, y pages.append(current_page) current_page = Image.new("RGB", (a4_w, a4_h), "white") x, y = margin_x, margin_y for passport_img, copies in passport_images: for _ in range(copies): # Move to next row if needed if x + paste_w > a4_w - margin_x: x = margin_x y += paste_h + spacing # Move to next page if needed if y + paste_h > a4_h - margin_y: new_page() current_page.paste(passport_img, (x, y)) print(f"DEBUG: Placed at x={x}, y={y}") x += paste_w + horizontal_gap pages.append(current_page) print(f"DEBUG: Total pages = {len(pages)}") # Export multi-page PDF output = BytesIO() if len(pages) == 1: pages[0].save(output, format="PDF", dpi=(300, 300)) else: pages[0].save( output, format="PDF", dpi=(300, 300), save_all=True, append_images=pages[1:], ) output.seek(0) print("DEBUG: Returning PDF to client") return send_file( output, mimetype="application/pdf", as_attachment=True, download_name="passport-sheet.pdf", ) if __name__ == "__main__": app.run(host="0.0.0.0", port=5000, debug=True)