Update app.py

app.py

@@ -5,7 +5,8 @@ import mimetypes
 from pydub import AudioSegment
 
 GEMINI_API_KEY = os.getenv("gemini")
-MODEL_NAME = os.getenv("modTH")
+MODEL_NAME_TH = os.getenv("modTH")
+MODEL_NAME = os.getenv("mod")
 
 genai.configure(api_key=GEMINI_API_KEY)
 
@@ -17,25 +18,26 @@ def transcribe_audio(audio_file):
         with open(audio_file, "rb") as f:
             audio_data = f.read()
         prompt_text = (
-            """зрабі транскрыпцыю аўдыяфайла для стварэння субтытрыў з часовымі меткамі srt. Стварай субтытры карацей за 70 сімвалаў. Прыклад фармату часу  00:00:01,589. Пасля кожнага субтытра пакідай пусты радок.
-Не пішы нічога, акрамя транскрыпцыі ў зададзеным фармаце! вось прыклад фармату:
+            """The user wants me to transcribe the audio into subtitles in SRT format, with a maximum of three words per one subtitle.
+I need to listen to the audio and create subtitles with timestamps. Check time format hours:minutes:seconds,milliseconds (00:00:00,000) after creating all.  Example SRT format:
 1
-00:00:15,670 --> 00:00:19,030
+00:00:01,670 --> 00:00:02,030
 За мяжою, за мяжою,
+
 2
-00:00:19,270 --> 00:00:22,850
+00:00:02,270 --> 00:00:03,850
 ні сваё і не чужое.
+
 3
-00:00:23,240 --> 00:00:29,760
+00:00:04,240 --> 00:00:05,760
 Хоць спявай ты, хоць ты грай,
-а навокал іншы край.
+
 4
-00:00:32,770 --> 00:00:39,250
-Пакінуў я край родны,
-спадзяюся, што не незваротна.
+00:00:05,770 --> 00:00:06,250
+а навокал іншы край.
             """
         )
-        model = genai.GenerativeModel(MODEL_NAME)
+        model = genai.GenerativeModel(MODEL_NAME_TH)
         response = model.generate_content(
             [prompt_text, {"mime_type": mime_type, "data": audio_data}]
         )