Update app.py (#1)

- Update app.py (df845522f44990e23aa5938eb5b05c08752338b7)


Co-authored-by: Keertan Balaji <keertan2610@users.noreply.huggingface.co>

app.py

@@ -5,3 +5,24 @@ app = FastAPI()
 @app.get("/")
 def greet_json():
     return {"Hello": "World!"}
+
+
+@app.post("/generate-image")
+async def generate_image(text: str = Form(...)):
+    # Create a blank image
+    image = Image.new('RGB', (512, 256), color=(73, 109, 137))
+    draw = ImageDraw.Draw(image)
+
+    # Use a default font
+    try:
+        font = ImageFont.truetype("arial.ttf", 30)
+    except:
+        font = ImageFont.load_default()
+
+    draw.text((10, 100), text, fill=(255, 255, 255), font=font)
+
+    # Convert image to bytes
+    buf = io.BytesIO()
+    image.save(buf, format="PNG")
+    buf.seek(0)
+    return StreamingResponse(buf, media_type="image/png")