Create app.py

app.py

@@ -0,0 +1,112 @@
+import gradio as gr
+from transformers import pipeline
+import re
+
+# Translation model
+translator = pipeline(
+    "translation",
+    model="google/translategemma-2b-it"
+)
+
+# -----------------------------
+# SRT PARSER
+# -----------------------------
+def parse_srt(srt_text):
+    pattern = re.compile(
+        r"(\d+)\s*\n"
+        r"(\d{2}:\d{2}:\d{2},\d{3}\s-->\s\d{2}:\d{2}:\d{2},\d{3})\s*\n"
+        r"(.*?)(?=\n\d+\n|\Z)",
+        re.DOTALL
+    )
+
+    matches = pattern.findall(srt_text.strip())
+
+    subtitles = []
+
+    for idx, timestamp, text in matches:
+        clean_text = text.replace("\n", " ").strip()
+
+        subtitles.append({
+            "index": idx,
+            "timestamp": timestamp,
+            "text": clean_text
+        })
+
+    return subtitles
+
+
+# -----------------------------
+# SHORTEN HINDI TEXT
+# -----------------------------
+def shorten_text(text, max_len):
+    words = text.split()
+
+    while len(text) > max_len and len(words) > 1:
+        words.pop(-1)
+        text = " ".join(words)
+
+    return text
+
+
+# -----------------------------
+# TRANSLATE SRT
+# -----------------------------
+def translate_srt(srt_text):
+
+    subtitles = parse_srt(srt_text)
+
+    output = []
+
+    for sub in subtitles:
+
+        english = sub["text"]
+
+        english_len = len(english)
+
+        # Max 130%
+        max_hindi_len = int(english_len * 1.3)
+
+        try:
+            result = translator(
+                english,
+                max_new_tokens=128
+            )
+
+            hindi = result[0]["translation_text"].strip()
+
+        except Exception as e:
+            hindi = english
+
+        # Length control
+        if len(hindi) > max_hindi_len:
+            hindi = shorten_text(hindi, max_hindi_len)
+
+        block = (
+            f'{sub["index"]}\n'
+            f'{sub["timestamp"]}\n'
+            f'{hindi}\n'
+        )
+
+        output.append(block)
+
+    return "\n".join(output)
+
+
+# -----------------------------
+# GRADIO UI
+# -----------------------------
+demo = gr.Interface(
+    fn=translate_srt,
+    inputs=gr.Textbox(
+        lines=20,
+        label="English SRT"
+    ),
+    outputs=gr.Textbox(
+        lines=20,
+        label="Hindi SRT"
+    ),
+    title="English to Hindi SRT Translator",
+    description="Translate English SRT to Hindi with subtitle length control."
+)
+
+demo.launch()