Create app.py

app.py

@@ -0,0 +1,59 @@
+import streamlit as st
+from transformers import pipeline
+from PIL import Image
+import time
+
+def generate_caption(image):
+    image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+    caption = image_to_text(image)[0]["generated_text"]
+    return caption
+
+def generate_story(caption):
+    pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
+    story = pipe(caption)[0]['generated_text']
+    return story
+
+def generate_audio(story):
+    pipe = pipeline("text-to-speech", model="facebook/mms-tts-eng")
+    audio = pipe(story)
+    return audio
+
+# Streamlit UI
+
+# Title of the Streamlit app
+st.title("Upload your image for an instant storytelling!")
+
+# Write a description
+st.write("This app is designed for 3-10 year-old kids by allowing them uploading image for fun storytelling entertainment.")
+
+# File uploader for image
+uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
+
+if uploaded_file is not None:
+    # Display the uploaded image
+    image = Image.open(uploaded_file)
+    st.image(image, caption="Uploaded Image", use_container_width=True)
+    
+    # Generate Image Caption
+    image_caption = generate_caption(image)
+
+    # Display results
+    st.subheader("Image Caption:")
+    st.write(f"{image_caption}")
+
+    # Generate Story
+    story_telling = generate_story(image_caption)
+
+    # Display results
+    st.subheader("Story:")
+    st.write(f"{story_telling}")
+
+    # Generate Audio
+    audio = generate_audio(story_telling)
+
+    # Display an audio file with a spinner effect
+    if st.button("Play Audio"):
+        st.audio(audio['audio'], 
+                format="audio/wav", 
+                start_time=0, 
+                sample_rate = audio['sampling_rate'])