Update app.py

app.py

@@ -2,7 +2,7 @@
 import streamlit as st
 from transformers import pipeline
 from gtts import gTTS
-import io
+import os
 
 # function part
 # img2text
@@ -14,22 +14,20 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    story_pipeline = pipeline("text-generation", model="perplexity-ai/r1-1776", trust_remote_code=True)
-    result = story_pipeline(text, max_length=200)
-    story_text = result[0]['generated_text']
+    story_text = "pipeline("text-generation", model="perplexity-ai/r1-1776", trust_remote_code=True)"   # to be completed
     return story_text
 
 # text2audio
 def text2audio(story_text):
-    tts = gTTS(text=story_text, lang='en')
-    audio_file = io.BytesIO()
-    tts.write_to_fp(audio_file)
-    audio_file.seek(0)
+    # Convert text to audio using gTTS
+    tts = gTTS(story_text, lang="en")
+    audio_file = "story_audio.wav"
+    tts.save(audio_file)
     return audio_file
 
 # main part
 st.set_page_config(page_title="Your Image to Audio Story",
-                   page_icon="🦜")  # prepare configuration
+                   page_icon="🦜") # prepare configuration
 st.header("Turn Your Image to Audio Story")
 
 # Upload image
@@ -45,19 +43,19 @@ if uploaded_file is not None:
     st.image(uploaded_file, caption="Uploaded Image",
              use_column_width=True)
 
-    # Stage 1: Image to Text
+    #Stage 1: Image to Text
     st.text('Processing img2text...')
     scenario = img2text(uploaded_file.name)
     st.write(scenario)
 
-    # Stage 2: Text to Story
+    #Stage 2: Text to Story
     st.text('Generating a story...')
     story = text2story(scenario)
     st.write(story)
 
-    # Stage 3: Story to Audio data
+    #Stage 3: Story to Audio data
     st.text('Generating audio data...')
-    audio_data = text2audio(story)
+    audio_data =text2audio(story)
 
     # Play button
     if st.button("Play Audio"):