Update app.py

app.py

@@ -4,8 +4,11 @@ import os
 import shutil
 import sys
 import yt_dlp  # для працы з YouTube
-import glob # Import glob for file searching
-import os # Import os for path manipulation
+import glob  # Import glob for file searching
+import os  # Import os for path manipulation
+
+# Атрыманне GEMINI API Key з пераменных асяроддзя
+GEMINI_API_KEY = os.getenv("gembeh")
 
 from sub_tools.media.converter import hls_to_media, media_to_signature, video_to_audio
 from sub_tools.media.segmenter import segment_audio
@@ -14,9 +17,6 @@ from sub_tools.system.directory import change_directory
 from sub_tools.system.console import header, success, error
 from sub_tools.transcribe import transcribe
 
-GEMINI_API_KEY = os.getenv("gembeh")
-print(GEMINI_API_KEY)
-
 # Вызначэнне функцый для дэактывацыі процілеглага поля
 def update_on_audio_change(audio):
     if audio is not None:
@@ -72,14 +72,13 @@ def main_logic(args: Namespace) -> tuple:
             header(f"{step}. Segment Audio")
             output_str += f"{step}. Segment Audio: Started\n"
 
-            print(f"Current working directory before segment_audio: {os.getcwd()}") # Debug CWD
-            print(f"Audio file path for segmentation: {args.audio_file}") # Debug audio file path
-            if not os.path.exists(args.audio_file): # Critical file existence check
+            print(f"Current working directory before segment_audio: {os.getcwd()}")  # Debug CWD
+            print(f"Audio file path for segmentation: {args.audio_file}")  # Debug audio file path
+            if not os.path.exists(args.audio_file):  # Critical file existence check
                 error_msg = f"Error: Audio file not found before segmentation: {args.audio_file}"
                 error(error_msg)
                 return (error_msg, None)
-            print(f"Audio file exists before segmentation: {args.audio_file}") # Debug file existence
-
+            print(f"Audio file exists before segmentation: {args.audio_file}")  # Debug file existence
 
             segment_audio(args.audio_file, args.audio_segment_prefix, args.audio_segment_format, args.audio_segment_length, args.overwrite)
             success("Done!")
