Upload folder using huggingface_hub

app.py

@@ -43,7 +43,7 @@ def send_generation_request(
 
     return response
 
-def generate_image(prompt, negative_prompt, seed, aspect_ratio, output_format, model):
+def generate_image(brand_name, type, coupon_info, address, prompt, negative_prompt, seed, aspect_ratio, output_format, model):
     """Generates an image using the SD3 API."""
 
     host = os.environ["STABILITY_HOST"]
@@ -72,6 +72,12 @@ def generate_image(prompt, negative_prompt, seed, aspect_ratio, output_format, m
     # Convert image to PIL format
     image = Image.open(BytesIO(output_image))
 
+    # TODO: 新增 品牌名称
+    # TODO: 新增 类别
+    # TODO: 新增 折扣信息
+    # TODO: 新增 实体店地址
+    # TODO: 新增 编号控制
+    # TODO: 新增 智能客服联系人QR Code
     return image
 
 """
@@ -85,6 +91,12 @@ Cost Rule
 interface = gr.Interface(
     fn=generate_image,
     inputs=[
+        gr.Textbox(label="品牌名称", placeholder="样例: AI原优舍"),
+        gr.Dropdown(label="类别", choices=["服饰", "饮食", "健身", "按摩", "公益"], value="服饰"),
+        gr.Textbox(label="折扣信息", placeholder="样例: 凭券全场任意两件正价八折"),
+        gr.Textbox(label="实体店地址", placeholder="样例: 广州市麦栏街20号野隐大楼3层"),
+        # TODO: 新增 编号控制按钮
+        # TODO: 新增 智能客服联系人QR Code
         gr.Textbox(label="广告整体正向描述", placeholder="This dreamlike digital art captures a vibrant, kaleidoscopic bird in a lush rainforest"),
         gr.Textbox(label="广告整体负向描述", placeholder="Optional"),
         gr.Number(label="算法随机种子", value=0),