Update app.py

app.py

@@ -1,14 +1,13 @@
 # Import required libraries
 import streamlit as st
 import cv2
-import numpy as np
 import tempfile
 import base64
 
 # Streamlit app title
-st.title("Video Interpolation App 🎥")
+st.title("Video Concatenator App 🎥")
 
-# Function to read video and return frames
+# Function to read video and return frames and properties
 def read_video(video_path):
     cap = cv2.VideoCapture(video_path)
     frames = []
@@ -17,8 +16,10 @@ def read_video(video_path):
         if not ret:
             break
         frames.append(frame)
+    fps = int(cap.get(cv2.CAP_PROP_FPS))
+    size = (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)))
     cap.release()
-    return frames
+    return frames, fps, size
 
 # Function to convert video file to base64
 def get_video_base64(video_path):
@@ -27,13 +28,9 @@ def get_video_base64(video_path):
     return video_base64
 
 # Function to save frames to video
-def save_frames_to_video(frames, output_video_path):
-    if not frames:
-        return
-
-    height, width, _ = frames[0].shape
+def save_frames_to_video(frames, output_video_path, fps, size):
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    out = cv2.VideoWriter(output_video_path, fourcc, 30.0, (width, height))
+    out = cv2.VideoWriter(output_video_path, fourcc, fps, size)
 
     for frame in frames:
         out.write(frame)
@@ -48,7 +45,7 @@ uploaded_files = st.file_uploader("Upload Video Files 📤", type=["mp4"], accep
 if uploaded_files:
     video_paths = []
     for file in uploaded_files:
-        tfile = tempfile.NamedTemporaryFile(delete=False)
+        tfile = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
         tfile.write(file.read())
         video_paths.append(tfile.name)
     
@@ -56,15 +53,17 @@ if uploaded_files:
 
     # Read videos and get frames
     all_frames = []
+    fps = 0
+    size = (0, 0)
     for path in video_paths:
-        frames = read_video(path)
+        frames, fps, size = read_video(path)
         all_frames.extend(frames)
 
     # Placeholder for future functionality
-    st.write("Interpolation will be performed here. ⏳")
+    st.write("Additional processing can be performed here. ⏳")
 
     # Save frames to video
-    save_frames_to_video(all_frames, output_video_name)
+    save_frames_to_video(all_frames, output_video_name, fps, size)
 
     # Provide download link for the video
     video_base64 = get_video_base64(output_video_name)