Update app.py

app.py

@@ -3,29 +3,48 @@ import tempfile
 import os
 import cv2
 import numpy as np
+import subprocess
 
+# Ensure FFmpeg is installed
+os.system("apt-get update && apt-get install -y ffmpeg")
+
+# Function to remove watermark from frames
 def remove_watermark(frame):
-    # Convert to grayscale
     gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
-    # Thresholding to create a mask
     _, mask = cv2.threshold(gray, 200, 255, cv2.THRESH_BINARY)
-    # Inpaint to remove watermark
     result = cv2.inpaint(frame, mask, 3, cv2.INPAINT_TELEA)
     return result
 
+# Function to enhance resolution to 1080p
 def enhance_resolution(input_path, output_path):
     os.system(f"ffmpeg -i {input_path} -vf scale=1920:1080 {output_path}")
 
-def trim_invidea_ai_branding(input_path, output_path, duration=None):
-    if duration is None:
-        duration = get_video_duration(input_path) - 3  # Trim last 3 seconds
-    os.system(f"ffmpeg -i {input_path} -t {duration} {output_path}")
-
+# Function to get video duration using subprocess (fix for ffmpeg.probe error)
 def get_video_duration(video_path):
-    import ffmpeg
-    probe = ffmpeg.probe(video_path)
-    return float(probe['format']['duration'])
+    try:
+        result = subprocess.run(
+            ["ffprobe", "-v", "error", "-show_entries", "format=duration",
+             "-of", "default=noprint_wrappers=1:nokey=1", video_path], 
+            stdout=subprocess.PIPE, 
+            stderr=subprocess.PIPE, 
+            text=True
+        )
+        return float(result.stdout.strip())  # Convert duration to float (seconds)
+    except Exception as e:
+        print("Error getting video duration:", e)
+        return None
 
+# Function to trim last 3 seconds (removes Invideo AI branding)
+def trim_invidea_ai_branding(input_path, output_path):
+    duration = get_video_duration(input_path)
+    
+    if duration:
+        duration = max(0, duration - 3)  # Trim last 3 seconds
+        os.system(f"ffmpeg -i {input_path} -t {duration} {output_path}")
+    else:
+        os.system(f"cp {input_path} {output_path}")  # Copy file without trimming
+
+# Function to process the video
 def process_video(input_video):
     cap = cv2.VideoCapture(input_video)
     if not cap.isOpened():
@@ -55,6 +74,7 @@ def process_video(input_video):
 
     return final_output.name
 
+# Streamlit UI
 st.title("AI Video Enhancement App 🎥")
 
 uploaded_file = st.file_uploader("Upload a video", type=["mp4", "avi", "mov"])
@@ -71,3 +91,4 @@ if uploaded_file:
         st.video(processed_video_path)
         with open(processed_video_path, "rb") as file:
             st.download_button("Download Processed Video", file, "enhanced_video.mp4", "video/mp4")
+