Update app.py

app.py

@@ -1,28 +1,20 @@
 import os
 import gradio as gr
-from google import genai
-from google.genai import types
+import google.generativeai as genai
 import mimetypes
 from pydub import AudioSegment
 
 # Атрыманне ключоў і мадэляў з пераменных асяроддзя
 GEMINI_API_KEY = os.getenv("gemini")
 MODEL_NAME_TH = os.getenv("modTH")  # Мадэль для транскрыпцыі
-MODEL_NAME = os.getenv(
-    "mod"
-)  # Мадэль для выпраўлення фармату і перакладу (можа быць той жа, што і modTH)
-PROMPT_TRANSCRIBE = os.getenv(
-    "p"
-)  # Промпт для транскрыпцыі (лепш мець зразумелае імя)
-
-
-# Стварэнне кліента (больш просты спосаб)
-client = genai.GenerativeModel(model_name=MODEL_NAME_TH, api_key=GEMINI_API_KEY)
-client_general = genai.GenerativeModel(model_name=MODEL_NAME, api_key=GEMINI_API_KEY)
+MODEL_NAME = os.getenv("mod")       # Мадэль для выпраўлення фармату і перакладу
+PROMPT_TRANSCRIBE = os.getenv("p")  # Промпт для транскрыпцыі
 
+# Наладжваем API-ключ
+genai.configure(api_key=GEMINI_API_KEY)
 
 def transcribe_audio(audio_file: str) -> str:
-    """Транскрыбуе аўдыяфайл з дапамогай Google GenAI."""
+    """Транскрыбуе аўдыяфайл з дапамогай Google Generative AI."""
     try:
         mime_type, _ = mimetypes.guess_type(audio_file)
         if not mime_type or not mime_type.startswith("audio"):
@@ -34,24 +26,18 @@ def transcribe_audio(audio_file: str) -> str:
         with open(audio_file, "rb") as f:
             audio_data = f.read()
 
-        response = client.generate_content(
-            [
-                PROMPT_TRANSCRIBE,
-                {"mime_type": mime_type, "data": audio_data},
-            ]  # ,  # Промпт + дадзеныя
-            # generation_config=types.GenerationConfig( # можна задаць дадатковыя параметры, напр. temperature
-            #     temperature=0.2,
-            # )
+        response = genai.generate_text(
+            model=MODEL_NAME_TH,
+            prompt=[PROMPT_TRANSCRIBE, {"mime_type": mime_type, "data": audio_data}],
+            # Пры жаданні можна дадаць дадатковыя параметры, напр. temperature=0.2
         )
-        response.resolve()  #  Яўнае дазвол прамісаў (калі выкарыстоўваецца асінхронны рэжым)
-        return response.text.strip()
+        return response.result.strip()
 
     except FileNotFoundError:
         return "Памылка: Файл не знойдзены."
     except Exception as e:
         return f"Нечаканая памылка: {e}"
 
-
 def fix_subtitles_format(transcript: str) -> str:
     """Выпраўляе фармат часу ў субцітрах."""
     prompt_fix = (
@@ -60,14 +46,12 @@ def fix_subtitles_format(transcript: str) -> str:
         f"{transcript}"
     )
     try:
-        response_fix = client_general.generate_content(prompt_fix)
-        response_fix.resolve()
-        return response_fix.text.strip()
+        response_fix = genai.generate_text(model=MODEL_NAME, prompt=prompt_fix)
+        return response_fix.result.strip()
     except Exception as e:
         print(f"Памылка пры выпраўленні субцітраў: {e}")  # Лагіраванне памылкі
         return transcript
 
-
 def create_srt(transcript: str, filename: str = "subtitles.srt") -> tuple[str, str]:
     """Стварае SRT-файл з транскрыпцыі."""
     try:
@@ -77,7 +61,6 @@ def create_srt(transcript: str, filename: str = "subtitles.srt") -> tuple[str, s
     except Exception as e:
         return f"Памылка пры запісе SRT-файла: {e}", ""
 
-
 def process_audio(audio_path: str) -> tuple[str, str]:
     """Апрацоўвае аўдыёфайл: транскрыбуе і стварае SRT."""
     transcript = transcribe_audio(audio_path)
@@ -88,7 +71,6 @@ def process_audio(audio_path: str) -> tuple[str, str]:
     text, srt_file = create_srt(fixed_transcript)
     return text, srt_file
 
-
 def extract_audio_from_video(video_file: str) -> tuple[str, str]:
     """Выдзяляе аўдыёдарожку з відэафайла."""
     try:
@@ -99,7 +81,6 @@ def extract_audio_from_video(video_file: str) -> tuple[str, str]:
     except Exception as e:
         return "", f"Памылка пры выдзяленні аўдыё з відэафайла: {e}"
 
-
 def process_video(video_path: str) -> tuple[str, str]:
     """Апрацоўвае відэафайл: выдзяляе аўдыё, транскрыбуе і стварае SRT."""
     audio_path, error = extract_audio_from_video(video_path)
@@ -107,7 +88,6 @@ def process_video(video_path: str) -> tuple[str, str]:
         return error, ""
     return process_audio(audio_path)
 
-
 def process_file(audio_path: str | None, video_path: str | None) -> tuple[str, str]:
     """Апрацоўвае файл (аўдыё або відэа)."""
     if audio_path:
@@ -117,17 +97,14 @@ def process_file(audio_path: str | None, video_path: str | None) -> tuple[str, s
     else:
         return "Няма файла для апрацоўкі.", ""
 
-
 def update_on_audio_change(audio_path: str | None) -> gr.update:
     """Абнаўляе інтэрфейс пры змене аўдыёфайла."""
     return gr.update(value=None, interactive=not bool(audio_path))
 
-
 def update_on_video_change(video_path: str | None) -> gr.update:
     """Абнаўляе інтэрфейс пры змене відэафайла."""
     return gr.update(value=None, interactive=not bool(video_path))
 
-
 def translate_transcript(
     transcript: str, target_language: str
 ) -> tuple[str, str]:
@@ -137,16 +114,13 @@ def translate_transcript(
         "Астатняе пакінь як ёсць.\nТэкст:\n{transcript}"
     )
     try:
-        response = client_general.generate_content(prompt_text)
-        response.resolve()
-        translated = response.text.strip()
+        response = genai.generate_text(model=MODEL_NAME, prompt=prompt_text)
+        translated = response.result.strip()
         translated_srt_filename = "translated_subtitles.srt"
         return create_srt(translated, translated_srt_filename)  # Перавыкарыстанне create_srt
     except Exception as e:
         return f"Памылка пры перакладзе: {e}", ""
 
-
-
 with gr.Blocks() as demo:
     gr.Markdown("# Транскрыпцыя аўдыя для беларускай мовы")
     gr.Markdown(
@@ -161,7 +135,7 @@ with gr.Blocks() as demo:
         audio_input = gr.Audio(type="filepath", label="Аўдыёфайл")
         video_input = gr.Video(label="Відэафайл")
 
-    #  Больш кампактны спосаб абнаўлення інтэрактыўнасці
+    # Больш кампактны спосаб абнаўлення інтэрактыўнасці
     audio_input.change(
         fn=update_on_audio_change, inputs=audio_input, outputs=video_input
     )
@@ -192,4 +166,4 @@ with gr.Blocks() as demo:
         outputs=[translation_output, translation_file_output],
     )
 
-demo.launch()
+demo.launch()