Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,73 +1,59 @@
+import altair as alt
+import numpy as np
+import pandas as pd
 import streamlit as st
 import os
 
 # Set your video directory here
 VIDEO_FOLDER = "./src/synthda_falling_realreal/"
 
-st.set_page_config(layout="centered")
+st.set_page_config(layout="wide")
 st.title("AutoSynthDa Pose Interpolation Viewer")
 
 st.markdown("""
-# AutoSynthDa Interpolation Viewer
+### AutoSynthDa Interpolation Viewer
 
 AutoSynthDa blends two input motion videos to **generate kinematically coherent, synthetic action videos**.
 
-Use the slider to view interpolated motion between two real motion capture inputs.
+Use the slider below to explore how the system interpolates motion from one video to another.
 
-View the code at: [github.com/nvidia/synthda](https://github.com/nvidia/synthda)
+Source: [github.com/nvidia/synthda](https://github.com/nvidia/synthda)
 """)
 
-# Custom styles for centering
-st.markdown("""
-<style>
-.centered-text {
-    text-align: center;
-    font-size: 18px;
-    font-weight: 500;
-    margin-top: 1em;
-    margin-bottom: 1em;
-}
-.slider-label {
-    text-align: center;
-    font-size: 16px;
-    color: gray;
-    margin-bottom: 10px;
-}
-video {
-    border-radius: 8px;
-    box-shadow: 0 0 10px rgba(0,0,0,0.1);
-}
-</style>
-""", unsafe_allow_html=True)
-
-# Slider label
-st.markdown('<p class="slider-label">Slide to interpolate between Input Video 1 (left) and Input Video 2 (right)</p>', unsafe_allow_html=True)
-
-# Slider
+# Add slider explanation
+st.markdown(
+    '<p style="text-align:center;"><strong>Move the slider to interpolate between Input Video 1 (left) and Input Video 2 (right)</strong></p>',
+    unsafe_allow_html=True
+)
+
+# Slider for selecting interpolation weight
 weight = st.slider("Interpolation Weight", 0.0, 1.0, step=0.1)
 
-# Calculate filenames
+# File paths
 filename_interp = f"videos_generated_{weight:.1f}.mp4"
+filename_input1 = "videos_generated_0.0.mp4"
+filename_input2 = "videos_generated_1.0.mp4"
+
 video_interp = os.path.join(VIDEO_FOLDER, filename_interp)
-video_input1 = os.path.join(VIDEO_FOLDER, "videos_generated_0.0.mp4")
-video_input2 = os.path.join(VIDEO_FOLDER, "videos_generated_1.0.mp4")
+video_input1 = os.path.join(VIDEO_FOLDER, filename_input1)
+video_input2 = os.path.join(VIDEO_FOLDER, filename_input2)
 
-# Display interpolation description
+exists_interp = os.path.exists(video_interp)
+exists_1 = os.path.exists(video_input1)
+exists_2 = os.path.exists(video_input2)
+
+# Description of blending
 if weight == 0.0:
-    interp_text = "Showing Input Video 1 (Original)"
+    st.success("Showing Input Video 1 (no interpolation)")
 elif weight == 1.0:
-    interp_text = "Showing Input Video 2 (Original)"
+    st.success("Showing Input Video 2 (no interpolation)")
 else:
     w2 = round(1.0 - weight, 1)
-    interp_text = f"Interpolated Video: {weight:.1f} from Input 1 + {w2:.1f} from Input 2"
-st.markdown(f'<div class="centered-text">{interp_text}</div>', unsafe_allow_html=True)
-
-# Check files
-exists_1 = os.path.exists(video_input1)
-exists_interp = os.path.exists(video_interp)
-exists_2 = os.path.exists(video_input2)
+    st.info(
+        f"Generated motion: {weight:.1f} from Input Video 1 + {w2:.1f} from Input Video 2"
+    )
 
-# Display videos side by side using columns
+# Show videos side by side
 col1, col2, col3 = st.columns(3)
 
 with col1:
@@ -78,4 +64,24 @@ with col1:
             <source src="{video_input1}" type="video/mp4">
         </video>""", unsafe_allow_html=True)
     else:
-        st.error("Video 1 not fo
+        st.error("Video 1 not found")
+
+with col2:
+    st.markdown("**Interpolated Video**", unsafe_allow_html=True)
+    if exists_interp:
+        st.markdown(f"""
+        <video autoplay loop muted playsinline width="100%">
+            <source src="{video_interp}" type="video/mp4">
+        </video>""", unsafe_allow_html=True)
+    else:
+        st.error("Interpolated video not found")
+
+with col3:
+    st.markdown("**Input Video 2**", unsafe_allow_html=True)
+    if exists_2:
+        st.markdown(f"""
+        <video autoplay loop muted playsinline width="100%">
+            <source src="{video_input2}" type="video/mp4">
+        </video>""", unsafe_allow_html=True)
+    else:
+        st.error("Video 2 not found")