app.py

from flask import Flask, render_template, request, send_from_directory, jsonify
from PIL import Image
import os
import uuid

app = Flask(__name__)

# Ensure output folders exist
os.makedirs('static/converted', exist_ok=True)
os.makedirs('static/stickers', exist_ok=True)

@app.route('/')
def index():
    return render_template('index.html')


@app.route('/convert', methods=['POST'])
def convert():
    image = request.files['image']
    output_format = request.form['format']  # png, jpg, gif, webp
    if image:
        img = Image.open(image)
        filename = f"{uuid.uuid4()}.{output_format}"
        save_path = os.path.join('static/converted', filename)

        # Convert to RGB if saving to JPEG (no transparency)
        if output_format == 'jpg':
            img = img.convert('RGB')
        
        img.save(save_path)
        return jsonify({"status": "success", "url": '/' + save_path})
    return jsonify({"status": "error"})


@app.route('/sticker', methods=['POST'])
def sticker():
    image = request.files['image']
    if image:
        img = Image.open(image).convert('RGBA')  # ensure transparency
        img = img.resize((512, 512))

        filename = f"{uuid.uuid4()}.png"
        save_path = os.path.join('static/stickers', filename)
        img.save(save_path, format='PNG')

        return jsonify({"status": "success", "url": '/' + save_path})
    return jsonify({"status": "error"})


if __name__ == '__main__':
    app.run(debug=True)