Update app.py

app.py

@@ -4,7 +4,7 @@ from PIL import Image, ImageDraw, ImageFont
 import time
 import os
 
-def generate_upi_qr(upi_code, amount, merchant_name=None):
+def generate_upi_qr_sticker(upi_code, amount, merchant_name=None):
     if not merchant_name:
         merchant_name = upi_code.split('@')[0]
 
@@ -16,50 +16,45 @@ def generate_upi_qr(upi_code, amount, merchant_name=None):
     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"
+    # Open your template
+    template = Image.open('temp.png').convert("RGB")
 
-    # # 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))
 
-    # # 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))
 
-    # # 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
 
-    # # 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}"
 
-    # 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)
 
-    # w_amount, h_amount = draw.textsize(text_amount, font=font)
-    # w_merchant, h_merchant = draw.textsize(text_merchant, font=font)
+    width, height = template.size
 
-    # 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
 
-    # 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)
 
-    # 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
+    # Save the result
+    final_image_path = os.path.join('static', 'qr_images', f'final_{upi_code}.png')
+    template.save(final_image_path)
 
+    return final_image_path
 
 iface = gr.Interface(
-    fn=generate_upi_qr,
+    fn=generate_upi_qr_sticker,
     inputs=["text", "text", "text"],
     outputs="image",
-    title="Generate UPI QR Code",
-    description="Generate a UPI QR code image."
+    title="Generate UPI QR Code Sticker",
+    description="Generate a UPI QR code sticker image."
 )
 
 if __name__ == "__main__":