Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import tempfile
 import time
 import mimetypes
+import subprocess
 from pathlib import Path
 
 # --- Get API key from environment variable or user input ---
@@ -20,6 +21,21 @@ class VideoProcessor:
         genai.configure(api_key=api_key)
         self.model = genai.GenerativeModel("gemini-2.0-flash")
 
+    def reduce_resolution(self, input_path, output_path, target_height=480):
+        """Reduce video resolution to speed up processing."""
+        try:
+            command = [
+                'ffmpeg', '-i', input_path,
+                '-vf', f'scale=-2:{target_height}',
+                '-c:a', 'copy',
+                '-y', output_path
+            ]
+            subprocess.run(command, check=True, capture_output=True)
+            return output_path
+        except subprocess.CalledProcessError:
+            # If ffmpeg fails, return original path
+            return input_path
+
     def upload_video(self, video_path, display_name="uploaded_video"):
         try:
             return genai.upload_file(path=video_path, display_name=display_name)
@@ -132,6 +148,7 @@ def main():
             # Process new video if different from current
             if st.session_state.video_name != uploaded_file.name:
                 tmp_path = None
+                reduced_path = None
                 try:
                     # Initialize processor if needed
                     if not st.session_state.video_processor:
@@ -149,9 +166,14 @@ def main():
                     with buffering_container.container():
                         show_buffering_animation()
                     
+                    # Reduce video resolution
+                    status_text.markdown("**Reducing video resolution...**")
+                    reduced_path = tmp_path.replace(Path(tmp_path).suffix, f"_reduced{Path(tmp_path).suffix}")
+                    video_to_upload = st.session_state.video_processor.reduce_resolution(tmp_path, reduced_path)
+                    
                     # Upload video
                     status_text.markdown("**Uploading video...**")
-                    video_file = st.session_state.video_processor.upload_video(tmp_path, uploaded_file.name)
+                    video_file = st.session_state.video_processor.upload_video(video_to_upload, uploaded_file.name)
                     
                     # Process video with status updates
                     processed_file = st.session_state.video_processor.wait_for_processing(video_file, status_text)
@@ -174,9 +196,11 @@ def main():
                     st.session_state.video_name = None
                     
                 finally:
-                    # Clean up temporary file
+                    # Clean up temporary files
                     if tmp_path and os.path.exists(tmp_path):
                         os.unlink(tmp_path)
+                    if reduced_path and os.path.exists(reduced_path):
+                        os.unlink(reduced_path)
 
             # Display video player
             st.video(uploaded_file.getvalue())