tcsenpai/youlama
1
0
Fork 0
mirror of https://github.com/tcsenpai/youlama.git synced 2025-06-10 21:17:38 +00:00
Code Issues 1 Packages Projects Releases Wiki Activity

refactored ollama gui

Browse Source
This commit is contained in:
tcsenpai 2025-05-23 10:34:21 +02:00
parent b99f47320b
commit 51e5e5eefe
1 changed files with 41 additions and 93 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

134
app.py
View File

@ -215,42 +215,24 @@ def create_interface():
value="Auto-detect", value="Auto-detect",
label="Language (optional)", label="Language (optional)",
) )
if OLLAMA_AVAILABLE: with gr.Group():
with gr.Group(): summarize_checkbox = gr.Checkbox(
summarize_checkbox = gr.Checkbox( label="Generate Summary",
label="Generate Summary", value=False value=False,
) interactive=OLLAMA_AVAILABLE,
ollama_model_dropdown = gr.Dropdown( )
choices=OLLAMA_MODELS, ollama_model_dropdown = gr.Dropdown(
value=DEFAULT_OLLAMA_MODEL, choices=(
label="Ollama Model", OLLAMA_MODELS
visible=False, if OLLAMA_AVAILABLE
) else ["No models available"]
),
def toggle_summary(checked): value=(
if checked and not ollama.is_available(): DEFAULT_OLLAMA_MODEL if OLLAMA_AVAILABLE else None
gr.Warning( ),
"Ollama is not available. Please check your Ollama server." label="Ollama Model",
) interactive=OLLAMA_AVAILABLE,
return False )
return checked
def update_model_visibility(visible):
if visible:
return gr.Dropdown.update(
visible=True, value=DEFAULT_OLLAMA_MODEL
)
return gr.Dropdown.update(visible=False)
summarize_checkbox.change(
fn=toggle_summary,
inputs=[summarize_checkbox],
outputs=[summarize_checkbox],
).then(
fn=update_model_visibility,
inputs=[summarize_checkbox],
outputs=[ollama_model_dropdown],
)
transcribe_btn = gr.Button("Transcribe", variant="primary") transcribe_btn = gr.Button("Transcribe", variant="primary")
with gr.Column(): with gr.Column():
@ -284,16 +266,8 @@ def create_interface():
audio_input, audio_input,
model_dropdown, model_dropdown,
language_dropdown, language_dropdown,
( summarize_checkbox,
summarize_checkbox ollama_model_dropdown,
if OLLAMA_AVAILABLE
else gr.Checkbox(value=False)
),
(
ollama_model_dropdown
if OLLAMA_AVAILABLE
else gr.Dropdown(value=None)
),
], ],
outputs=[ outputs=[
output_text, output_text,
@ -331,42 +305,24 @@ def create_interface():
value="Auto-detect", value="Auto-detect",
label="Language (optional)", label="Language (optional)",
) )
if OLLAMA_AVAILABLE: with gr.Group():
with gr.Group(): yt_summarize_checkbox = gr.Checkbox(
yt_summarize_checkbox = gr.Checkbox( label="Generate Summary",
label="Generate Summary", value=False value=False,
) interactive=OLLAMA_AVAILABLE,
yt_ollama_model_dropdown = gr.Dropdown( )
choices=OLLAMA_MODELS, yt_ollama_model_dropdown = gr.Dropdown(
value=DEFAULT_OLLAMA_MODEL, choices=(
label="Ollama Model", OLLAMA_MODELS
visible=False, if OLLAMA_AVAILABLE
) else ["No models available"]
),
def toggle_yt_summary(checked): value=(
if checked and not ollama.is_available(): DEFAULT_OLLAMA_MODEL if OLLAMA_AVAILABLE else None
gr.Warning( ),
"Ollama is not available. Please check your Ollama server." label="Ollama Model",
) interactive=OLLAMA_AVAILABLE,
return False )
return checked
def update_yt_model_visibility(visible):
if visible:
return gr.Dropdown.update(
visible=True, value=DEFAULT_OLLAMA_MODEL
)
return gr.Dropdown.update(visible=False)
yt_summarize_checkbox.change(
fn=toggle_yt_summary,
inputs=[yt_summarize_checkbox],
outputs=[yt_summarize_checkbox],
).then(
fn=update_yt_model_visibility,
inputs=[yt_summarize_checkbox],
outputs=[yt_ollama_model_dropdown],
)
yt_process_btn = gr.Button("Process Video", variant="primary") yt_process_btn = gr.Button("Process Video", variant="primary")
with gr.Column(): with gr.Column():
@ -399,16 +355,8 @@ def create_interface():
youtube_url, youtube_url,
yt_model_dropdown, yt_model_dropdown,
yt_language_dropdown, yt_language_dropdown,
( yt_summarize_checkbox,
yt_summarize_checkbox yt_ollama_model_dropdown,
if OLLAMA_AVAILABLE
else gr.Checkbox(value=False)
),
(
yt_ollama_model_dropdown
if OLLAMA_AVAILABLE
else gr.Dropdown(value=None)
),
], ],
outputs=[ outputs=[
yt_output_text, yt_output_text,
@ -428,7 +376,7 @@ def create_interface():
- Maximum audio duration is {MAX_DURATION // 60} minutes - Maximum audio duration is {MAX_DURATION // 60} minutes
- YouTube videos will first try to use available subtitles - YouTube videos will first try to use available subtitles
- If no subtitles are available, the video will be transcribed - If no subtitles are available, the video will be transcribed
{"- Ollama summarization is available for both local files and YouTube videos" if OLLAMA_AVAILABLE else ""} {"- Ollama summarization is available for both local files and YouTube videos" if OLLAMA_AVAILABLE else "- Ollama summarization is currently unavailable"}
### Status: ### Status:
- Device: {DEVICE} - Device: {DEVICE}