diff --git a/app.py b/app.py index ea2856b..3b8375b 100644 --- a/app.py +++ b/app.py @@ -47,7 +47,7 @@ Use only russian language! faster_whisper_models_list = ['tiny', 'base', 'small', 'medium', 'large-v1', 'large-v2', 'large-v3', 'large', 'distil-large-v2', 'distil-large-v3', 'distil-large-v3.5', 'large-v3-turbo', 'turbo'] llm_models_list = ['openai/gpt-oss-120b', 'Qwen/Qwen3-235B-A22B-Thinking-2507', 'deepseek-ai/DeepSeek-R1-0528', 'meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8', 'openai/gpt-oss-20b', 'Intel/Qwen3-Coder-480B-A35B-Instruct-int4-mixed-ar', 'meta-llama/Llama-3.2-90B-Vision-Instruct', 'mistralai/Mistral-Nemo-Instruct-2407', 'Qwen/Qwen2.5-VL-32B-Instruct', 'meta-llama/Llama-3.3-70B-Instruct', 'mistralai/Devstral-Small-2505', 'mistralai/Magistral-Small-2506', 'mistralai/Mistral-Large-Instruct-2411', 'CohereForAI/aya-expanse-32b'] - +is_pipeline_enabled = True # Функция транскрибации def recognize(model, audioFile, beamSize, vadFilter, minSilenceDurationMs, speechPadMs, temp0, temp1, temp2, wordTimestamps, noSpeechThreshold, conditionOnPreviousText): @@ -74,6 +74,23 @@ def recognize(model, audioFile, beamSize, vadFilter, minSilenceDurationMs, speec return(text) +def generateByCondition(api_key, llm_model, system_prompt, recognized_text, llm_temperature, is_pipeline_enabled, trigger): + llm = Llm(api_key) + + # если чекбокс включен и событие было change → обрабатываем + if is_pipeline_enabled and trigger == "change": + result, md = llm.generate(llm_model, system_prompt, recognized_text, llm_temperature) + saveFile("output.txt", result) + return result, result + + # если чекбокс выключен и событие было click → обрабатываем + if not is_pipeline_enabled and trigger == "click": + result, md = llm.generate(llm_model, system_prompt, recognized_text, llm_temperature) + saveFile("output.txt", result) + return result, md + + # если нет чекбокса и было событие change + return gr.skip(), gr.skip() def format_timestamp(seconds: float) -> str: millis = int(seconds * 1000) @@ -90,7 +107,15 @@ def saveFile(filename, text): filePath.parent.mkdir(parents=True, exist_ok=True) filePath.write_text(text, encoding='utf-8') +# Функция для динамического обновления кнопки в зависимости от состояния checkbox +def updateButton(isChecked): + if not isChecked: + variant = 'primary' + else: + variant = 'secondary' + return gr.update(interactive=not isChecked, variant=variant) +# Интерфейс with gr.Blocks() as demo: gr.HTML('''