Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,6 @@ from faster_whisper import WhisperModel
|
|
5 |
from moviepy.video.io.VideoFileClip import VideoFileClip
|
6 |
import logging
|
7 |
import google.generativeai as genai
|
8 |
-
from google.api_core import exceptions
|
9 |
|
10 |
# Suppress moviepy logs
|
11 |
logging.getLogger("moviepy").setLevel(logging.ERROR)
|
@@ -23,7 +22,7 @@ generation_config = {
|
|
23 |
}
|
24 |
|
25 |
model = genai.GenerativeModel(
|
26 |
-
model_name="gemini-
|
27 |
generation_config=generation_config,
|
28 |
)
|
29 |
|
@@ -98,14 +97,9 @@ def format_timestamp(seconds):
|
|
98 |
|
99 |
def translate_text(text, target_language):
|
100 |
"""Translate text using Gemini."""
|
101 |
-
|
102 |
-
|
103 |
-
|
104 |
-
return response.text
|
105 |
-
except exceptions.ResourceExhausted:
|
106 |
-
return f"Translation failed: API quota exceeded. Please try again later."
|
107 |
-
except Exception as e:
|
108 |
-
return f"Translation failed: {str(e)}"
|
109 |
|
110 |
def translate_srt(srt_text, target_language):
|
111 |
"""Translate an SRT file while preserving timestamps."""
|
|
|
5 |
from moviepy.video.io.VideoFileClip import VideoFileClip
|
6 |
import logging
|
7 |
import google.generativeai as genai
|
|
|
8 |
|
9 |
# Suppress moviepy logs
|
10 |
logging.getLogger("moviepy").setLevel(logging.ERROR)
|
|
|
22 |
}
|
23 |
|
24 |
model = genai.GenerativeModel(
|
25 |
+
model_name="gemini-exp-1206", # Updated model name
|
26 |
generation_config=generation_config,
|
27 |
)
|
28 |
|
|
|
97 |
|
98 |
def translate_text(text, target_language):
|
99 |
"""Translate text using Gemini."""
|
100 |
+
prompt = f"Translate the following text into {target_language}:\n\n{text}"
|
101 |
+
response = model.generate_content(prompt)
|
102 |
+
return response.text
|
|
|
|
|
|
|
|
|
|
|
103 |
|
104 |
def translate_srt(srt_text, target_language):
|
105 |
"""Translate an SRT file while preserving timestamps."""
|