Update app.py

app.py

@@ -71,7 +71,7 @@ async def upload_image(image_file:UploadFile):
     image_bytes = await image_file.read()
     uploaded_image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
     
-    return JSONResponse(content={"message": "Logo uploaded successfully"})
+    return JSONResponse(content={"message": "Image uploaded successfully"})
     
 @app.get("/get_image")
 async def get_image(response_class=StreamingResponse):
@@ -165,7 +165,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 created_image is not None:
+    if uploaded_image is not None and uploaded_logo is not None and generated_image is not None:
         # Create LOGO
         print("Drawing")
         #logo_width, logo_height = logo_image.size