@@ -132,7 +131,7 @@ def run_subtools(
     languages,
     overwrite,
     retry,
-    gemini_api_key, # Тут замяняем
+    gemini_api_key,
     debug,
     audio_segment_prefix,
     audio_segment_format,
@@ -158,7 +157,7 @@ def run_subtools(
         languages=languages,
         overwrite=overwrite,
         retry=retry,
-        gemini_api_key=GEMINI_API_KEY, # Замянілі тут
+        gemini_api_key=gemini_api_key,
         debug=debug,
         audio_segment_prefix=audio_segment_prefix,
         audio_segment_format=audio_segment_format,
@@ -176,9 +175,9 @@ def transcribe_youtube(youtube_url: str) -> tuple:
     if not youtube_url:
         return "Не ўведзена спасылка", None
 
-    output_dir = "output_youtube_downloads" # Define a specific output directory
-    os.makedirs(output_dir, exist_ok=True) # Create the directory if it doesn't exist
-    outtmpl = os.path.join(output_dir, "temp_youtube_audio.%(ext)s") # Path within the output dir
+    output_dir = "output_youtube_downloads"  # Define a specific output directory
+    os.makedirs(output_dir, exist_ok=True)  # Create the directory if it doesn't exist
+    outtmpl = os.path.join(output_dir, "temp_youtube_audio.%(ext)s")  # Path within the output dir
 
     try:
         ydl_opts = {
@@ -187,7 +186,7 @@ def transcribe_youtube(youtube_url: str) -> tuple:
             #'quiet': True,  # Прыбіраем quiet для адладкі
             'postprocessors': [{
                 'key': 'FFmpegExtractAudio',
-                'preferredcodec': 'mp3', # Download as MP3
+                'preferredcodec': 'mp3',  # Download as MP3
                 'preferredquality': '0',  # лепшая якасць
             }],
         }
@@ -202,11 +201,11 @@ def transcribe_youtube(youtube_url: str) -> tuple:
 
         # Шукаем створаны аўдыёфайл у output_dir
         audio_files = glob.glob(os.path.join(output_dir, "temp_youtube_audio.*"))
-        print(f"Знойдзеныя аўдыёфайлы ў {output_dir}: {audio_files}") # Дадаем вывад для адладкі
+        print(f"Знойдзеныя аўдыёфайлы ў {output_dir}: {audio_files}")  # Дадаем вывад для адладкі
         if not audio_files:
             return ("Памылка: файл аўдыё не створаны", None)
 
-        audio_file_rel = None # Relative path
+        audio_file_rel = None  # Relative path
         for f in audio_files:
             if f.endswith(".mp3"):
                 audio_file_rel = f
@@ -214,11 +213,11 @@ def transcribe_youtube(youtube_url: str) -> tuple:
         if audio_file_rel is None:
             audio_file_rel = audio_files[0]
 
-        audio_file_abs = os.path.abspath(audio_file_rel) # Get absolute path here!
+        audio_file_abs = os.path.abspath(audio_file_rel)  # Get absolute path here!
 
-        print(f"Абраны аўдыёфайл (relative): {audio_file_rel}") # Дадаем вывад для адладкі
-        print(f"Абраны аўдыёфайл (absolute): {audio_file_abs}") # Дадаем вывад для адладкі
-        return ("Спампоўка YouTube аўдыё завершана", audio_file_abs) # Return absolute path!
+        print(f"Абраны аўдыёфайл (relative): {audio_file_rel}")  # Дадаем вывад для адладкі
+        print(f"Абраны аўдыёфайл (absolute): {audio_file_abs}")  # Дадаем вывад для адладкі
+        return ("Спампоўка YouTube аўдыё завершана", audio_file_abs)  # Return absolute path!
     except Exception as e:
         err_msg = f"Памылка пры апрацоўцы YouTube (yt_dlp): {e}. Калі ласка, пераканайцеся, што спасылка сапраўдная, і праверце, ці даступныя відэа."
         print(err_msg)
@@ -242,7 +241,7 @@ def process_youtube_url(youtube_url):
 
     output_text_file, output_file_file = process_uploaded_file(audio=full_audio_file_path, video=None)
 
-    return output_text_file, output_file_file # Correct return order: text, file_path
+    return output_text_file, output_file_file  # Correct return order: text, file_path
 
 
 def process_uploaded_file(audio, video):
@@ -274,7 +273,7 @@ def process_uploaded_file(audio, video):
         languages="be",
         overwrite=False,
         retry=50,
-        gemini_api_key=GEMINI_API_KEY, # Замянілі тут
+        gemini_api_key=GEMINI_API_KEY,
         debug=False,
         audio_segment_prefix="audio_segment",
         audio_segment_format="mp3",
@@ -286,8 +285,8 @@ with gr.Blocks() as demo:
     with gr.Tabs():
         with gr.Tab("Файл"):
             with gr.Row():
-                audio_input = gr.Audio(type="filepath", label="Аўдыёфайл") # Keep type="filepath"
-                video_input = gr.Video(label="Відэафайл", interactive=True) # Reset to interactive True initially
+                audio_input = gr.Audio(type="filepath", label="Аўдыёфайл")  # Keep type="filepath"
+                video_input = gr.Video(label="Відэафайл", interactive=True)  # Reset to interactive True initially
 
             audio_input.change(fn=update_on_audio_change, inputs=audio_input, outputs=video_input)
             video_input.change(fn=update_on_video_change, inputs=video_input, outputs=audio_input)
@@ -306,8 +305,7 @@ with gr.Blocks() as demo:
             submit_btn_youtube.click(
                 fn=process_youtube_url,
                 inputs=youtube_url_input,
-                outputs=[output_text_youtube, output_file_youtube] # Correct output order
+                outputs=[output_text_youtube, output_file_youtube]  # Correct output order
             )
 
-
-demo.launch(allowed_paths=["output_youtube_downloads"])
+demo.launch(allowed_paths=["output_youtube_downloads"])