Update app.py

app.py

@@ -189,17 +189,33 @@ def transcribe_to_srt(audio_file, language="hindi"):
     
     return srt_content, preview, status
 
-def process_video(video_file, language="hindi", progress=gr.Progress()):
+def process_video(video_file, language="hindi", subtitle_position="bottom-center", progress=gr.Progress()):
     """Full video processing with subtitle burning"""
     
     if not video_file:
         return None, None, "Please upload a video"
     
+    # Position mapping: alignment and vertical margin
+    POSITION_MAP = {
+        "top-left": (7, 80),
+        "top-center": (8, 80),
+        "top-right": (9, 80),
+        "middle-left": (4, 0),
+        "middle-center": (5, 0),
+        "middle-right": (6, 0),
+        "bottom-left": (1, 50),
+        "bottom-center": (2, 50),
+        "bottom-right": (3, 50),
+    }
+    
+    alignment, margin = POSITION_MAP.get(subtitle_position, (2, 50))
+    
     output_dir = tempfile.mkdtemp()
     print(f"\n{'='*60}")
     print(f"VIDEO PROCESSING STARTED")
     print(f"{'='*60}")
     print(f"  Output directory: {output_dir}")
+    print(f"  Subtitle position: {subtitle_position} (alignment={alignment}, margin={margin})")
     print(f"  Using: {'faster-whisper' if USE_FASTER_WHISPER else 'transformers pipeline'}")
     print(f"{'='*60}\n")
     
@@ -250,6 +266,34 @@ def process_video(video_file, language="hindi", progress=gr.Progress()):
         ], check=True, capture_output=True)
         print(f"✅ ASS created: {ass_path} ({os.path.getsize(ass_path)} bytes)")
         
+        # Modify ASS file to change subtitle position
+        print(f"\nModifying subtitle position to: {subtitle_position}...")
+        with open(ass_path, 'r', encoding='utf-8') as f:
+            ass_content = f.read()
+        
+        # Replace default style with custom position
+        # Format: Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, 
+        #         Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, 
+        #         Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding
+        
+        # Find and replace the style line
+        import re
+        style_pattern = r'Style: Default,[^,]+,\d+,[^,]+,[^,]+,[^,]+,[^,]+,-?\d+,-?\d+,-?\d+,-?\d+,\d+,\d+,\d+,\d+,\d+,\d+,\d+,\d+,\d+,\d+,\d+,\d+'
+        
+        def replace_style(match):
+            parts = match.group(0).split(',')
+            # Replace alignment (index 17) and MarginV (index 20)
+            parts[17] = str(alignment)
+            parts[20] = str(margin)
+            return ','.join(parts)
+        
+        ass_content = re.sub(style_pattern, replace_style, ass_content)
+        
+        with open(ass_path, 'w', encoding='utf-8') as f:
+            f.write(ass_content)
+        
+        print(f"✅ Subtitle position updated: {subtitle_position}")
+        
         # Step 5: Burn subtitles
         progress(0.9, desc="Burning subtitles...")
         print("\nStep 5: Burning subtitles into video...")
@@ -339,6 +383,15 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Hindi Subtitle Generator") as demo
                         value="hindi",
                         label="Language"
                     )
+                    subtitle_pos = gr.Dropdown(
+                        [
+                            "top-left", "top-center", "top-right",
+                            "middle-left", "middle-center", "middle-right",
+                            "bottom-left", "bottom-center", "bottom-right"
+                        ],
+                        value="bottom-center",
+                        label="Subtitle Position"
+                    )
                     video_btn = gr.Button("Generate Subtitles", variant="primary")
                 
                 with gr.Column():
@@ -348,7 +401,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Hindi Subtitle Generator") as demo
             
             video_btn.click(
                 fn=process_video,
-                inputs=[video_input, video_lang],
+                inputs=[video_input, video_lang, subtitle_pos],
                 outputs=[srt_output, video_output, status_output]
             )