Update app.py

app.py

@@ -5,6 +5,7 @@ from faster_whisper import WhisperModel
 from moviepy.video.io.VideoFileClip import VideoFileClip
 import logging
 import google.generativeai as genai
+from google.api_core import exceptions
 
 # Suppress moviepy logs
 logging.getLogger("moviepy").setLevel(logging.ERROR)
@@ -14,8 +15,8 @@ genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # Create the Gemini model
 generation_config = {
-    "temperature": 0.7,  # Lower temperature for more focused translations
-    "top_p": 0.9,
+    "temperature": 1,
+    "top_p": 0.95,
     "top_k": 40,
     "max_output_tokens": 8192,
     "response_mime_type": "text/plain",
@@ -97,10 +98,14 @@ def format_timestamp(seconds):
 
 def translate_text(text, target_language):
     """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)
-    return response.text
+    try:
+        prompt = f"Translate the following text into {target_language}:\n\n{text}"
+        response = model.generate_content(prompt)
+        return response.text
+    except exceptions.ResourceExhausted:
+        return f"Translation failed: API quota exceeded. Please try again later."
+    except Exception as e:
+        return f"Translation failed: {str(e)}"
 
 def translate_srt(srt_text, target_language):
     """Translate an SRT file while preserving timestamps."""