import os
from diffusers import FluxPipeline # type: ignore
import gradio as gr # type: ignore
from huggingface_hub import login, InferenceClient

token = os.getenv("HF_TOKEN")

client = InferenceClient(
	provider="together",
	api_key=token,
    model="black-forest-labs/FLUX.1-dev")

def generate_image(prompt):
    image = client.text_to_image(prompt)
    return image

gradio_app = gr.Interface(
    fn=generate_image,
    inputs=gr.Textbox(label="Entrez une description"),
    outputs=gr.Image(label="Image générée"),
    title="Générateur d'images IA",
    description="Entrez une description et générez une image correspondante."
)

if __name__ == "__main__":
    gradio_app.launch()