Update app.py

app.py

@@ -31,16 +31,20 @@ def home():
         img = Image.new("RGB", (256, 256), color=(255, 255, 255))
         draw = ImageDraw.Draw(img)
 
-        # Use system font installed in Docker
-        font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 20)
-        w, h = draw.textsize(text, font=font)
-        draw.text(((256-w)/2, (256-h)/2), text, fill="black", font=font)
-
-        # Convert image to base64 to embed in HTML
-        img_byte_arr = io.BytesIO()
-        img.save(img_byte_arr, format="PNG")
-        img_byte_arr.seek(0)
-        img_b64 = base64.b64encode(img_byte_arr.read()).decode()
+        # Safe font handling
+        try:
+            font = ImageFont.load_default()
+        except:
+            font = None
+
+        # Draw text
+        draw.text((10, 120), text, fill="black", font=font)
+
+        # Convert image to base64
+        img_bytes = io.BytesIO()
+        img.save(img_bytes, format="PNG")
+        img_bytes.seek(0)
+        img_b64 = base64.b64encode(img_bytes.read()).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)