Spaces:
Runtime error
Runtime error
| import io | |
| from process_media import MediaProcessor | |
| import streamlit as st | |
| from summarizer import BARTSummarizer | |
| from Utils import fetch_article_text, get_text_from_youtube_url | |
| st.markdown( | |
| """ | |
| <style> | |
| section[data-testid="stSidebar"] div[role="radiogroup"] label { | |
| padding: 0px 0px 20px 20px; | |
| } | |
| section[data-testid="stSidebar"] h2 { | |
| margin: 10px; | |
| } | |
| section.main div[role="radiogroup"] label { | |
| padding: 10px 10px 10px 0px; | |
| } | |
| </style> | |
| """, | |
| unsafe_allow_html=True, | |
| ) | |
| with st.sidebar: | |
| st.header("CHOOSE INPUT TYPE") | |
| input_type = st.radio("", ["Text", "Media"], label_visibility = "hidden") | |
| text_to_summarize = None | |
| if input_type == "Text": | |
| st.header("Summarize from text or URL") | |
| text_type = st.radio("", ["Raw Text", "URL", "Document"], key="text_type", horizontal=True, label_visibility = "hidden") | |
| if text_type == "Raw Text": | |
| text = st.text_area("Enter raw text here", height=240, max_chars=10000, placeholder="Enter a paragraph to summarize") | |
| if text: | |
| text_to_summarize = text | |
| elif text_type == "URL": | |
| url = st.text_input("Enter URL here", placeholder="Enter URL to an article, blog post, etc.") | |
| if url: | |
| article_text = fetch_article_text(url) | |
| if article_text: | |
| st.markdown("#### Text from url:") | |
| st.write(article_text) | |
| text_to_summarize = article_text | |
| else: | |
| ## TODO: Add file upload option | |
| pass | |
| elif input_type == "Media": | |
| st.header("Summarize from file or YouTube URL") | |
| media_type = st.radio("", ["Audio file", "Video file", "Youtube video link"], key="media_type", horizontal=True, label_visibility = "hidden") | |
| if media_type == "Audio file": | |
| audio_file = st.file_uploader("Upload an audio file", type=["mp3", "wav"], label_visibility="visible") | |
| if audio_file is not None: | |
| with st.spinner("Fetching text from audio..."): | |
| # print(audio_file.read()) | |
| wav_bytes = None | |
| media_processor = MediaProcessor() | |
| if audio_file.type == "audio/mpeg": | |
| wav_bytes = media_processor.get_wav_from_audio(audio_file.read()) | |
| else: | |
| wav_bytes = audio_file.read() | |
| text = media_processor.process_audio(wav_bytes) | |
| st.markdown("#### Text from audio:") | |
| st.write(text) | |
| elif media_type == "Video file": | |
| video_file = st.file_uploader("Upload a video file", type=["mp4"], label_visibility="visible") | |
| if video_file is not None: | |
| with st.spinner("Fetching text from video..."): | |
| media_processor = MediaProcessor() | |
| text = media_processor.process_video(video_file.read()) | |
| st.markdown("#### Text from video:") | |
| st.write(text) | |
| else: | |
| youtube_url = st.text_input("Enter YouTube URL here", placeholder="Enter URL to an YouTube video", label_visibility="visible") | |
| if youtube_url: | |
| with st.spinner("Fetching text from video..."): | |
| try: | |
| text_to_summarize = get_text_from_youtube_url(youtube_url) | |
| st.markdown("#### Text from video:") | |
| st.markdown('<div style="height: 300px; overflow: auto; margin-bottom: 20px;">' + text_to_summarize + '</div>', unsafe_allow_html=True) | |
| except: | |
| st.error("Unable to fetch text from video. Please try a different video.") | |
| text_to_summarize = None | |
| if text_to_summarize is not None: | |
| overall_summary = st.button("Overall summary") | |
| auto_chapters_summary = st.button("Auto Chapters summary") | |
| if overall_summary: | |
| with st.spinner("Summarizing..."): | |
| # time.sleep(2) | |
| # st.write(text_to_summarize) | |
| summarizer = BARTSummarizer() | |
| summary = summarizer.chunk_summarize(text_to_summarize) | |
| st.markdown("#### Summary:") | |
| st.write(summary) | |
| elif auto_chapters_summary: | |
| with st.spinner("Summarizing..."): | |
| # time.sleep(2) | |
| # st.write(text_to_summarize) | |
| summarizer = BARTSummarizer() | |
| summary = summarizer.auto_chapters_summarize(text_to_summarize) | |
| st.markdown("#### Summary:") | |
| st.write(summary) |