Update app.py

app.py

@@ -14,7 +14,7 @@ def process_video(video_path):
     fps = int(cap.get(cv2.CAP_PROP_FPS))
 
     # Create a video writer to save the output
-    output_path = os.path.join(os.getcwd(),"output.mp4")
+    output_path = os.path.join(os.getcwd(), "output.mp4")
     fourcc = cv2.VideoWriter_fourcc(*"mp4v")
     out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
 
@@ -54,27 +54,32 @@ def process_video(video_path):
 st.title("Fall Detection App")
 st.write("Upload a video or choose from the default videos to detect falls.")
 
-# Video selection
-default_videos = {
-    "Video 1": os.path.join(os.getcwd(),"fall_test_01.mp4"),
-    "Video 2": os.path.join(os.getcwd(),"fall_test_02.mp4"),
-    "Video 3": "video3.mp4",
-}
-
-option = st.selectbox("Choose a video", list(default_videos.keys()))
-uploaded_video = st.file_uploader("Or upload your own video", type=["mp4", "avi", "mov"])
-
-if uploaded_video is not None:
-    video_path = uploaded_video.name
-    with open(video_path, 'wb') as f:
-        f.write(uploaded_video.getbuffer())
-    st.success(f"Uploaded {uploaded_video.name}")
-else:
-    video_path = default_videos[option]
-
-if st.button("Process Video"):
-    output_video = process_video(video_path)
-    st.video(output_video)
-    st.write("Download the processed video:")
-    with open(output_video, "rb") as video_file:
-        st.download_button("Download", video_file, "workouts.mp4")
+# Create two columns
+left_column, right_column = st.columns(2)
+
+# Right column for video selection
+with right_column:
+    default_videos = {
+        "Video 1": os.path.join(os.getcwd(), "fall_test_01.mp4"),
+        "Video 2": os.path.join(os.getcwd(), "fall_test_02.mp4"),
+        "Video 3": "video3.mp4",
+    }
+
+    option = st.selectbox("Choose a video", list(default_videos.keys()))
+    uploaded_video = st.file_uploader("Or upload your own video", type=["mp4", "avi", "mov"])
+
+    if uploaded_video is not None:
+        video_path = uploaded_video.name
+        with open(video_path, 'wb') as f:
+            f.write(uploaded_video.getbuffer())
+        st.success(f"Uploaded {uploaded_video.name}")
+    else:
+        video_path = default_videos[option]
+
+    if st.button("Process Video"):
+        output_video = process_video(video_path)
+        left_column.video(output_video)  # Display video in the left column
+        left_column.write("Download the processed video:")
+        with open(output_video, "rb") as video_file:
+            left_column.download_button("Download", video_file, "output.mp4")
+