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

status bar

Browse Source
This commit is contained in:
tcsenpai 2025-05-23 10:43:51 +02:00
parent 8c32277e85
commit 647f7c7b22
1 changed files with 38 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
app.py
View File

@ -187,7 +187,7 @@ def create_interface():
"### A powerful tool for transcribing and summarizing audio/video content" "### A powerful tool for transcribing and summarizing audio/video content"
) )
with gr.Tabs() as tabs: with gr.Tabs(selected=1) as tabs: # Set YouTube tab as default (index 1)
with gr.TabItem("Local File"): with gr.TabItem("Local File"):
gr.Markdown( gr.Markdown(
""" """
@ -323,6 +323,15 @@ def create_interface():
label="Ollama Model", label="Ollama Model",
interactive=OLLAMA_AVAILABLE, interactive=OLLAMA_AVAILABLE,
) )
# Add status bar
yt_status = gr.Textbox(
label="Status",
value="Waiting for input...",
interactive=False,
elem_classes=["status-bar"],
)
yt_process_btn = gr.Button("Process Video", variant="primary") yt_process_btn = gr.Button("Process Video", variant="primary")
with gr.Column(): with gr.Column():
@ -343,13 +352,33 @@ def create_interface():
# Set up the event handler # Set up the event handler
def process_yt_with_summary(url, model, lang, summarize, ollama_model): def process_yt_with_summary(url, model, lang, summarize, ollama_model):
result = process_youtube_url( try:
url, model, lang, summarize, ollama_model # Update status for each step
) yield "Checking URL and fetching video information...", None, None, None, None
if len(result) == 4: result = process_youtube_url(
text, lang, source, summary = result url, model, lang, summarize, ollama_model
return text, lang, source, summary if summary else "" )
return result[0], result[1], result[2], ""
if len(result) == 4:
text, lang, source, summary = result
if source == "Subtitles":
yield f"Processing subtitles...", None, None, None, None
else:
yield f"Transcribing video...", None, None, None, None
if summarize and summary:
yield f"Generating summary...", None, None, None, None
yield text, lang, source, (
summary if summary else ""
), "Processing complete!"
else:
yield result[0], result[1], result[
2
], "", f"Error: {result[0]}"
except Exception as e:
logger.error(f"Error in process_yt_with_summary: {str(e)}")
yield f"Error: {str(e)}", None, None, "", "Processing failed!"
yt_process_btn.click( yt_process_btn.click(
fn=process_yt_with_summary, fn=process_yt_with_summary,
@ -365,6 +394,7 @@ def create_interface():
yt_detected_language, yt_detected_language,
yt_source, yt_source,
yt_summary_text if OLLAMA_AVAILABLE else gr.Textbox(), yt_summary_text if OLLAMA_AVAILABLE else gr.Textbox(),
yt_status,
], ],
) )