import streamlit as st
from transformers import pipeline

st.set_page_config(page_title="Generate Your Tweet and Toxicity Analysis")

st.header("Please input your first word of a Tweet :performing_arts:")
input = st.text_input("In put your first word...")

if input is not None:
    print(Your word is welcome :slightly_smiling_face:)

    #Stage 1: Image to Text
    st.text('Processing img2text...')
    scenario = img2text(uploaded_file.name)
    st.write(scenario)

    #Stage 2: Text to Story
    st.text('Generating a story...')
    story = text2story(scenario)
    st.write(story)

    #Stage 3: Story to Audio data
    st.text('Generating audio data...')
    audio_data =text2audio(story)

    # Play button
    if st.button("Play Audio"):
        st.audio(audio_data['audio'],
                    format="audio/wav",
                    start_time=0,
                    sample_rate = audio_data['sampling_rate'])