jlengrand/tldw
1
0
Fork 0
mirror of https://github.com/jlengrand/tldw.git synced 2026-03-10 00:41:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

First stab at allowing video downloads in the UI

Browse Source
This commit is contained in:
Robert
2024-05-10 14:09:37 -07:00
parent ecc3630dad
commit 747cefca0f
3 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
Tests/HF/app.py → HF/app.py
View File

@@ -360,7 +360,7 @@ def process_url(input_path, num_speakers=2, whisper_model="small.en", offset=0,
try:
results = main(input_path, api_name=api_name, api_key=api_key, num_speakers=num_speakers, whisper_model=whisper_model, offset=offset, vad_filter=vad_filter, download_video_flag=download_video_flag)
if results:
transcription_result = results[0]
json_file_path = transcription_result['audio_file'].replace('.wav', '.segments.json')
@@ -1281,7 +1281,10 @@ def launch_ui(demo_mode=False):
iface = gr.Interface(
# fn=lambda url, num_speakers, whisper_model, offset, api_name, api_key: process_url(url, num_speakers, whisper_model, offset, api_name=api_name, api_key=api_key, demo_mode=demo_mode),
fn=lambda *args: process_url(*args, demo_mode=demo_mode),
fn=lambda url, num_speakers, whisper_model, offset, api_name, api_key: process_url(url, num_speakers, whisper_model, offset, api_name, api_key, demo_mode=demo_mode),
# fn=lambda *args: process_url(*args, demo_mode=demo_mode),
inputs=inputs,
outputs=[
gr.components.Textbox(label="Transcription", value=lambda: "", max_lines=10),

0
Tests/HF/requirements.txt → HF/requirements.txt
View File

15
summarize.py
View File

@@ -359,14 +359,16 @@ def process_url(input_path, num_speakers=2, whisper_model="small.en", offset=0,
summary_file_path = json_file_path.replace('.segments.json', '_summary.txt')
if os.path.exists(summary_file_path):
return json_data, summary_file_path, json_file_path, summary_file_path
video_file_path = transcription_result['video_path'] if download_video_flag else None
return json_data, summary_file_path, json_file_path, summary_file_path, video_file_path
else:
return json_data, "Summary not available.", json_file_path, None
video_file_path = transcription_result['video_path'] if download_video_flag else None
return json_data, "Summary not available.", json_file_path, None, video_file_path
else:
return None, "No results found.", None, None
return None, "No results found.", None, None, None
except Exception as e:
error_message = f"An error occurred: {str(e)}"
return None, error_message, None, None
return None, error_message, None, None, None
@@ -474,7 +476,7 @@ def download_video(video_url, download_path, info_dict, download_video_flag):
output_file_path = os.path.join(download_path, f"{title}.mp4")
if userOS == "Windows":
if sys.platform.startswith('win'):
logging.debug("Running ffmpeg on Windows...")
ffmpeg_command = [
'.\\Bin\\ffmpeg.exe',
@@ -1233,7 +1235,8 @@ def launch_ui(demo_mode=False):
gr.components.Textbox(label="Transcription", value=lambda: "", max_lines=10),
gr.components.Textbox(label="Summary or Status Message"),
gr.components.File(label="Download Transcription as JSON"),
gr.components.File(label="Download Summary as text", visible=lambda summary_file_path: summary_file_path is not None)
gr.components.File(label="Download Summary as text", visible=lambda summary_file_path: summary_file_path is not None),
gr.components.File(label="Download Video", visible=lambda video_file_path: video_file_path is not None)
],
title="Video Transcription and Summarization",
description="Submit a video URL for transcription and summarization.",