import gradio as gr
from PIL import Image
import numpy as np

def generate_image(prompt: str) -> Image.Image:
    """
    Generate an image based on the given text prompt.
    This is a placeholder function. Replace it with your actual image generation logic.
    
    :param prompt: A text prompt describing the image to generate.
    :return: A PIL Image object.
    """
    # Placeholder: Create a blank image with a random color
    width, height = 256, 256
    random_color = np.random.randint(0, 256, 3)
    image = Image.new("RGB", (width, height), tuple(random_color))
    return image

def main():
    # Define the Gradio interface
    iface = gr.Interface(
        fn=generate_image,  # The function to call for generating images
        inputs=gr.inputs.Textbox(lines=2, placeholder="Enter a description of the image"),  # Input component
        outputs=gr.outputs.Image(type="pil"),  # Output component
        title="Image Generator",
        description="Generate an image from a text prompt."
    )
    
    # Launch the interface
    iface.launch()

if __name__ == "__main__":
    main()