Spaces:
Build error
Build error
| import logging | |
| import streamlit as st | |
| from src import file_readers, predict, output, config | |
| logging.basicConfig(filename='app.log', | |
| filemode='w', | |
| format='%(name)s - %(levelname)s - %(message)s', | |
| level=logging.INFO, | |
| force=True) | |
| st.title('Audiobook Generation Tool') | |
| text_file = open(config.INSTRUCTIONS, "r") | |
| readme_text = text_file.read() | |
| text_file.close() | |
| st.markdown(readme_text) | |
| st.header('1. Upload your document') | |
| uploaded_file = st.file_uploader( | |
| label="File types accepted: epub, txt, pdf)", | |
| type=['epub', 'txt', 'pdf']) | |
| model = predict.load_model() | |
| st.header('2. Please select voice') | |
| speaker = st.radio('Available voices:', config.SPEAKER_LIST.keys(), horizontal=True) | |
| audio_path = config.resource_path / f'speaker_{config.SPEAKER_LIST.get(speaker)}.wav' | |
| audio_file = open(audio_path, 'rb') | |
| audio_bytes = audio_file.read() | |
| st.audio(audio_bytes, format='audio/ogg') | |
| st.header('3. Run the app to generate audio') | |
| if st.button('Click to run!'): | |
| file_ext = uploaded_file.type | |
| file_title = uploaded_file.name | |
| if file_ext == 'application/epub+zip': | |
| text, file_title = file_readers.read_epub(uploaded_file) | |
| elif file_ext == 'text/plain': | |
| file = uploaded_file.read() | |
| text = file_readers.preprocess_text(file) | |
| elif file_ext == 'application/pdf': | |
| text = file_readers.read_pdf(uploaded_file) | |
| else: | |
| st.warning('Invalid file type', icon="⚠️") | |
| st.success('Reading file complete!') | |
| with st.spinner('Generating audio...'): | |
| predict.generate_audio(text, file_title, model, config.SPEAKER_LIST.get(speaker)) | |
| st.success('Audio generation complete!') | |
| with st.spinner('Building zip file...'): | |
| zip_file = output.assemble_zip(file_title) | |
| title_name = f'{file_title}.zip' | |
| st.success('Zip file prepared!') | |
| with open(zip_file, "rb") as fp: | |
| btn = st.download_button( | |
| label="Download Audiobook", | |
| data=fp, | |
| file_name=title_name, | |
| mime="application/zip" | |
| ) | |