Update app.py

app.py

@@ -27,24 +27,24 @@ def home():
     if request.method == "POST":
         text = request.form.get("prompt", "Hello!")
 
-        # Create image
+        # Create blank image
         img = Image.new("RGB", (256, 256), color=(255, 255, 255))
         draw = ImageDraw.Draw(img)
 
-        # Safe font handling
-        try:
-            font = ImageFont.load_default()
-        except:
-            font = None
+        # Use Pillow default font (always works in slim Python)
+        font = ImageFont.load_default()
 
-        # Draw text
-        draw.text((10, 120), text, fill="black", font=font)
+        # Draw text centered
+        w, h = draw.textsize(text, font=font)
+        draw.text(((256 - w) / 2, (256 - h) / 2), text, fill="black", font=font)
 
-        # Convert image to base64
+        # Save image to bytes
         img_bytes = io.BytesIO()
         img.save(img_bytes, format="PNG")
         img_bytes.seek(0)
-        img_b64 = base64.b64encode(img_bytes.read()).decode()
+
+        # Encode to base64 for embedding
+        img_b64 = base64.b64encode(img_bytes.getvalue()).decode()
         image_tag = f"<h3>Generated Image:</h3><img src='data:image/png;base64,{img_b64}' alt='Generated Image'>"
 
     return HTML_TEMPLATE.format(image_tag=image_tag)