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__)

# Ensure temp directory exists
TEMP_DIR = "temp"
os.makedirs(TEMP_DIR, exist_ok=True)

@app.route('/api')
def generate_profile():
    # 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)  # Profile picture URL

    # Load background
    bg = Image.open("bg.png").convert("RGBA")
    draw = ImageDraw.Draw(bg)
    
    # Load font
    font_large = ImageFont.truetype("font.ttf", 50)  # Larger font for name
    font_small = ImageFont.truetype("font.ttf", 40)  # Smaller font for details

    # Get image size
    W, H = bg.size  

    # Profile picture settings
    pfp_size = 150  # Size of the profile picture
    pfp_y = 50  # Position from top
    pfp_x = (W - pfp_size) // 2  # Centering horizontally

    # Load profile picture or fallback
    try:
        if pfp_url:
            response = requests.get(pfp_url, timeout=5)
            pfp = Image.open(BytesIO(response.content)).convert("RGBA")
        else:
            pfp = Image.open("fallback.png").convert("RGBA")  # Fallback image

        pfp = pfp.resize((pfp_size, pfp_size), Image.LANCZOS)

        # Create circular mask for profile picture
        mask = Image.new("L", (pfp_size, pfp_size), 0)
        draw_mask = ImageDraw.Draw(mask)
        draw_mask.ellipse((0, 0, pfp_size, pfp_size), fill=255)

        # Apply mask and paste profile picture
        pfp = Image.composite(pfp, Image.new("RGBA", pfp.size, (0, 0, 0, 0)), mask)
        bg.paste(pfp, (pfp_x, pfp_y), pfp)
    except Exception as e:
        print(f"Error loading profile picture: {e}")

    # Define text positions
    y_start = pfp_y + pfp_size + 30  # Start below the profile picture
    spacing = 80  # Vertical gap between lines
    text_color = (255, 255, 255)  # White color

    # Centered text positions
    def draw_centered_text(text, y, font):
        text_width, _ = draw.textsize(text, font=font)
        x = (W - text_width) // 2  # Center the text
        draw.text((x, y), text, font=font, fill=text_color)

    # Draw text
    draw_centered_text(f"Name: {name}", y_start, font_large)
    draw_centered_text(f"Rank: {rank}", y_start + spacing, font_small)
    draw_centered_text(f"Balance: {balance}", y_start + 2 * spacing, font_small)
    draw_centered_text(f"Next Rank: {next_rank}", y_start + 3 * spacing, font_small)

    # Generate a random filename
    output_path = os.path.join(TEMP_DIR, f"profile_{uuid.uuid4().hex}.png")
    bg.save(output_path)

    # Send and delete image after sending
    response = send_file(output_path, mimetype='image/png')
    os.remove(output_path)  # Clean up file after sending
    return response

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000, debug=True)