Update app.py

app.py

@@ -55,6 +55,25 @@ def transcribe_audio(audio_file):
     except Exception as e:
         return f"Нечаканая памылка: {str(e)}"
 
+def fix_subtitles_format(transcript):
+    """
+    Дадатковы запыт да мадэлі, які выпраўляе фармат часоў у субцітрах.
+    """
+    try:
+        prompt_fix = (
+            f"выправі часовы фармат у субцітрах на правільны, вось прыклад 00:00:01,589\n"
+            f"напішы толькі выпраўленныя субцітры {transcript}"
+        )
+        model = genai.GenerativeModel(MODEL_NAME)
+        response_fix = model.generate_content(prompt_fix)
+        if response_fix.text:
+            fixed_transcript = response_fix.text.strip()
+        else:
+            fixed_transcript = transcript
+        return fixed_transcript
+    except Exception as e:
+        return transcript
+
 def create_srt(transcript, filename="subtitles.srt"):
     try:
         with open(filename, "w", encoding="utf-8") as f:
@@ -67,7 +86,9 @@ def process_audio(audio):
     transcript = transcribe_audio(audio)
     if transcript.startswith("Памылка"):
         return transcript, None
-    text, srt_file = create_srt(transcript)
+    # Дадаем другі запыт для выпраўлення фармату часоў у субцітрах
+    fixed_transcript = fix_subtitles_format(transcript)
+    text, srt_file = create_srt(fixed_transcript)
     return text, srt_file
 
 def extract_audio_from_video(video_file):
@@ -106,13 +127,9 @@ def update_on_video_change(video):
         return gr.update(interactive=True)
 
 def translate_transcript(transcript, target_language):
-    """
-    Функцыя адпраўляе тэкст транскрыпцыі з промптам для перакладу.
-    Атрыманая перакладзеная версія запісваецца ў новы SRT-файл.
-    """
     try:
         prompt_text = (
-            f"перакладзі толькі тэксты субцітраў на {target_language} мову. Выправі фармат часовых метак дзе неабходна (правільны фармат HH:MM:SS,mmm  --> HH:MM:SS,mmm). Астатняя пакінь як ёсць."
+            f"перакладзі толькі тэксты субцітраў на {target_language} мову. Астатняя пакінь як ёсць."
             f"Тэкст:\n{transcript}"
         )
         model = genai.GenerativeModel(MODEL_NAME)
@@ -149,7 +166,7 @@ with gr.Blocks() as demo:
     file_output = gr.File(label="SRT-файл")
     btn.click(fn=process_file, inputs=[audio_input, video_input], outputs=[transcript_output, file_output])
     
-    gr.Markdown("## Пераклад субтытраў")
+    gr.Markdown("## Пераклад субцітраў")
     with gr.Row():
         language_dropdown = gr.Dropdown(
             choices=["English", "Руcкая", "Польская", "Літоўская", "Нямецкая"],