diff --git a/app.py b/app.py index 7f9cb8e..906d2a4 100644 --- a/app.py +++ b/app.py @@ -71,8 +71,15 @@ def updateButton(isChecked): def updateTextbox(isChecked): return gr.update(visible=isChecked) +################################################### +# ____ ___.___ ___. .__ # +#| | \ | \_ |__ ____ | | ______ _ __# +#| | / | | __ \_/ __ \| | / _ \ \/ \/ /# +#| | /| | | \_\ \ ___/| |_( <_> ) / # +#|______/ |___| |___ /\___ >____/\____/ \/\_/ # +# \/ \/ # +################################################### -# Интерфейс with gr.Blocks() as demo: gr.HTML('''
@@ -83,72 +90,82 @@ with gr.Blocks() as demo: ''') with gr.Row(): - with gr.Column(): - # Колонка в левой части экрана с основным взаимодействием - with gr.Accordion(label='Recognization'): - with gr.Column(): - audioFile = gr.Audio(label='Load audio for transcribe', type="filepath") - recognizeBtn = gr.Button('recognize', variant='primary') + # Вкладка с основным взаимодействием + with gr.Tab('Actions'): + isPipelineEnabledCheckbox = gr.Checkbox(label='is pipeline enabled', value=True, interactive=True) - with gr.Accordion(label='Recognized text'): - recognizedText = gr.TextArea(label='') + with gr.Row(): + with gr.Accordion(label='Recognization and integration'): + with gr.Column(): + audioFile = gr.Audio(label='Load audio for transcribe', type="filepath") + images = gr.Files(label='Upload images', file_types=['image']) + recognizeBtn = gr.Button('recognize and integrate', variant='primary') - # Колонка в правой части экрана с настройками + with gr.Accordion(label='Recognized text'): + recognizedText = gr.TextArea(label='') + + with gr.Accordion(label='LLM'): + with gr.Column(): + refineTextBtn = gr.Button('refine text', variant='secondary', interactive=False) + + with gr.Accordion(label='Refined text raw'): + refinedText = gr.Textbox(label='', show_copy_button=True) + + with gr.Accordion(label='Refined text md formated'): + refinedTextMD = gr.Markdown(label='') + + # Вкладка с настройками + with gr.Tab('Settings'): with gr.Column(): - with gr.Accordion(label='Settings'): - # Первое поле на всю ширину в акордионе настроек - with gr.Group(): - saveFileCheckbox = gr.Checkbox(label='save file', value=True, interactive=True) - filename = gr.Textbox(label='Output filename', value='output.txt', interactive=True) - filenamePdf = gr.Textbox(label='Output filename for pdf', value='output.pdf', interactive=True) + # Первое поле на всю ширину в акордионе настроек + with gr.Accordion('File settings'): + saveFileCheckbox = gr.Checkbox(label='save file', value=True, interactive=True) + filename = gr.Textbox(label='Output filename', value='output.txt', interactive=True) + filenamePdf = gr.Textbox(label='Output filename for pdf', value='output.pdf', interactive=True) + # Акордион настроек faster whisper + with gr.Accordion(label='Faster whisper settings'): + with gr.Row(): + # Левая колонка в акордионе + with gr.Column(): + fastWhisperModel = gr.Dropdown(label='Model', choices=FAST_WHISPER_MODELS, value=FAST_WHISPER_MODELS[11], interactive=True) - # Акордион настроек faster whisper - with gr.Accordion(label='Faster whisper settings'): - with gr.Row(): - # Левая колонка в акордионе - with gr.Column(): - fastWhisperModel = gr.Dropdown(label='Model', choices=FAST_WHISPER_MODELS, value=FAST_WHISPER_MODELS[11], interactive=True) + beamSize = gr.Number(label='beam_size', value=8, interactive=True) + noSpeechThreshold = gr.Number(label='no_speech_threshold', value=0.5, interactive=True) + vadFilter = gr.Checkbox(label='vad_filter', value=True, interactive=True) + wordTimestamps = gr.Checkbox(label='word_timestamps', value=True, interactive=True) + conditionOnPreviousText = gr.Checkbox(label='condition_on_previous_text', value=False, interactive=True) - beamSize = gr.Number(label='beam_size', value=8, interactive=True) - noSpeechThreshold = gr.Number(label='no_speech_threshold', value=0.5, interactive=True) - vadFilter = gr.Checkbox(label='vad_filter', value=True, interactive=True) - wordTimestamps = gr.Checkbox(label='word_timestamps', value=True, interactive=True) - conditionOnPreviousText = gr.Checkbox(label='condition_on_previous_text', value=False, interactive=True) + # Правая колонка в акордионе + with gr.Column(): + with gr.Accordion(label='Vad parameters'): + minSilenceDurationMs = gr.Number(label='min_silence_duration_ms', value=300, interactive=True) + speechPadMs = gr.Number(label='speech_pad_ms', value=200, interactive=True) - # Правая колонка в акордионе - with gr.Column(): - with gr.Accordion(label='Vad parameters'): - minSilenceDurationMs = gr.Number(label='min_silence_duration_ms', value=300, interactive=True) - speechPadMs = gr.Number(label='speech_pad_ms', value=200, interactive=True) + with gr.Accordion(label='Temperature'): + temp0 = gr.Number(label='temp_0', value=0.0, interactive=True) + temp1 = gr.Number(label='temp_1', value=0.2, interactive=True) + temp2 = gr.Number(label='temp_2', value=0.4, interactive=True) - with gr.Accordion(label='Temperature'): - temp0 = gr.Number(label='temp_0', value=0.0, interactive=True) - temp1 = gr.Number(label='temp_1', value=0.2, interactive=True) - temp2 = gr.Number(label='temp_2', value=0.4, interactive=True) + # Нижний акордион настроек для api ключа llm + with gr.Accordion(label='ai.io.net api settings'): + apiKey = gr.Textbox(label='API key', value=DEFAULT_API_KEY, interactive=True) - # Нижний акордион настроек для api ключа llm - with gr.Accordion(label='ai.io.net api settings'): - apiKey = gr.Textbox(label='API key', value=DEFAULT_API_KEY, interactive=True) + with gr.Accordion(label='System prompt'): + systemPrompt = gr.Textbox(label='', value=DEFAULT_SYSTEM_PROMPT, interactive=True) - with gr.Accordion(label='System prompt'): - systemPrompt = gr.Textbox(label='', value=DEFAULT_SYSTEM_PROMPT, interactive=True) - - with gr.Row(): - llmModel = gr.Dropdown(label='models', choices=LLM_MODELS, value=LLM_MODELS[1], interactive=True) - llmTemperature = gr.Number(label='Temperature', value=0.8, interactive=True ) - - with gr.Column(): - with gr.Accordion(label='LLM'): - isPipelineEnabledCheckbox = gr.Checkbox(label='is pipeline enabled', value=True, interactive=True) - refineTextBtn = gr.Button('refine text', variant='secondary', interactive=False) - - with gr.Accordion(label='Refined text raw'): - refinedText = gr.Textbox(label='', show_copy_button=True) - - with gr.Accordion(label='Refined text md formated'): - refinedTextMD = gr.Markdown(label='') + with gr.Row(): + llmModel = gr.Dropdown(label='models', choices=LLM_MODELS, value=LLM_MODELS[1], interactive=True) + llmTemperature = gr.Number(label='Temperature', value=0.8, interactive=True ) + ###################################################################### + #.____ .__ ___. .__ # + #| | ____ ____ |__| ____ \_ |__ ____ | | ______ _ __# + #| | / _ \ / ___\| |/ ___\ | __ \_/ __ \| | / _ \ \/ \/ /# + #| |__( <_> ) /_/ > \ \___ | \_\ \ ___/| |_( <_> ) / # + #|_______ \____/\___ /|__|\___ > |___ /\___ >____/\____/ \/\_/ # + # \/ /_____/ \/ \/ \/ # + ###################################################################### isPipelineEnabledCheckbox.change(updateButton, inputs=[isPipelineEnabledCheckbox], outputs=refineTextBtn) saveFileCheckbox.change(updateTextbox, inputs=saveFileCheckbox, outputs=filename)