Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 import cv2
 import numpy as np
+import tempfile
+import os
 
 def cartoonize_image(image):
     # Convert the image to grayscale
@@ -30,20 +32,21 @@ def main():
     video_file = st.file_uploader("Choose a video file", type=["mp4"])
 
     if video_file is not None:
-        # Convert the video file to bytes
-        video_bytes = video_file.read()
+        # Save the uploaded video to a temporary file
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_video:
+            temp_video.write(video_file.read())
+            temp_video_path = temp_video.name
 
-        # Convert the bytes to a numpy array
-        video_np = np.frombuffer(video_bytes, np.uint8)
+        # Process and display the cartoonized video
+        cartoon_video_path = cartoonize_video(temp_video_path)
 
-        # Decode the numpy array to a video file
-        video_path = "temp_video.mp4"
-        with open(video_path, "wb") as f:
-            f.write(video_bytes)
+        # Display the cartoonized video
+        with open(cartoon_video_path, "rb") as f:
+            st.video(f.read(), format="video/mp4", start_time=0)
 
-        # Process and display the cartoonized video
-        cartoon_video = cartoonize_video(video_path)
-        st.video(cartoon_video, format="video/mp4", start_time=0)
+        # Clean up the temporary files
+        os.remove(temp_video_path)
+        os.remove(cartoon_video_path)
 
 def cartoonize_video(video_path):
     cap = cv2.VideoCapture(video_path)
@@ -53,8 +56,8 @@ def cartoonize_video(video_path):
     width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
 
-    # Create VideoWriter object to save the cartoonized video
-    cartoon_video_path = "cartoon_video.mp4"
+    # Create a temporary file to save the cartoonized video
+    cartoon_video_path = tempfile.mktemp(suffix=".mp4")
     cartoon_video = cv2.VideoWriter(cartoon_video_path, cv2.VideoWriter_fourcc(*"mp4v"), fps, (width, height))
 
     while True:
@@ -66,20 +69,13 @@ def cartoonize_video(video_path):
         # Cartoonize the frame
         cartoon_frame = cartoonize_image(frame)
 
-        # Display the cartoonized frame
-        st.image(cartoon_frame, channels="BGR", use_column_width=True)
-
         # Write the cartoonized frame to the VideoWriter
         cartoon_video.write(cartoon_frame)
 
     cap.release()
     cartoon_video.release()
 
-    # Read the cartoonized video and return the bytes
-    with open(cartoon_video_path, "rb") as f:
-        cartoon_video_bytes = f.read()
-
-    return cartoon_video_bytes
+    return cartoon_video_path
 
 if __name__ == "__main__":
     main()