Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,8 +1,9 @@
 import streamlit as st
 import os
 
-# Video directory
-VIDEO_FOLDER = "./src/synthda_falling_realreal/"
+# Video directories
+VIDEO_FOLDER_1 = "./src/synthda_falling_realreal/"
+VIDEO_FOLDER_2 = "./src/synthda_demo_fall_2/"
 
 # Set page layout
 st.set_page_config(layout="wide")
@@ -38,45 +39,58 @@ else:
 
 st.markdown(f'<p style="text-align: center; color: #444;"><strong>{interp_text}</strong></p>', unsafe_allow_html=True)
 
-# Filepaths
+# ---------- Row 1: synthda_falling_realreal ----------
 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, filename_input1)
-video_input2 = os.path.join(VIDEO_FOLDER, filename_input2)
+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)
 
-# Layout: 3 columns for video display
+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 <span style="font-weight: normal;">(Generated with generative AI)</span>
-        </div>
-    """, unsafe_allow_html=True)
-    
-    if exists_1:
-        st.video(video_input1)
-    else:
-        st.error("Video 1 not found")
+    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")
 
 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")
+    st.video(video_interp) if exists_interp else st.error("Interpolated video not found")
 
 with col3:
-    st.markdown("""<div style='text-align: center; font-weight: bold;'>Input Video 2 <span style="font-weight: normal;">(Obtained from real-world dataset)</span>
-    </div>""", unsafe_allow_html=True)
-    if exists_2:
-        st.video(video_input2)
-    else:
-        st.error("Video 2 is not found")
+    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")
+
+# ---------- Row 2: synthda_demo_fall_2 ----------
+filename_interp2 = f"videos_generated_{weight:.1f}.mp4"
+filename_input1_2 = "videos_generated_0.0.mp4"
+filename_input2_2 = "videos_generated_1.0.mp4"
+
+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")
+
+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")
+
+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")