Update app.py

app.py

@@ -22,7 +22,7 @@ generation_config = {
 }
 
 model = genai.GenerativeModel(
-    model_name="gemini-exp-1206",  # Updated model name
+    model_name="gemini-2.0-flash-exp",
     generation_config=generation_config,
 )
 
@@ -95,25 +95,14 @@ def format_timestamp(seconds):
     milliseconds = int((seconds - int(seconds)) * 1000)
     return f"{hours:02}:{minutes:02}:{int(seconds):02},{milliseconds:03}"
 
-def translate_text(text, target_language):
-    """Translate text using Gemini."""
-    prompt = f"Translate the following text into {target_language}:\n\n{text}"
-    response = model.generate_content(prompt)
-    return response.text
-
 def translate_srt(srt_text, target_language):
     """Translate an SRT file while preserving timestamps."""
-    translated_srt = ""
-    for line in srt_text.split("\n"):
-        if " --> " in line:  # Timestamp line
-            translated_srt += line + "\n"
-        elif line.strip().isdigit():  # Subtitle index line
-            translated_srt += line + "\n"
-        elif line.strip():  # Text line
-            translated_srt += translate_text(line, target_language) + "\n"
-        else:  # Empty line
-            translated_srt += "\n"
-    return translated_srt
+    # Magic prompt for Gemini
+    prompt = f"Translate the following SRT subtitles into {target_language}. Preserve the SRT format (timestamps and structure). Translate only the text after the timestamp. Do not add explanations or extra text.\n\n{srt_text}"
+    
+    # Send the prompt to Gemini
+    response = model.generate_content(prompt)
+    return response.text
 
 def process_video(video_file, language="Auto Detect", translate_to=None):
     """Process a video file to generate and translate subtitles."""