tcsenpai/youlama
1
0
Fork 0
mirror of https://github.com/tcsenpai/youlama.git synced 2025-06-07 03:35:41 +00:00
Code Issues 1 Packages Projects Releases Wiki Activity

youtu.be safer management and & params remover

Browse Source
This commit is contained in:
tcsenpai 2024-12-15 16:00:46 +01:00
parent 4e89d75962
commit 70ae2fb11a
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/main.py
View File

@ -76,15 +76,19 @@ if "messages" not in st.session_state:
# Create a single header container # Create a single header container
header = st.container() header = st.container()
def show_warning(message): def show_warning(message):
update_header("⚠️ " + message) update_header("⚠️ " + message)
def show_error(message): def show_error(message):
update_header("🚫 " + message) update_header("🚫 " + message)
def show_info(message): def show_info(message):
update_header("✅ " + message) update_header("✅ " + message)
def update_header(message): def update_header(message):
with header: with header:
st.markdown( st.markdown(
@ -116,6 +120,7 @@ def update_header(message):
unsafe_allow_html=True, unsafe_allow_html=True,
) )
# Initialize the header with a ready message # Initialize the header with a ready message
update_header("✅ Ready to summarize!") update_header("✅ Ready to summarize!")
@ -158,7 +163,15 @@ def get_ollama_models(ollama_url):
def summarize_video( def summarize_video(
video_url, model, ollama_url, fallback_to_whisper=True, force_whisper=False video_url, model, ollama_url, fallback_to_whisper=True, force_whisper=False
): ):
video_id = video_url.split("v=")[-1] video_id = None
# Get the video id from the url if it's a valid youtube or invidious or any other url that contains a video id
if "v=" in video_url:
video_id = video_url.split("v=")[-1]
# Support short urls as well
elif "youtu.be/" in video_url:
video_id = video_url.split("youtu.be/")[-1]
# Also cut out any part of the url after the video id
video_id = video_id.split("&")[0]
st.write(f"Video ID: {video_id}") st.write(f"Video ID: {video_id}")
with st.spinner("Fetching transcript..."): with st.spinner("Fetching transcript..."):
@ -350,9 +363,7 @@ URL: {video_url}
{summary['transcript']}""" {summary['transcript']}"""
paste_url = create_paste( paste_url = create_paste(f"Transcript: {summary['title']}", content)
f"Transcript: {summary['title']}", content
)
st.success( st.success(
f"Transcript shared successfully! [View here]({paste_url})" f"Transcript shared successfully! [View here]({paste_url})"
) )