import streamlit as st
import requests
from PIL import Image
from io import BytesIO

API_KEY = "1143a102dbe21628248d4bb992b391a49dc058c584181ea72e17c2ccd49be9ca69ccf4a2b97fc82c89ff1029578abbea"
API_URL = "https://clipdrop-api.co/text-to-image/v1"

def generate_image(prompt):
    headers = {"x-api-key": API_KEY}
    files = {"prompt": (None, prompt, "text/plain")}

    try:
        response = requests.post(API_URL, files=files, headers=headers)
        response.raise_for_status()

        # Get the generated image
        image = Image.open(BytesIO(response.content))

        return image
    except requests.exceptions.RequestException as e:
        st.error(f"Error occurred during image generation: {str(e)}")
        return None

def main():
    st.title("Text-to-Image Generator")

    # Text prompt input
    prompt = st.text_input("Enter a text prompt")

    if prompt:
        # Generate image when the "Generate Image" button is clicked
        if st.button("Generate Image"):
            st.write("Generating image...")
            image = generate_image(prompt)

            if image:
                # Display the generated image
                st.image(image, caption="Generated Image", use_column_width=True)

if __name__ == "__main__":
    main()