Update app.py

app.py

@@ -85,8 +85,8 @@ def add_text_to_image(image, product_name, tagline, cta_text, font_size=50):
     
     return image
 
-# Gradio Interface Function
-def generate_ad(brand_title, tagline, cta, brand_logo, product_image, custom_prompt):
+# Main function to generate advertisement
+def generate_ad(brand_title, tagline, cta, brand_logo=None, product_image=None, custom_prompt=None):
     """
     Generate an advertisement image with the provided details.
     
@@ -94,9 +94,9 @@ def generate_ad(brand_title, tagline, cta, brand_logo, product_image, custom_pro
         brand_title (str): Brand title for the advertisement.
         tagline (str): Tagline for the advertisement.
         cta (str): Call to action text.
-        brand_logo (file): Brand logo image (optional).
-        product_image (file): Product image (optional).
-        custom_prompt (str): Custom prompt for image generation.
+        brand_logo (file, optional): Brand logo image (optional).
+        product_image (file, optional): Product image (optional).
+        custom_prompt (str, optional): Custom prompt for image generation.
 
     Returns:
         PIL Image: Final advertisement image.
@@ -124,7 +124,7 @@ def generate_ad(brand_title, tagline, cta, brand_logo, product_image, custom_pro
     
     return final_image
 
-# Create Gradio Interface
+# Gradio Interface
 iface = gr.Interface(
     fn=generate_ad,
     inputs=[
@@ -138,5 +138,5 @@ iface = gr.Interface(
     outputs=gr.Image(label="Generated Advertisement"),
 )
 
-# Launch Gradio Interface
+# Launch the Gradio Interface
 iface.launch()