first 'working' version
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
import os
|
||||
|
||||
import streamlit as st
|
||||
|
||||
from utils import get_logger
|
||||
from transcription.job_utils import create_job
|
||||
from transcription.transcription_helper import transcribe
|
||||
|
||||
logger = get_logger(__name__)
|
||||
|
||||
|
||||
def create_new_task(filepath: str, app_config, overwrite: bool):
|
||||
"""
|
||||
Create a new task
|
||||
:param filepath: path to the audio file
|
||||
:param app_config: app configuration
|
||||
:param overwrite: overwrite existing task
|
||||
:return: None
|
||||
"""
|
||||
logger.info(f"Creating new task for file {filepath}")
|
||||
|
||||
if filepath is None:
|
||||
st.error("No file uploaded")
|
||||
return
|
||||
|
||||
transcribe(filepath, app_config, blocking=False, overwrite=overwrite)
|
||||
|
||||
|
||||
def render_new_task(container, app_config: dict):
|
||||
"""
|
||||
Render the new task page
|
||||
:param container: container object to place the content in
|
||||
:param app_config: app configuration
|
||||
:return: None
|
||||
"""
|
||||
|
||||
with container.container():
|
||||
st.write("Create a new task")
|
||||
# add checkbox for overwrite
|
||||
with st.form(key="new_task_form", clear_on_submit=True):
|
||||
overwrite = st.checkbox("Overwrite existing task", value=False, key="overwrite")
|
||||
uploaded_file = st.file_uploader("Audio File", type=["wav", "mp3", "flac", "ogg"], key="audio_file")
|
||||
file_path = None
|
||||
|
||||
if uploaded_file is not None:
|
||||
# save files to upload directory
|
||||
upload_directory = app_config["data"]["upload_directory"]
|
||||
|
||||
# create the upload directory if it does not exist
|
||||
if not os.path.exists(upload_directory):
|
||||
logger.info(f"Creating upload directory {upload_directory}")
|
||||
os.makedirs(upload_directory)
|
||||
|
||||
logger.info(f"Saving file {uploaded_file.name} to {upload_directory}")
|
||||
file_path = f"{upload_directory}/{uploaded_file.name}"
|
||||
with open(file_path, "wb") as file:
|
||||
file.write(uploaded_file.getbuffer())
|
||||
|
||||
st.success(f"File saved to {file_path}")
|
||||
|
||||
st.form_submit_button("Create Task", on_click=create_new_task, args=(file_path, app_config, overwrite))
|
||||
Reference in New Issue
Block a user