app.py

from huggingface_hub import InferenceClient
from PIL import Image
import gradio as gr
import os

# Load token from environment
token = os.environ["HF_TOKEN"]

# Create the client
client = InferenceClient(
    model="artificialguybr/TshirtDesignRedmond-V2",
    provider="fal-ai",
    token=token,
)

# Trigger word for model
trigger_word = "T shirt design, TshirtDesignAF, "

def generate_image(prompt):
    full_prompt = f"{prompt} {trigger_word}"
    print("Generating image with:", full_prompt)

    # Call the HF inference client directly with correct parameters
    image = client.text_to_image(
        prompt=full_prompt,
        negative_prompt="(worst quality, low quality, lowres, bad photo, ...)",
        num_inference_steps=30,
        guidance_scale=7.5,
    )
    return image

# Gradio interface
iface = gr.Interface(
    fn=generate_image,
    inputs=gr.Textbox(lines=2, placeholder="Type your prompt here..."),
    outputs="image",
    title="TShirt Design XL Image Generator",
    description="Powered by Redmond.AI — Generates T-shirt designs from prompts. Trigger tags: 'T shirt design, TshirtDesignAF'.",
    examples=[["Cute Panda"], ["Skull"]]
)

print("Launching Gradio interface...")
iface.launch()