Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
-import google.genai as genai
+from google import genai
+from google.genai import types
 import mimetypes
 from pydub import AudioSegment
 
@@ -10,13 +11,8 @@ MODEL_NAME_TH = os.getenv("modTH")  # Мадэль для транскрыпцы
 MODEL_NAME = os.getenv("mod")  # Мадэль для выпраўлення фармату і перакладу
 PROMPT_TRANSCRIBE = os.getenv("p")  # Промпт для транскрыпцыі
 
-# Канфігурацыя API ключа
-genai.configure(api_key=GEMINI_API_KEY)
-
-# Стварэнне мадэляў
-model_transcribe = genai.GenerativeModel(MODEL_NAME_TH)
-model_general = genai.GenerativeModel(MODEL_NAME)
-
+# Стварэнне кліента
+client = genai.Client(api_key=GEMINI_API_KEY)
 
 def transcribe_audio(audio_file: str) -> str:
     """Транскрыбуе аўдыяфайл з дапамогай Google GenAI."""
@@ -31,14 +27,20 @@ def transcribe_audio(audio_file: str) -> str:
         with open(audio_file, "rb") as f:
             audio_data = f.read()
 
-        # Стварэнне кантэнта для перадачы ў API
-        content = [
+        # Стварэнне contents для перадачы ў API
+        contents = [
             {"text": PROMPT_TRANSCRIBE},
             {"mime_type": mime_type, "data": audio_data},
         ]
 
         # Выклік API
-        response = model_transcribe.generate_content(content)
+        response = client.models.generate_content(
+            model=MODEL_NAME_TH,
+            contents=contents,
+            config=types.GenerateContentConfig(
+                temperature=0.2
+            )
+        )
         
         # Атрыманне адказу
         return response.text.strip()
@@ -57,7 +59,13 @@ def fix_subtitles_format(transcript: str) -> str:
         f"{transcript}"
     )
     try:
-        response_fix = model_general.generate_content(prompt_fix)
+        response_fix = client.models.generate_content(
+            model=MODEL_NAME,
+            contents=[{"text": prompt_fix}],
+            config=types.GenerateContentConfig(
+                temperature=0.2
+            )
+        )
         return response_fix.text.strip()
     except Exception as e:
         print(f"Памылка пры выпраўленні субцітраў: {e}")
@@ -131,7 +139,13 @@ def translate_transcript(transcript: str, target_language: str) -> tuple[str, st
         "Астатняе пакінь як ёсць.\nТэкст:\n{transcript}"
     )
     try:
-        response = model_general.generate_content(prompt_text)
+        response = client.models.generate_content(
+            model=MODEL_NAME,
+            contents=[{"text": prompt_text}],
+            config=types.GenerateContentConfig(
+                temperature=0.2
+            )
+        )
         translated = response.text.strip()
         translated_srt_filename = "translated_subtitles.srt"
         return create_srt(translated, translated_srt_filename)