Spaces:
Runtime error
Runtime error
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() | |