app.py

from __future__ import annotations
from huggingface_hub import whoami
import gradio as gr
from diffusers import DiffusionPipeline

def generate_image_and_prompt(prompt: str):
    # Create an "improved" prompt (simple example)
    improved_prompt = f"Improved: {prompt}"

    # Load the DiffusionPipeline
    pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
    pipe.load_lora_weights("EvanZhouDev/open-genmoji")

    # Generate an image from the prompt
    image = pipe(prompt).images[0]

    # Return both the improved prompt and the image
    return improved_prompt, image

# Create the Gradio interface
with gr.Blocks() as demo:
    gr.LoginButton()
    # Input Textbox for the user prompt
    input_prompt = gr.Textbox(label="Input Prompt", placeholder="Enter your prompt here")
    
    # Output Textbox for the generated improved prompt
    improved_prompt_output = gr.Textbox(label="Improved Prompt")
    
    # Output Image for the generated image
    generated_image_output = gr.Image(label="Generated Image")
    
    # Button to trigger the generation
    generate_btn = gr.Button("Generate")
    
    # Set up the event listener for the button click
    generate_btn.click(
        fn=generate_image_and_prompt,
        inputs=input_prompt,
        outputs=[improved_prompt_output, generated_image_output]
    )

# Launch the Gradio app
if __name__ == "__main__":
    demo.launch(show_error=True)