Update src/streamlit_app.py

src/streamlit_app.py

@@ -14,33 +14,27 @@ st.markdown("""
     <h3 style="text-align: center;">SynthDa Interpolation Demo Viewer</h3>
     <p style="text-align: center;">
         AutoSynthDa blends two input motion videos to <strong>generate kinematically coherent, synthetic action videos</strong>.<br>
-        Use the slider below to explore how the system interpolates motion from one video to another.<br>
+        Use the sliders below to explore how the system interpolates motion from one video to another.<br>
         Source: <a href="https://github.com/nvidia/synthda" target="_blank">github.com/nvidia/synthda</a>
     </p>
 """, unsafe_allow_html=True)
 
-# Slider instruction (centered)
-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
-)
+# ---------- Row 1: synthda_falling_realreal ----------
+st.markdown("## Demo Set 1: Falling Real-Real Interpolation")
 
-# Slider (starts at 0.5)
-weight = st.slider("Interpolation Weight", 0.1, 0.9, 0.5, step=0.1)
+weight1 = st.slider("Interpolation Weight (Set 1)", 0.0, 1.0, 0.5, step=0.1)
 
-# Interpolation explanation (centered)
-if weight == 0.0:
-    interp_text = "Showing Input Video 1 (no interpolation)"
-elif weight == 1.0:
-    interp_text = "Showing Input Video 2 (no interpolation)"
+if weight1 == 0.0:
+    interp_text1 = "Showing Input Video 1 (no interpolation)"
+elif weight1 == 1.0:
+    interp_text1 = "Showing Input Video 2 (no interpolation)"
 else:
-    w2 = round(1.0 - weight, 1)
-    interp_text = f"Generated motion: {weight:.1f} from Input Video 1 + {w2:.1f} from Input Video 2"
+    w2 = round(1.0 - weight1, 1)
+    interp_text1 = f"Generated motion: {weight1:.1f} from Input Video 1 + {w2:.1f} from Input Video 2"
 
-st.markdown(f'<p style="text-align: center; color: #444;"><strong>{interp_text}</strong></p>', unsafe_allow_html=True)
+st.markdown(f"**{interp_text1}**")
 
-# ---------- Row 1: synthda_falling_realreal ----------
-filename_interp = f"videos_generated_{weight:.1f}.mp4"
+filename_interp = f"videos_generated_{weight1:.1f}.mp4"
 filename_input1 = "videos_generated_0.0.mp4"
 filename_input2 = "videos_generated_1.0.mp4"
 
@@ -48,27 +42,45 @@ video_interp = os.path.join(VIDEO_FOLDER_1, filename_interp)
 video_input1 = os.path.join(VIDEO_FOLDER_1, filename_input1)
 video_input2 = os.path.join(VIDEO_FOLDER_1, filename_input2)
 
-exists_interp = os.path.exists(video_interp)
-exists_1 = os.path.exists(video_input1)
-exists_2 = os.path.exists(video_input2)
-
-st.markdown("### Demo Set 1: Falling Real-Real Interpolation")
 col1, col2, col3 = st.columns(3)
 
 with col1:
-    st.markdown("<div style='text-align: center; font-weight: bold;'>Input Video 1 (Generated)</div>", unsafe_allow_html=True)
-    st.video(video_input1) if exists_1 else st.error("Video 1 not found")
+    st.markdown("**Input Video 1 (Generated)**")
+    if os.path.exists(video_input1):
+        st.video(video_input1)
+    else:
+        st.error("Video 1 not found")
 
 with col2:
-    st.markdown("<div style='text-align: center; font-weight: bold;'>Interpolated Video</div>", unsafe_allow_html=True)
-    st.video(video_interp) if exists_interp else st.error("Interpolated video not found")
+    st.markdown("**Interpolated Video**")
+    if os.path.exists(video_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 (Real)</div>", unsafe_allow_html=True)
-    st.video(video_input2) if exists_2 else st.error("Video 2 not found")
+    st.markdown("**Input Video 2 (Real)**")
+    if os.path.exists(video_input2):
+        st.video(video_input2)
+    else:
+        st.error("Video 2 not found")
 
 # ---------- Row 2: synthda_demo_fall_2 ----------
-filename_interp2 = f"videos_generated_{weight:.1f}.mp4"
+st.markdown("## Demo Set 2: Fall Demo Interpolation")
+
+weight2 = st.slider("Interpolation Weight (Set 2)", 0.0, 1.0, 0.5, step=0.1)
+
+if weight2 == 0.0:
+    interp_text2 = "Showing Input Video 1 (no interpolation)"
+elif weight2 == 1.0:
+    interp_text2 = "Showing Input Video 2 (no interpolation)"
+else:
+    w2 = round(1.0 - weight2, 1)
+    interp_text2 = f"Generated motion: {weight2:.1f} from Input Video 1 + {w2:.1f} from Input Video 2"
+
+st.markdown(f"**{interp_text2}**")
+
+filename_interp2 = f"videos_generated_{weight2:.1f}.mp4"
 filename_input1_2 = "videos_generated_0.0.mp4"
 filename_input2_2 = "videos_generated_1.0.mp4"
 
@@ -76,21 +88,25 @@ video_interp2 = os.path.join(VIDEO_FOLDER_2, filename_interp2)
 video_input1_2 = os.path.join(VIDEO_FOLDER_2, filename_input1_2)
 video_input2_2 = os.path.join(VIDEO_FOLDER_2, filename_input2_2)
 
-exists_interp2 = os.path.exists(video_interp2)
-exists_1_2 = os.path.exists(video_input1_2)
-exists_2_2 = os.path.exists(video_input2_2)
-
-st.markdown("### Demo Set 2: Fall Demo Interpolation")
 col4, col5, col6 = st.columns(3)
 
 with col4:
-    st.markdown("<div style='text-align: center; font-weight: bold;'>Input Video 1 (Generated)</div>", unsafe_allow_html=True)
-    st.video(video_input1_2) if exists_1_2 else st.error("Video 1 not found")
+    st.markdown("**Input Video 1 (Generated)**")
+    if os.path.exists(video_input1_2):
+        st.video(video_input1_2)
+    else:
+        st.error("Video 1 not found")
 
 with col5:
-    st.markdown("<div style='text-align: center; font-weight: bold;'>Interpolated Video</div>", unsafe_allow_html=True)
-    st.video(video_interp2) if exists_interp2 else st.error("Interpolated video not found")
+    st.markdown("**Interpolated Video**")
+    if os.path.exists(video_interp2):
+        st.video(video_interp2)
+    else:
+        st.error("Interpolated video not found")
 
 with col6:
-    st.markdown("<div style='text-align: center; font-weight: bold;'>Input Video 2 (Real)</div>", unsafe_allow_html=True)
-    st.video(video_input2_2) if exists_2_2 else st.error("Video 2 not found")
+    st.markdown("**Input Video 2 (Real)**")
+    if os.path.exists(video_input2_2):
+        st.video(video_input2_2)
+    else:
+        st.error("Video 2 not found")