Update app.py

app.py

@@ -5,8 +5,6 @@ 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)
@@ -16,8 +14,8 @@ genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # Create the Gemini model
 generation_config = {
-    "temperature": 1,
-    "top_p": 0.95,
+    "temperature": 0.7,  # Lower temperature for more focused translations
+    "top_p": 0.9,
     "top_k": 40,
     "max_output_tokens": 8192,
     "response_mime_type": "text/plain",
@@ -97,12 +95,11 @@ 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 retry logic."""
-    prompt = f"Translate the following text into {target_language}:\n\n{text}"
+    """Translate text using Gemini."""
+    # Magic prompt to ensure concise and accurate translations
+    prompt = f"Translate the following text into {target_language}. Only provide the translated text, no explanations or additional text:\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):
@@ -135,14 +132,10 @@ 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":
-        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
+        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)
     
     # Clean up extracted audio file
     os.remove(audio_file)