Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,22 +1,22 @@
 import streamlit as st
 import os
 
-# Video folder path
+# Video directory
 VIDEO_FOLDER = "./src/synthda_falling_realreal/"
 
-# Streamlit page setup
+# Set page layout
 st.set_page_config(layout="wide")
 st.title("AutoSynthDa Pose Interpolation Viewer")
 
 st.markdown("""
 ### AutoSynthDa Interpolation Viewer
 
-AutoSynthDa blends two input motion videos to **generate kinematically coherent, synthetic action videos**.
+AutoSynthDa blends two input motion videos to **generate kinematically coherent, synthetic action videos**.  
 Use the slider below to explore how the system interpolates motion from one video to another.  
 Source: [github.com/nvidia/synthda](https://github.com/nvidia/synthda)
 """)
 
-# Slider explanation
+# Slider instruction
 st.markdown(
     '<p style="text-align:center;"><strong>Use the slider to control the interpolation between Input Video 1 (left) and Input Video 2 (right).</strong></p>',
     unsafe_allow_html=True
@@ -54,24 +54,24 @@ else:
 # Layout: 3 columns for videos
 col1, col2, col3 = st.columns(3)
 
-# Helper to show video with autoplay
-def render_video_column(col, label, video_path, exists):
-    with col:
-        st.markdown(f"<div style='text-align: center; font-weight: bold;'>{label}</div>", unsafe_allow_html=True)
-        if exists:
-            st.markdown(
-                f"""
-                <video width="100%" autoplay loop muted playsinline>
-                    <source src="{video_path}" type="video/mp4">
-                    Your browser does not support the video tag.
-                </video>
-                """,
-                unsafe_allow_html=True
-            )
-        else:
-            st.error(f"{label} not found")
+# Centered labels and video playback
+with col1:
+    st.markdown("<div style='text-align: center; font-weight: bold;'>Input Video 1</div>", unsafe_allow_html=True)
+    if exists_1:
+        st.video(video_input1)
+    else:
+        st.error("Video 1 not found")
 
-# Show videos
-render_video_column(col1, "Input Video 1", video_input1, exists_1)
-render_video_column(col2, "Interpolated Video", video_interp, exists_interp)
-render_video_column(col3, "Input Video 2", video_input2, exists_2)
+with col2:
+    st.markdown("<div style='text-align: center; font-weight: bold;'>Interpolated Video</div>", unsafe_allow_html=True)
+    if exists_interp:
+        st.video(video_interp)
+    else:
+        st.error("Interpolated video not found")
+
+with col3:
+    st.markdown("<div style='text-align: center; font-weight: bold;'>Input Video 2</div>", unsafe_allow_html=True)
+    if exists_2:
+        st.video(video_input2)
+    else:
+        st.error("Video 2 not found")