Update app.py

app.py

@@ -1,13 +1,29 @@
 import os
 from huggingface_hub import InferenceClient
+import gradio as gr
 
+# Hugging Face Inference Client
 client = InferenceClient(
     provider="nebius",
-    api_key=os.environ["HF_TOKEN"],
+    api_key=os.environ["HF_TOKEN"],  # HF_TOKEN Secrets এ দেওয়া থাকবে
 )
 
-# output is a PIL.Image object
-image = client.text_to_image(
-    "Astronaut riding a horse",
-    model="black-forest-labs/FLUX.1-dev",
-)
+# Function যা Prompt থেকে ছবি জেনারেট করবে
+def generate_image(prompt):
+    image = client.text_to_image(
+        prompt,
+        model="black-forest-labs/FLUX.1-dev",
+    )
+    return image
+
+# Gradio UI
+demo = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(label="Enter your prompt"),
+    outputs=gr.Image(type="pil", label="Generated Image"),
+    title="FLUX.1 Text-to-Image Generator",
+    description="Type a prompt and generate an image using FLUX.1 model."
+)
+
+# Launch the app
+demo.launch()