Update app.py

app.py

@@ -103,7 +103,7 @@ def prompt_fix_grammar(language="English"):
     """
     prompt = f"""
 -------------- You are a professional subtitle editor for **video dubbing**.
-Fix the grammar, spelling, and awkward phrasing in the following `.srt` subtitle file while preserving timing, meaning, and emotional tone.  
+Fix the grammar, spelling, and awkward phrasing in the following `.srt` subtitle file while preserving timing, meaning, and emotional tone.  
 Do NOT translate — keep everything in {language}.
 
 Output in JSON format exactly like this:
@@ -127,6 +127,40 @@ Output in JSON format exactly like this:
     return prompt
 
 
+def prompt_srt_to_romanized(language="Hindi"):
+  """
+  Generates a prompt for converting a .srt subtitle file
+  from any language to a Romanized (Latin letters) version,
+  preserving timing, meaning, punctuation, and formatting.
+  """
+  prompt = f"""
+-------------- You are a professional subtitle editor tasked with converting subtitles to Romanized text.
+Your task is to convert a `.srt` subtitle file from {language} to **Romanized {language}**, 
+keeping everything exactly the same except using Latin letters for all words.
+
+**Instructions:**
+1. Preserve the original timestamp of each subtitle.
+2. Keep the original meaning, punctuation, and formatting intact.
+3. Convert **only the original subtitle text** to Roman letters, word by word.
+4. Do not add, remove, or change any words.
+5. Output in strict JSON format exactly like this:
+
+```json
+{{
+"subtitle sequence number": {{
+"timestamp": "original timestamp",
+"original subtitle text": "original {language} subtitle line",
+"dubbing": "Romanized, {language} line of original subtitle text"
+}}
+}}
+````
+
+Focus entirely on **accurate Romanization**; do not modify anything else.
+"""
+  return prompt
+
+
+
 import pysrt
 
 def prompt_maker(srt_path, target_language, task="Translation"):
@@ -140,6 +174,8 @@ def prompt_maker(srt_path, target_language, task="Translation"):
             f.write(f"{sub.text}\n\n")
         if task == "Translation":
             f.write(prompt_translation(target_language))
+        if task=="Romanization":
+            f.write(prompt_srt_to_romanized(target_language))
         else:
             f.write(prompt_fix_grammar(target_language))
 
@@ -196,10 +232,10 @@ def ui2():
                 gr.Markdown("### Step 1: Generate Prompt")
                 srt_file = gr.File(label="Upload .srt file generated by Whisper", file_types=[".srt"])
                 task = gr.Dropdown(
-                    ["Translation", "Fix Grammar [English to English for dubbing]"],
-                    label="Select Task",
-                    value="Translation",
-                )
+                      ["Translation","Romanization","Fix Grammar [English to English for dubbing]"],
+                      label="Select Task",
+                      value="Translation",
+                  )
                 language = gr.Dropdown(target_lang_list, label="Select the language you want to translate into", value="English")
                 generate_btn = gr.Button("Generate Prompt")
                 output_prompt = gr.Textbox(