Update app.py

app.py

@@ -11,14 +11,21 @@ def img2text(img):
     return text
 
 # text2story
-def text2story(text):
-    story_text = ""   # to be completed
-    return story_text
+def text2story(scenario):
+    generator = pipeline("text-generation", model="gpt2")
+    story = generator(
+        f"Create a children's story based on: {scenario}",
+        max_length=300,
+        num_return_sequences=1
+    )
+    return story[0]["generated_text"]
 
 # text2audio
 def text2audio(story_text):
-    audio_data = ""     # to be completed
-    return audio_data
+    tts = gTTS(text=story_text, lang="en", slow=False)
+    audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+    tts.save(audio_file.name)
+    return audio_file.name
 
 
 # main 
@@ -42,13 +49,15 @@ if uploaded_file is not None:
 
     #Stage 2: Text to Story
     st.text('Generating a story...')
-    #story = text2story(scenario)
-    #st.write(story)
+    story = text2story(scenario)
+    st.subheader("Generated Story")
+    st.success(story)
 
     #Stage 3: Story to Audio data
-    #st.text('Generating audio data...')
-    #audio_data =text2audio(story)
-
+    st.text('Generating audio data...')
+    audio_data =text2audio(story)
+    st.subheader("Audio Story")
+    st.audio(audio_path, format="audio/mp3")
     # Play button
     if st.button("Play Audio"):
         #st.audio(audio_data['audio'],