Create app.py

app.py

@@ -0,0 +1,92 @@
+from fastapi import FastAPI
+from fastapi.responses import StreamingResponse
+from pydantic import BaseModel
+from PIL import Image, ImageDraw, ImageFont
+import requests
+from io import BytesIO
+import os
+
+app = FastAPI()
+
+class WelcomeRequest(BaseModel):
+    username: str
+    avatar_url: str | None = None
+
+def load_font(size: int = 60) -> ImageFont.FreeTypeFont:
+    font_paths = [
+        "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf",
+        "/usr/share/fonts/truetype/liberation/LiberationSans-Bold.ttf",
+        "/usr/share/fonts/truetype/freefont/FreeSans.ttf",
+        "/System/Library/Fonts/Arial.ttf",  # macOS
+        "C:/Windows/Fonts/arial.ttf",       # Windows
+        "arial.ttf"
+    ]
+
+    for path in font_paths:
+        try:
+            if os.path.exists(path):
+                return ImageFont.truetype(path, size)
+        except Exception as e:
+            print(f"Failed to load font {path}: {e}")
+            continue
+
+    # If no TrueType font found, try to use default but warn
+    print(f"Warning: Using default font, size may not scale properly")
+    try:
+        return ImageFont.load_default(size)
+    except:
+        return ImageFont.load_default()
+
+@app.post("/generate-welcome")
+async def generate_welcome(data: WelcomeRequest):
+    try:
+        base_image = Image.open("ChatGPT Image May 27, 2025, 05_24_11 AM.png").convert("RGBA")
+    except FileNotFoundError:
+        return {"error": "base_image.png not found"}
+
+    width, height = base_image.size
+    draw = ImageDraw.Draw(base_image)
+
+    # === Big, centered username font ===
+    username_font_size = 50
+    font = load_font(username_font_size)
+
+    # Debug: Check if font loaded properly
+    print(f"Font loaded: {type(font)}, size: {username_font_size}")
+
+    text_color = (148, 255, 255)
+
+    # === Position elements in lower center area ===
+    username_with_at = f"@{data.username}"  # Add @ symbol
+    text_width = draw.textlength(username_with_at, font=font)
+    username_x = (width - text_width) // 2
+
+    # Position username in lower portion of image
+    username_y = height - 120  # 120px from bottom
+
+    # === Optional avatar ===
+    avatar_size = 120
+    if data.avatar_url:
+        try:
+            response = requests.get(data.avatar_url)
+            avatar = Image.open(BytesIO(response.content)).convert("RGBA").resize((avatar_size, avatar_size))
+
+            mask = Image.new("L", (avatar_size, avatar_size), 0)
+            draw_mask = ImageDraw.Draw(mask)
+            draw_mask.ellipse((0, 0, avatar_size, avatar_size), fill=255)
+
+            # Position avatar above the username in lower center
+            avatar_x = width // 2 - avatar_size // 2
+            avatar_y = username_y - avatar_size - 30  # 30px gap above username
+
+            base_image.paste(avatar, (avatar_x, avatar_y), mask)
+        except Exception as e:
+            print("Error loading avatar:", e)
+
+    # === Draw large centered username with @ symbol in lower area ===
+    draw.text((username_x, username_y), username_with_at, fill=text_color, font=font)
+
+    output = BytesIO()
+    base_image.save(output, format='PNG')
+    output.seek(0)
+    return StreamingResponse(output, media_type="image/png")