Update app.py

app.py

@@ -66,12 +66,17 @@ def get_video_source(source_type, source_path=None):
     if source_type == "Webcam":
         return cv2.VideoCapture(0)
     elif source_type == "Video File" and source_path:
-        # Create a temporary file
+        # Create a temporary file with a specific extension
         temp_dir = tempfile.gettempdir()
         temp_path = os.path.join(temp_dir, 'temp_video.mp4')
         with open(temp_path, 'wb') as f:
             f.write(source_path.getvalue())
-        return cv2.VideoCapture(temp_path)
+        
+        cap = cv2.VideoCapture(temp_path)
+        if not cap.isOpened():
+            st.error("Error: Could not open video file. Please ensure it's a supported format (MP4 with H.264 encoding recommended)")
+            return None
+        return cap
     elif source_type == "RTSP Stream" and source_path:
         return cv2.VideoCapture(source_path)
     return None
@@ -131,18 +136,10 @@ def main():
         
         source_path = None
         uploaded_file = None
-        if source_type == "Video File" and source_path:
-            # Create a temporary file with a specific extension
-            temp_dir = tempfile.gettempdir()
-            temp_path = os.path.join(temp_dir, 'temp_video.mp4')
-            with open(temp_path, 'wb') as f:
-                f.write(source_path.getvalue())
-            
-            cap = cv2.VideoCapture(temp_path)
-            if not cap.isOpened():
-                st.error("Error: Could not open video file. Please ensure it's a supported format (MP4 with H.264 encoding recommended)")
-                return None
-            return cap
+        if source_type == "Video File":
+            uploaded_file = st.file_uploader("Choose a video file", type=['mp4', 'avi', 'mov'])
+            if uploaded_file:
+                source_path = BytesIO(uploaded_file.getvalue())
         elif source_type == "RTSP Stream":
             source_path = st.text_input("Enter RTSP URL", placeholder="rtsp://your-camera-url")