Update app.py

app.py

@@ -166,7 +166,7 @@ async def generate_new_img(hex_code: str, prompt: str = Query(..., description="
 @app.get("/create_ad_template",response_class=StreamingResponse)
 async def create_ad_template(punchline: str, punchline_color:str, button_text:str,button_color:str):
     
-    if uploaded_image is not None and uploaded_logo is not None and generated_image is not None:
+    if uploaded_logo is not None and generated_image is not None:
         # Create LOGO
         print("Drawing")
         #logo_width, logo_height = logo_image.size
@@ -219,11 +219,9 @@ async def create_ad_template(punchline: str, punchline_color:str, button_text:st
         
     else:
         color = (255, 0, 0)  # This represents red
-        generated_image = Image.new("RGB", (image_width, image_height), color)
-        color = (0, 255, 0)  # This represents red
-        uploaded_image_image = Image.new("RGB", (image_width, image_height), color)
+        generated_image = Image.new("RGB", (350, 350), color)
         color = (0, 0, 255)  # This represents red
-        uploaded_logo = Image.new("RGB", (image_width, image_height), color)
+        uploaded_logo = Image.new("RGB", (100, 100), color)
         # Create LOGO
         print("Drawing newwww imagess........")
         #logo_width, logo_height = logo_image.size