File size: 1,711 Bytes
e9d63b2 d6e07a2 e9d63b2 d6e07a2 e9d63b2 d6e07a2 e9d63b2 d6e07a2 e9d63b2 d6e07a2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
import streamlit as st
from transformers import pipeline
# Load Hugging Face pipelines
text_generator = pipeline("text-generation")
text_to_speech = pipeline("text-to-speech")
text_to_image = pipeline("text2image")
# Streamlit app
st.title("Children's Storytelling App")
# Input prompt for story
story_prompt = st.text_area("Write the beginning of your story:")
# Generate story button
if st.button("Generate Story"):
if story_prompt:
# Generate text
generated_story = text_generator(story_prompt, max_length=100, num_return_sequences=1)
st.write("Here's your story:")
st.write(generated_story[0]["generated_text"])
else:
st.warning("Please enter a story prompt.")
# Text-to-speech button
if st.button("Text to Speech"):
if story_prompt:
# Convert text to speech
st.audio(text_to_speech(story_prompt)[0]["audio"], format="audio/wav")
else:
st.warning("Please enter a story prompt.")
# Text-to-text button
if st.button("Text to Text"):
if story_prompt:
# Convert text to a different text
transformed_text = text_to_image(story_prompt)[0]["generated_text"]
st.write("Transformed text:")
st.write(transformed_text)
else:
st.warning("Please enter a story prompt.")
# Text-to-image
st.sidebar.title("Text to Image")
image_prompt = st.sidebar.text_area("Enter text for image:")
if st.sidebar.button("Generate Image"):
if image_prompt:
# Convert text to image
image = text_to_image(image_prompt)[0]["image"]
st.sidebar.image(image, caption="Generated Image", use_column_width=True)
else:
st.sidebar.warning("Please enter text for the image.")
|