Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,3 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
 import os
 
@@ -8,26 +5,23 @@ import os
 VIDEO_FOLDER = "./src/synthda_falling_realreal/"
 
 st.set_page_config(layout="wide")
-st.title("AutoSynthDa Pose Interpolation Viewer")
 
+# Title and instructions
 st.markdown("""
-### AutoSynthDa Interpolation Viewer
-
-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)
-""")
-
-# 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)
+<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)
+
+# Slider starts at 0.5 by default
+weight = st.slider("Interpolation Weight", 0.0, 1.0, step=0.1, value=0.5)
 
 # File paths
 filename_interp = f"videos_generated_{weight:.1f}.mp4"
@@ -42,37 +36,37 @@ exists_interp = os.path.exists(video_interp)
 exists_1 = os.path.exists(video_input1)
 exists_2 = os.path.exists(video_input2)
 
-# Description of blending
+# Interpolation info text
 if weight == 0.0:
-    st.success("Showing Input Video 1 (no interpolation)")
+    interp_text = "Displaying Input Video 1 (no interpolation)"
 elif weight == 1.0:
-    st.success("Showing Input Video 2 (no interpolation)")
+    interp_text = "Displaying Input Video 2 (no interpolation)"
 else:
     w2 = round(1.0 - weight, 1)
-    st.info(
-        f"Generated motion: {weight:.1f} from Input Video 1 + {w2:.1f} from Input Video 2"
-    )
+    interp_text = f"Interpolated 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)
 
-# Show videos side by side
+# Layout with 3 video panels
 col1, col2, col3 = st.columns(3)
 
 with col1:
-    st.markdown("**Input Video 1**", unsafe_allow_html=True)
+    st.markdown("**Input Video 1**")
     if exists_1:
-         st.video(video_input1)
+        st.video(video_input1)
     else:
         st.error("Video 1 not found")
 
 with col2:
-    st.markdown("**Interpolated Video**", unsafe_allow_html=True)
+    st.markdown("**Interpolated Video**")
     if exists_interp:
-         st.video(video_interp)
+        st.video(video_interp)
     else:
         st.error("Interpolated video not found")
 
 with col3:
-    st.markdown("**Input Video 2**", unsafe_allow_html=True)
+    st.markdown("**Input Video 2**")
     if exists_2:
-         st.video(video_input2)
+        st.video(video_input2)
     else:
         st.error("Video 2 not found")