Update app.py

app.py

@@ -30,8 +30,22 @@ def get_video_base64(video_path):
         video_base64 = base64.b64encode(file.read()).decode()
     return video_base64
 
+# Function to save frames to video
+def save_frames_to_video(frames, output_video_path):
+    height, width, _ = frames[0].shape
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(output_video_path, fourcc, 20.0, (width, height))
+
+    for frame in frames:
+        out.write(frame)
+
+    out.release()
+
+# Textbox for user to specify output video name
+output_video_name = st.text_input("Enter the name for the output video:", "Output_Video.mp4")
+
 # Upload videos
-uploaded_files = st.file_uploader("Upload Video Files", type=["mp4"], accept_multiple_files=True)
+uploaded_files = st.file_uploader("Upload Video Files 📤", type=["mp4"], accept_multiple_files=True)
 
 if uploaded_files:
     video_paths = []
@@ -40,25 +54,22 @@ if uploaded_files:
         tfile.write(file.read())
         video_paths.append(tfile.name)
     
-    st.write("Uploaded video files:", video_paths)
+    st.write("Uploaded video files: 📂", video_paths)
 
     # Read videos and get frames
     all_frames = []
     for path in video_paths:
         frames = read_video(path)
-        all_frames.append(frames)
+        all_frames.extend(frames)
 
     # Placeholder for future functionality
-    st.write("Interpolation will be performed here.")
+    st.write("Interpolation will be performed here. ⏳")
+
+    # Save frames to video
+    save_frames_to_video(all_frames, output_video_name)
 
-    # Compile frames into new video (placeholder)
-    st.write("The new video will be compiled here.")
-    
-    # Assume output_video_path contains the path to the compiled video
-    output_video_path = "output_video.mp4"
-    
     # Provide download link for the video
-    video_base64 = get_video_base64(output_video_path)
+    video_base64 = get_video_base64(output_video_name)
     st.markdown(f'### Download Output Video 📥')
-    href = f'<a href="data:video/mp4;base64,{video_base64}" download="output_video.mp4">Click here to download the video</a>'
+    href = f'<a href="data:video/mp4;base64,{video_base64}" download="{output_video_name}">Click here to download the video</a>'
     st.markdown(href, unsafe_allow_html=True)