Update app.py

app.py

@@ -1,53 +1,52 @@
 import streamlit as st
 from transformers import pipeline
-from responsivevoice import ResponsiveVoice
-import random
 
-# Pre-trained text-to-speech model
-tts_model = ResponsiveVoice()
+# Load Hugging Face pipelines
+text_generator = pipeline("text-generation")
+text_to_speech = pipeline("text-to-speech")
+text_to_image = pipeline("text2image")
 
-# Choose a pre-trained language model for story generation
-model_name = "gpt2"
+# Streamlit app
+st.title("Children's Storytelling App")
 
-# Initialize pipeline for text generation
-generator = pipeline("text-generation", model=model_name)
-
-# Define supported languages
-languages = ["en", "fr", "es", "de", "it"]
-
-def translate_text(text, target_language):
-    translator = Translator()
-    translated_text = translator.translate(text, dest=target_language).text
-    return translated_text
-
-def generate_and_narrate_story(prompt, language):
-    # Generate story based on prompt
-    story = generator(prompt, max_length=1024)[0]["generated_text"]
-
-    # Translate story to chosen language
-    if language != "en":
-        translated_story = translate_text(story, language)
-    else:
-        translated_story = story
-
-    # Speak the story using the text-to-speech model
-    tts_model.speak(translated_story, language)
-
-# Streamlit app initialization
-st.title("AI Storytelling App")
-
-# Prompt input
-prompt = st.text_input("Start your story with...")
-
-# Language selector
-language = st.selectbox("Choose narration language:", languages)
+# Input prompt for story
+story_prompt = st.text_area("Write the beginning of your story:")
 
 # Generate story button
-if st.button("Generate and Narrate Story"):
-    with st.spinner("Generating and narrating your story..."):
-        generate_and_narrate_story(prompt, language)
-
-# Disclaimer
-st.write("* This app is still under development and may not always generate accurate or coherent results.")
-st.write("* Please be mindful of the content generated by the AI model.")
+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.")