Upload folder using huggingface_hub

app.py

@@ -6,6 +6,7 @@ from PIL import Image
 import requests
 import time
 import getpass
+from PIL import ImageDraw, ImageFont
 
 # Get the API key
 STABILITY_API_KEY = os.environ.get('STABILITY_API_KEY')  # Try to get from environment variable
@@ -72,7 +73,11 @@ def generate_image(brand_name, type, coupon_info, address, prompt, negative_prom
     # Convert image to PIL format
     image = Image.open(BytesIO(output_image))
 
-    # TODO: 新增 品牌名称
+    # Add brand info to the upper left corner
+    draw = ImageDraw.Draw(image)
+    font = ImageFont.truetype("./arial-font/arial.ttf", 20)  # You can change the font and size
+    draw.text((10, 10), brand_name, fill=(255, 255, 255), font=font)  # Adjust position and color as needed
+
     # TODO: 新增 类别
     # TODO: 新增 折扣信息
     # TODO: 新增 实体店地址