import gradio as gr
import qrcode
from PIL import Image, ImageDraw, ImageFont
import time
import os

def generate_upi_qr(upi_code, amount, merchant_name=None):
    if not merchant_name:
        merchant_name = upi_code.split('@')[0]

    # Generate UPI QR code
    qr = qrcode.QRCode(version=1, box_size=10, border=5)
    qr_data = f"upi://pay?pa={upi_code}&pn={merchant_name}&am={amount}&cu=INR"

    qr.add_data(qr_data)
    qr.make(fit=True)
    qr_image = qr.make_image(fill="black", back_color="white").convert("RGB")

    # # Generate unique image name based on UPI ID, amount, and timestamp
    # timestamp = str(int(time.time()))
    # image_name = f"qr_{upi_code}_{amount}_{timestamp}.png"

    # # Load your template
    # template = Image.open(gradio.interface.assets.get("Asset 6-1.png")).convert("RGB")

    # # Resize the QR code if needed
    # qr_image = qr_image.resize((700, 700))

    # # Paste the QR code onto your template
    # template.paste(qr_image, (225, 450))

    # # Draw text for amount and merchant name
    # draw = ImageDraw.Draw(template)
    # font = ImageFont.truetype("arial.ttf", 50)  # Specify the font file and size

    # text_amount = f"Amount: {amount} INR"
    # text_merchant = f"Merchant: {merchant_name}"

    # w_amount, h_amount = draw.textsize(text_amount, font=font)
    # w_merchant, h_merchant = draw.textsize(text_merchant, font=font)

    # width, height = template.size

    # position_amount = ((width - w_amount) // 2, height - 1145)  # Adjust the height as needed
    # position_merchant = ((width - w_merchant) // 2, height - 1205)  # Adjust the height as needed

    # draw.text(position_amount, text_amount, (0, 0, 0), font=font)
    # draw.text(position_merchant, text_merchant, (0, 0, 0), font=font)

    # # Save the result
    # final_image_path = os.path.join('static', 'qr_images', f'final_{image_name}')
    # template.save(final_image_path)

    return qr_image


iface = gr.Interface(
    fn=generate_upi_qr,
    inputs=["text", "text", "text"],
    outputs="image",
    title="Generate UPI QR Code",
    description="Generate a UPI QR code image."
)

if __name__ == "__main__":
    iface.launch()