fe-test / app.py
pionero-thien-nguyenduc's picture
init file
68a6cd6
import streamlit as st
import tempfile
st.set_page_config(
page_title="PC-AVS",
layout="wide",
initial_sidebar_state="collapsed"
)
st.header("PC-AVS")
col1, col2 = st.columns(2)
result = None
# Upload video and audio
with col1:
st.header("Video Input")
video_file = st.file_uploader("Upload video", type=["mp4", "avi", "mov"])
if video_file is not None:
# Show video
temp_video = tempfile.NamedTemporaryFile(delete=False)
temp_video.write(video_file.read())
st.video(temp_video.name)
else:
st.write("No video have been uploaded yet.")
st.header("Audio Input")
audio_file = st.file_uploader("Upload audio", type=["mp3", "wav", "ogg"])
if audio_file is not None:
temp_audio = tempfile.NamedTemporaryFile(delete=False)
temp_audio.write(audio_file.read())
st.audio(temp_audio.name)
else:
st.write("No audio have been uploaded yet.")
if st.button("Submit", key="replay-button"):
if video_file is not None and audio_file is not None:
result = video_file
else:
st.write("Please upload video and audio")
# Show result
with col2:
st.header("Result")
if result is not None:
# Show video
result_video = tempfile.NamedTemporaryFile(delete=False)
result_video.write(result.read())
st.video(result_video.name)
else:
st.write("No result")