first 'working' version

This commit is contained in:
2024-12-29 23:43:26 +01:00
parent 6200763f0b
commit b3c30398c2
11 changed files with 469 additions and 36 deletions
+61
View File
@@ -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))