added upload image

app.py

@@ -12,6 +12,8 @@ from dotenv import load_dotenv
 HF_TOKEN = os.getenv('HF_TOKEN')
 app = FastAPI()
 
+init_image = None
+
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -40,22 +42,31 @@ async def generate_image(prompt: str = Query(..., description="Text prompt for i
     image.save(image_data, format="PNG")
     image_data.seek(0)
     return StreamingResponse(image_data, media_type="image/png")
-    
-@app.get("/generate_ad",response_class=StreamingResponse)
-async def generate_ad(image_file:UploadFile, hex_code:str, prompt: str = Query(..., description="Text prompt for image generation")):
-    
+
+@app.post("/uploadImage", response_class=StreamingResponse)
+async def upload_image(image_file:UploadFile):
+    global init_image
     image_bytes = await image_file.read()
     init_image = Image.open(io.BytesIO(image_bytes)).convert("RGBA")
-    #init_image.thumbnail((768, 768))
-    #Generate the image using the text-to-image model
-    generator = torch.Generator(device=device).manual_seed(1024)
-    ad_prompt = f"""Your system prompt is this: {prompt} Consider your system prompt first. 
-                    Then from init image create me a new image that will attract customers to put in a (ad template) 
-                    Also, use this hex code {hex_code} in the image"""
+    init_image.thumbnail((768, 768))
+    return {"message": "You have uploaded the image!"}
+
     
-    image = pipe(prompt=ad_prompt, image=init_image, strength=0.75, guidance_scale=7.5, generator=generator).images[0]
+@app.get("/generate_ad",response_class=StreamingResponse)
+async def generate_ad(hex_code:str, prompt: str = Query(..., description="Text prompt for image generation")):
     
-    image_data = io.BytesIO()
-    image.save(image_data, format="PNG")
-    image_data.seek(0)
-    return StreamingResponse(image_data, media_type="image/png")
+    if init_image is not None:
+        #Generate the image using the text-to-image model
+        generator = torch.Generator(device=device).manual_seed(1024)
+        ad_prompt = f"""Your system prompt is this: {prompt} Consider your system prompt first. 
+                        Then from the initial image create a new image that will attract customers to put in an (ad template) 
+                        Also, use this hex code {hex_code} in the image"""
+        
+        image = pipe(prompt=ad_prompt, image=init_image, strength=0.75, guidance_scale=7.5, generator=generator).images[0]
+        
+        image_data = io.BytesIO()
+        image.save(image_data, format="PNG")
+        image_data.seek(0)
+        return StreamingResponse(image_data, media_type="image/png")
+    else
+        return {"message": "You have not uploaded the image!"}