Update app.py

app.py

@@ -1,9 +1,7 @@
 import os
 import uuid
-import requests
 from flask import Flask, request, send_file
 from PIL import Image, ImageDraw, ImageFont
-from io import BytesIO
 
 app = Flask(__name__)
 
@@ -11,6 +9,10 @@ app = Flask(__name__)
 TEMP_DIR = "temp"
 os.makedirs(TEMP_DIR, exist_ok=True)
 
+# Available images
+PFP_OPTIONS = {str(i): f"pfp{i}.png" for i in range(1, 6)}
+BG_OPTIONS = {str(i): f"bg{i}.png" for i in range(1, 4)}
+
 # Load fonts safely
 def load_font(font_path, size):
     try:
@@ -20,19 +22,20 @@ def load_font(font_path, size):
 
 @app.route('/api')
 def generate_profile():
-    """ Generate a profile card with user details and a profile picture. """
+    """ Generate a profile card with user details and a chosen profile picture & background. """
     # Get query parameters
     name = request.args.get('name', 'Unknown')
     rank = request.args.get('rank', 'Unranked')
     balance = request.args.get('balance', '0')
     next_rank = request.args.get('next_rank', 'Unknown')
-    pfp_url = request.args.get('pfp', None)
+    pfp_choice = request.args.get('pfp', '1')
+    bg_choice = request.args.get('bg', '1')
 
-    # Load background image
-    try:
-        bg = Image.open("bg.png").convert("RGBA")  # Ensure transparency support
-    except IOError:
-        return "Error: Background image 'bg.png' not found.", 500
+    # Validate and load background
+    bg_path = BG_OPTIONS.get(bg_choice, "bg1.png")
+    if not os.path.exists(bg_path):
+        return f"Error: Background '{bg_choice}' not found.", 400
+    bg = Image.open(bg_path).convert("RGBA")
 
     draw = ImageDraw.Draw(bg)
     W, H = bg.size  # Get background dimensions
@@ -47,44 +50,32 @@ def generate_profile():
     pfp_x = 50
     pfp_y = H // 2 - pfp_size // 2  # Center vertically
 
-    # Download and save profile picture if provided
-    pfp_path = os.path.join(TEMP_DIR, f"pfp_{uuid.uuid4().hex}.png")
-
-    try:
-        if pfp_url:
-            response = requests.get(pfp_url, timeout=5)
-            if response.status_code == 200:
-                with open(pfp_path, "wb") as f:
-                    f.write(response.content)
-                pfp = Image.open(pfp_path).convert("RGBA")
-            else:
-                raise ValueError("Invalid profile picture URL")
-        else:
-            pfp = Image.open("fallback.png").convert("RGBA")
-
-        # Resize and mask the profile picture (circular crop)
-        pfp = pfp.resize((pfp_size, pfp_size), Image.LANCZOS)
-        mask = Image.new("L", (pfp_size, pfp_size), 0)
-        mask_draw = ImageDraw.Draw(mask)
-        mask_draw.ellipse((0, 0, pfp_size, pfp_size), fill=255)
-
-        # Create a circular profile picture
-        circular_pfp = Image.new("RGBA", (pfp_size, pfp_size), (0, 0, 0, 0))
-        circular_pfp.paste(pfp, (0, 0), mask)
-
-        # Create a white border
-        border_size = 10
-        border = Image.new("RGBA", (pfp_size + border_size, pfp_size + border_size), (255, 255, 255, 255))
-        border_mask = Image.new("L", (pfp_size + border_size, pfp_size + border_size), 0)
-        border_draw = ImageDraw.Draw(border_mask)
-        border_draw.ellipse((0, 0, pfp_size + border_size, pfp_size + border_size), fill=255)
-        border.paste(circular_pfp, (border_size // 2, border_size // 2), mask)
-
-        # Paste profile picture onto background
-        bg.paste(border, (pfp_x - border_size // 2, pfp_y - border_size // 2), border_mask)
-    except Exception as e:
-        print(f"Error loading profile picture: {e}")
-        pfp_path = None  # Ensure the file is not deleted if it wasn't created
+    # Validate and load profile picture
+    pfp_path = PFP_OPTIONS.get(pfp_choice, "pfp1.png")
+    if not os.path.exists(pfp_path):
+        return f"Error: Profile picture '{pfp_choice}' not found.", 400
+    pfp = Image.open(pfp_path).convert("RGBA")
+
+    # Resize and mask the profile picture (circular crop)
+    pfp = pfp.resize((pfp_size, pfp_size), Image.LANCZOS)
+    mask = Image.new("L", (pfp_size, pfp_size), 0)
+    mask_draw = ImageDraw.Draw(mask)
+    mask_draw.ellipse((0, 0, pfp_size, pfp_size), fill=255)
+
+    # Create a circular profile picture
+    circular_pfp = Image.new("RGBA", (pfp_size, pfp_size), (0, 0, 0, 0))
+    circular_pfp.paste(pfp, (0, 0), mask)
+
+    # Create a white border
+    border_size = 10
+    border = Image.new("RGBA", (pfp_size + border_size, pfp_size + border_size), (255, 255, 255, 255))
+    border_mask = Image.new("L", (pfp_size + border_size, pfp_size + border_size), 0)
+    border_draw = ImageDraw.Draw(border_mask)
+    border_draw.ellipse((0, 0, pfp_size + border_size, pfp_size + border_size), fill=255)
+    border.paste(circular_pfp, (border_size // 2, border_size // 2), mask)
+
+    # Paste profile picture onto background
+    bg.paste(border, (pfp_x - border_size // 2, pfp_y - border_size // 2), border_mask)
 
     # **TEXT POSITIONING**
     text_x = pfp_x + pfp_size + 40  # Place text beside the profile picture
@@ -100,9 +91,10 @@ def generate_profile():
         return font
 
     # Adjust name font size dynamically
-    name_font = fit_text(name, W - text_x - 50, font_large)
+    max_text_width = W - text_x - 50
+    name_font = fit_text(name, max_text_width, font_large)
 
-    # **Draw text**
+    # **Draw text (Ensuring alignment)**
     draw.text((text_x, text_y), name, font=name_font, fill=text_color)  
     draw.text((text_x, text_y + text_spacing), f"Rank: {rank}", font=font_medium, fill=text_color)  
     draw.text((text_x, text_y + 2 * text_spacing), f"Balance: {balance}", font=font_medium, fill=text_color)  
@@ -118,8 +110,6 @@ def generate_profile():
     # Cleanup saved files
     if os.path.exists(output_path):
         os.remove(output_path)
-    if pfp_path and os.path.exists(pfp_path):  # Delete downloaded pfp if it was created
-        os.remove(pfp_path)
 
     return response