Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,72 +1,77 @@
 import streamlit as st
 import os
 
-# Set your video directory here
+# Video folder path
 VIDEO_FOLDER = "./src/synthda_falling_realreal/"
 
+# Streamlit page setup
 st.set_page_config(layout="wide")
+st.title("AutoSynthDa Pose Interpolation Viewer")
 
-# Title and instructions
 st.markdown("""
-<h1 style='text-align: center;'>AutoSynthDa Pose Interpolation Viewer</h1>
-<p style='text-align: center; font-size: 18px;'>
-AutoSynthDa blends two input motion videos to generate <strong>kinematically coherent, synthetic action sequences</strong>.
-</p>
-<p style='text-align: center; font-size: 16px;'>
-Use the slider to explore how the motion transitions from Input Video 1 (left) to Input Video 2 (right).
-</p>
-<p style='text-align: center;'>
-<a href='https://github.com/nvidia/synthda' target='_blank'>View the code on GitHub</a>
-</p>
-""", unsafe_allow_html=True)
+### AutoSynthDa Interpolation Viewer
 
-# Slider starts at 0.5 by default
-weight = st.slider("Interpolation Weight", 0.1, 0.9, step=0.1, value=0.5)
+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)
+""")
 
-# File paths
+# Slider explanation
+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
+)
+
+# Slider (starts at 0.5)
+weight = st.slider("Interpolation Weight", 0.0, 1.0, 0.5, step=0.1)
+
+# Filenames
 filename_interp = f"videos_generated_{weight:.1f}.mp4"
 filename_input1 = "videos_generated_0.0.mp4"
 filename_input2 = "videos_generated_1.0.mp4"
 
+# Full paths
 video_interp = os.path.join(VIDEO_FOLDER, filename_interp)
 video_input1 = os.path.join(VIDEO_FOLDER, filename_input1)
 video_input2 = os.path.join(VIDEO_FOLDER, filename_input2)
 
+# File checks
 exists_interp = os.path.exists(video_interp)
 exists_1 = os.path.exists(video_input1)
 exists_2 = os.path.exists(video_input2)
 
-# Interpolation info text
+# Interpolation status
 if weight == 0.0:
-    interp_text = "Displaying Input Video 1 (no interpolation)"
+    st.success("Showing Input Video 1 (no interpolation)")
 elif weight == 1.0:
-    interp_text = "Displaying Input Video 2 (no interpolation)"
+    st.success("Showing Input Video 2 (no interpolation)")
 else:
     w2 = round(1.0 - weight, 1)
-    interp_text = f"Interpolated motion: {weight:.1f} from Input Video 1 + {w2:.1f} from Input Video 2"
+    st.info(
+        f"Generated motion: {weight:.1f} from Input Video 1 + {w2:.1f} from Input Video 2"
+    )
 
-st.markdown(f"<p style='text-align: center; font-size: 16px;'><strong>{interp_text}</strong></p>", unsafe_allow_html=True)
-
-# Layout with 3 video panels
+# Layout: 3 columns for videos
 col1, col2, col3 = st.columns(3)
 
-with col1:
-    st.markdown("**Input Video 1**")
-    if exists_1:
-        st.video(video_input1)
-    else:
-        st.error("Video 1 not found")
-
-with col2:
-    st.markdown("**Interpolated Video**")
-    if exists_interp:
-        st.video(video_interp)
-    else:
-        st.error("Interpolated video not found")
+# 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")
 
-with col3:
-    st.markdown("**Input Video 2**")
-    if exists_2:
-        st.video(video_input2)
-    else:
-        st.error("Video 2 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)