Update app.py

app.py

@@ -3,7 +3,7 @@ from PIL import Image, ImageDraw, ImageFont
 import torch
 from diffusers import StableDiffusionImg2ImgPipeline, DiffusionPipeline
 from starlette.responses import StreamingResponse
-from fastapi.responses import FileResponse
+from fastapi.responses import FileResponse, JSONResponse
 import io
 from fastapi.middleware.cors import CORSMiddleware
 import os
@@ -43,13 +43,13 @@ async def generate_image(prompt: str = Query(..., description="Text prompt for i
     image_data.seek(0)
     return StreamingResponse(image_data, media_type="image/png")
 
-@app.post("/uploadImage", response_class=StreamingResponse)
+@app.post("/uploadImage", response_class=JSONResponse)
 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))
-    return {"message": "You have uploaded the image!"}
+    return JSONResponse(content={"message": "File uploaded successfully"})
 
     
 @app.get("/generate_ad",response_class=StreamingResponse)
@@ -69,4 +69,4 @@ async def generate_ad(hex_code:str, prompt: str = Query(..., description="Text p
         image_data.seek(0)
         return StreamingResponse(image_data, media_type="image/png")
     else:
-        return {"message": "You have not uploaded the image!"}
+        return StreamingResponse("You have not uploaded the image!", media_type="text/plain")