| import os |
| import sys |
|
|
| |
| sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))) |
|
|
| import pysrt |
| import google.generativeai as genai |
| from app.services.translators.gemini_adapter import GeminiAdapter |
| from dotenv import load_dotenv |
|
|
| load_dotenv() |
| adapter = GeminiAdapter() |
|
|
| subs = pysrt.open('app/subtitles/08-52-AM--10-05-2026/nikhil kamath clip_test_hi.srt', encoding='utf-8') |
| lines = [sub.text for sub in subs[:30]] |
|
|
| print(f"Translating {len(lines)} lines") |
|
|
| lang_name = "Hindi" |
| non_empty = [(i, line) for i, line in enumerate(lines) if line.strip()] |
| numbered_block = "\n".join( |
| f"[{idx+1}] {line}" for idx, (_, line) in enumerate(non_empty) |
| ) |
|
|
| system_instruction = ( |
| f"You are an expert translator specializing in {lang_name}. " |
| f"You will receive numbered English subtitle lines from a conversation. " |
| f"Translate ALL {len(non_empty)} lines to natural, colloquial {lang_name}. " |
| f"Use the surrounding lines as context to pick the right tone, pronouns, and expressions. " |
| f"Return ONLY the translations in the exact same numbered format: [1] translation, [2] translation, etc. " |
| f"Do NOT add explanations, notes, or extra text. " |
| f"You MUST translate exactly {len(non_empty)} lines. Do not stop until you have output all of them." |
| ) |
|
|
| user_prompt = f"Here are the lines:\n{numbered_block}" |
| model = genai.GenerativeModel("gemini-2.5-flash", system_instruction=system_instruction) |
|
|
| response = model.generate_content( |
| user_prompt, |
| generation_config=genai.types.GenerationConfig( |
| temperature=0.3, |
| ) |
| ) |
|
|
| raw_output = response.text.strip() |
| print("RAW OUTPUT:") |
| print("---") |
| print(raw_output) |
| print("---") |
| print("Finish Reason:", response.candidates[0].finish_reason) |
|
|
| translated_dict = adapter._parse_numbered_block(raw_output) |
| print(f"Parsed {len(translated_dict)} lines.") |
|
|
