Update app.py

app.py

@@ -5,6 +5,8 @@ from faster_whisper import WhisperModel
 from moviepy.video.io.VideoFileClip import VideoFileClip
 import logging
 import google.generativeai as genai
+import time
+from google.api_core import retry
 
 # Suppress moviepy logs
 logging.getLogger("moviepy").setLevel(logging.ERROR)
@@ -14,8 +16,8 @@ genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # Create the Gemini model
 generation_config = {
-    "temperature": 0.7,  # Lower temperature for more accurate translations
-    "top_p": 0.9,
+    "temperature": 1,
+    "top_p": 0.95,
     "top_k": 40,
     "max_output_tokens": 8192,
     "response_mime_type": "text/plain",
@@ -95,11 +97,12 @@ def format_timestamp(seconds):
     milliseconds = int((seconds - int(seconds)) * 1000)
     return f"{hours:02}:{minutes:02}:{int(seconds):02},{milliseconds:03}"
 
+@retry.Retry()
 def translate_text(text, target_language):
-    """Translate text using Gemini with a magic prompt."""
-    # Magic prompt for clean translations
-    prompt = f"Translate the following text into {target_language}. Do not add any extra words like 'Here is the translation:' or 'Translation:'. Just provide the translated text:\n\n{text}"
+    """Translate text using Gemini with retry logic."""
+    prompt = f"Translate the following text into {target_language}:\n\n{text}"
     response = model.generate_content(prompt)
+    time.sleep(1)  # Add a delay to avoid hitting the rate limit
     return response.text
 
 def translate_srt(srt_text, target_language):
@@ -132,10 +135,14 @@ def process_video(video_file, language="Auto Detect", translate_to=None):
     # Translate subtitles if a target language is provided
     translated_srt_file = None
     if translate_to and translate_to != "None":
-        translated_subtitles = translate_srt(subtitles, translate_to)
-        translated_srt_file = "translated_subtitles.srt"
-        with open(translated_srt_file, "w", encoding="utf-8") as f:
-            f.write(translated_subtitles)
+        try:
+            translated_subtitles = translate_srt(subtitles, translate_to)
+            translated_srt_file = "translated_subtitles.srt"
+            with open(translated_srt_file, "w", encoding="utf-8") as f:
+                f.write(translated_subtitles)
+        except Exception as e:
+            print(f"Translation failed: {e}")
+            translated_srt_file = None
     
     # Clean up extracted audio file
     os.remove(audio_file)