Spaces:
Runtime error
Runtime error
File size: 1,623 Bytes
5c7be38 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
import streamlit as st
import time
import requests
from utils.audio import *
from utils.preprocessing import *
from PIL import Image
play, names = preprocess_play("assets/FathersDay.csv")
colors = ['violet', 'orange', 'blue', 'red', 'green']
name_to_color = {}
i = 0
for name in names:
name_to_color[name] = colors[i]
i += 1
play_title = "Father's Day"
play_author = "D. M. Larson"
st.markdown(f"<h1 style='text-align: center; color: white;'>{play_title}</h1>", unsafe_allow_html=True)
st.markdown(f"<h5 style='text-align: center; color: white;'>{play_author}</h5>", unsafe_allow_html=True)
initial_img = Image.open('images/meeting.png')
ai_img = st.image(initial_img)
character_1 = st.markdown('')
quote_1 = st.markdown('')
character_2 = st.markdown('')
quote_2 = st.markdown('')
character_3 = st.markdown('')
quote_3 = st.markdown('')
url = 'http://127.0.0.1:5000/getDialog'
dialogues = [[character_1, quote_1], [character_2, quote_2], [character_3, quote_3]]
time.sleep(1)
counter = 0
for i in range(1, len(play)):
rem = (i-1) % 3
item = play[i]
curr_color = name_to_color[item[0]]
time.sleep(1)
dialogues[rem][0].markdown(':' + curr_color + '[' + item[0] +']')
dataToSend = {
"play_name": play_title,
"line_no": counter
}
audio = requests.post(url, data = dataToSend)
counter += 1
time.sleep(1)
dialogues[rem][1].markdown(item[1])
autoplay_audio(audio.content)
time.sleep(get_audio_duration(audio.content))
if rem == 2:
for dialog in dialogues:
dialog[0].markdown('')
dialog[1].markdown('')
|