Update app.py

app.py

@@ -1,8 +1,17 @@
-import fal_client
+from huggingface_hub import InferenceClient
 from PIL import Image
 import gradio as gr
-from io import BytesIO
-import base64
+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, "
@@ -19,32 +28,25 @@ def generate_image(dress_type, fabric_type, color_type, design):
     full_prompt = f"{base_description} {trigger_word}"
     print("Generating image with:", full_prompt)
 
-    # Call the FAL API with subscribe
-    result = fal_client.subscribe(
-        "fal-ai/fast-sdxl",
-        arguments={"prompt": full_prompt}
+    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
 
-    # Decode the returned image
-    if result and "image" in result:
-        image_data = result["image"].split(",")[1]
-        image = Image.open(BytesIO(base64.b64decode(image_data)))
-        return image
-    else:
-        raise RuntimeError("Image generation failed or response was invalid.")
-
-# Gradio interface
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Textbox(label="Dress Type", placeholder="e.g., T-shirt"),
-        gr.Textbox(label="Fabric Type", placeholder="e.g., soft cotton"),
-        gr.Textbox(label="Color Type", placeholder="e.g., emerald-green"),
-        gr.Textbox(label="Design", placeholder="e.g., fire design"),
+        gr.Textbox(label="Dress Type"),
+        gr.Textbox(label="Fabric Type"),
+        gr.Textbox(label="Color Type"),
+        gr.Textbox(label="Design"),
     ],
     outputs="image",
     title="TShirt Design XL Image Generator",
-    description="Powered by Redmond.AI — Generate stunning T-shirt designs from simple attributes using FAL-AI.",
+    description="Powered by Redmond.AI — Generate stunning T-shirt designs from simple attributes.",
 )
 
 print("Launching Gradio interface...")