2

app.py

@@ -53,17 +53,15 @@ def run_command(command, working_dir, progress_bar, progress_text, step_start_pr
                             try:
                                 progress = float(line.strip().split("PROGRESS:")[1]) / 100
                                 if Path(command[1]).name == 'gen_skes.py':
-                                    # Scale 0-100% (2D) to 0-60%, 100-200% (3D) to 60-80%
                                     if progress <= 1.0:
-                                        adjusted_progress = step_start_progress + (progress * 0.6)  # 60% for 2D
+                                        adjusted_progress = step_start_progress + (progress * 0.6)
                                     else:
-                                        adjusted_progress = step_start_progress + 0.6 + ((progress - 1.0) * 0.2)  # 20% for 3D
+                                        adjusted_progress = step_start_progress + 0.6 + ((progress - 1.0) * 0.2)
                                 else:
-                                    # Scale 0-100% to 20% for conver_bvh.py
                                     adjusted_progress = step_start_progress + (progress * step_weight)
                                 total_progress = min(adjusted_progress, step_start_progress + step_weight)
                                 progress_bar.progress(total_progress)
-                                progress_text.text(f"{int(total_progress * 100)}%")
+                                progress_text.text(f"Progress: {int(total_progress * 100)}%")
                             except ValueError:
                                 pass
                     elif source == 'stderr':
@@ -81,7 +79,7 @@ def run_command(command, working_dir, progress_bar, progress_text, step_start_pr
         
         if show_progress:
             progress_bar.progress(step_start_progress + step_weight)
-            progress_text.text(f"{int((step_start_progress + step_weight) * 100)}%")
+            progress_text.text(f"Progress: {int((step_start_progress + step_weight) * 100)}%")
         return True
     
     except Exception as e:
@@ -132,7 +130,7 @@ def process_video(video_file):
         ]
         
         progress_bar = st.progress(0.0)
-        progress_text = st.empty()  # Added for percentage display
+        progress_text = st.empty()
         total_progress = 0.0
         
         for i, step in enumerate(pipeline_steps):
@@ -162,7 +160,7 @@ def process_video(video_file):
             total_progress += step["weight"]
             if step["show_progress"]:
                 progress_bar.progress(min(total_progress, 1.0))
-                progress_text.text(f"{int(total_progress * 100)}%")
+                progress_text.text(f"Progress: {int(total_progress * 100)}%")
         
         bvh_output_dir = output_dir / "bvh"
         bvh_file = bvh_output_dir / "output.bvh"
@@ -185,42 +183,367 @@ def process_video(video_file):
 def cleanup_immediate(output_dir):
     if output_dir and os.path.exists(output_dir):
         shutil.rmtree(output_dir, ignore_errors=True)
-        st.success("")
+        st.success("Output folder cleaned up successfully.")
     else:
-        st.warning("")
+        st.warning("No output folder to clean up.")
 
 def main():
-    st.title("Video to BVH Converter")
-    st.write("Upload a video file to convert it to BVH format")
+    st.set_page_config(
+        page_title="Motion Capture Studio | Video to BVH Converter",
+        page_icon="🎬",
+        layout="wide",
+        initial_sidebar_state="collapsed"
+    )
+
+    st.markdown("""
+    <style>
+    :root {
+        --bg-color: #1a1a1a;
+        --card-bg: #252525;
+        --primary-color: #bb86fc;
+        --secondary-color: #03dac6;
+        --error-color: #cf6679;
+        --text-color: #e0e0e0;
+        --text-secondary: #a0a0a0;
+        --border-color: #404040;
+        --shadow-color: rgba(0, 0, 0, 0.5);
+    }
+    
+    .stApp {
+        background-color: var(--bg-color);
+        font-family: 'Arial', sans-serif;
+    }
+    
+    h1, h2, h3, h4, h5, h6, p, li, div {
+        color: var(--text-color) !important;
+    }
+    
+    .card {
+        background-color: var(--card-bg);
+        border-radius: 20px;
+        padding: 2rem;
+        margin: 1rem auto;
+        border: 1px solid var(--border-color);
+        box-shadow: 0 8px 30px var(--shadow-color);
+        max-width: 1200px;
+    }
+    
+    .main-title {
+        font-size: 3.5rem;
+        font-weight: 900;
+        background: linear-gradient(135deg, var(--primary-color), var(--secondary-color));
+        -webkit-background-clip: text;
+        -webkit-text-fill-color: transparent;
+        text-align: center;
+        margin: 1.5rem 0 0.5rem;
+        text-shadow: 0 2px 10px rgba(187, 134, 252, 0.3);
+    }
+    
+    .subtitle {
+        font-size: 1.3rem;
+        color: var(--text-secondary);
+        text-align: center;
+        margin-bottom: 2.5rem;
+        font-weight: 300;
+        letter-spacing: 0.5px;
+    }
+    
+    .section-title {
+        font-size: 1.5rem;
+        font-weight: 700;
+        color: var(--primary-color) !important;
+        margin-bottom: 1.2rem;
+        text-transform: uppercase;
+        letter-spacing: 1px;
+    }
+    
+    .stButton > button {
+        background: linear-gradient(135deg, var(--primary-color), #9b59f5);
+        color: #fff !important;
+        border-radius: 12px;
+        padding: 0.8rem 2.5rem;
+        font-weight: 600;
+        font-size: 1.2rem;
+        border: none;
+        transition: all 0.3s ease;
+        width: 100%;
+        box-shadow: 0 4px 15px rgba(187, 134, 252, 0.3);
+    }
+    
+    .stButton > button:hover {
+        transform: translateY(-3px);
+        box-shadow: 0 6px 20px rgba(187, 134, 252, 0.5);
+        background: linear-gradient(135deg, #9b59f5, var(--primary-color));
+    }
+    
+    .stDownloadButton > button {
+        background: linear-gradient(135deg, var(--secondary-color), #02b3a3);
+        color: #fff !important;
+        border-radius: 12px;
+        padding: 0.8rem 2.5rem;
+        font-weight: 600;
+        font-size: 1.2rem;
+        border: none;
+        transition: all 0.3s ease;
+        width: 100%;
+        box-shadow: 0 4px 15px rgba(3, 218, 198, 0.3);
+    }
+    
+    .stDownloadButton > button:hover {
+        transform: translateY(-3px);
+        box-shadow: 0 6px 20px rgba(3, 218, 198, 0.5);
+        background: linear-gradient(135deg, #02b3a3, var(--secondary-color));
+    }
+    
+    .upload-container {
+        border: 2px dashed var(--border-color);
+        border-radius: 15px;
+        padding: 2.5rem;
+        text-align: center;
+        background-color: rgba(255, 255, 255, 0.05);
+        transition: all 0.3s ease;
+    }
+    
+    .upload-container:hover {
+        border-color: var(--primary-color);
+        box-shadow: 0 0 20px rgba(187, 134, 252, 0.2);
+    }
+    
+    .video-container {
+        border-radius: 15px;
+        overflow: hidden;
+        box-shadow: 0 6px 25px var(--shadow-color);
+        margin: 1rem 0;
+        border: 1px solid var(--border-color);
+        background-color: #000;
+    }
+    
+    .status-indicator {
+        padding: 1rem;
+        border-radius: 12px;
+        margin: 0.8rem 0;
+        display: flex;
+        align-items: center;
+        gap: 0.5rem;
+        font-size: 1.1rem;
+    }
+    
+    .status-indicator.info {
+        background-color: rgba(187, 134, 252, 0.1);
+        border-left: 5px solid var(--primary-color);
+    }
+    
+    .status-indicator.success {
+        background-color: rgba(3, 218, 198, 0.1);
+        border-left: 5px solid var(--secondary-color);
+    }
+    
+    .control-section {
+        text-align: center;
+        margin: 2rem 0;
+        padding: 1rem;
+        background-color: rgba(255, 255, 255, 0.03);
+        border-radius: 12px;
+        border: 1px solid var(--border-color);
+    }
+    
+    .stProgress {
+        width: 100%;
+        height: 12px;
+        border-radius: 6px;
+        background-color: #333;
+        margin: 1rem 0;
+    }
+    
+    .stProgress > div > div {
+        background: linear-gradient(135deg, var(--primary-color), var(--secondary-color));
+        border-radius: 6px;
+    }
     
-    uploaded_file = st.file_uploader("Choose a video file", type=['mp4', 'avi', 'mov'])
+    .progress-text {
+        font-weight: 600;
+        padding: 0.5rem 1rem;
+        background: linear-gradient(135deg, var(--primary-color), var(--secondary-color));
+        color: #fff !important;
+        border-radius: 20px;
+        box-shadow: 0 2px 10px rgba(187, 134, 252, 0.3);
+        display: inline-block;
+    }
     
-    if uploaded_file is not None:
-        st.video(uploaded_file)
+    .separator {
+        height: 2px;
+        background: linear-gradient(90deg, transparent, var(--border-color), transparent);
+        margin: 2.5rem 0;
+    }
+    
+    .footer {
+        text-align: center;
+        padding: 2.5rem 0 1.5rem;
+        font-size: 0.95rem;
+        color: var(--text-secondary);
+        border-top: 1px solid var(--border-color);
+        letter-spacing: 0.5px;
+    }
+    
+    .pipeline-step {
+        display: flex;
+        align-items: center;
+        background-color: rgba(255, 255, 255, 0.05);
+        padding: 1rem;
+        border-radius: 12px;
+        margin-bottom: 1rem;
+        border: 1px solid var(--border-color);
+        transition: all 0.3s ease;
+    }
+    
+    .pipeline-step:hover {
+        background-color: rgba(255, 255, 255, 0.08);
+        border-color: var(--primary-color);
+        box-shadow: 0 4px 15px rgba(187, 134, 252, 0.1);
+    }
+    
+    .step-number {
+        background: linear-gradient(135deg, var(--primary-color), var(--secondary-color));
+        color: #fff;
+        border-radius: 50%;
+        width: 40px;
+        height: 40px;
+        display: flex;
+        align-items: center;
+        justify-content: center;
+        margin-right: 1.2rem;
+        font-weight: 600;
+        font-size: 1.2rem;
+        box-shadow: 0 2px 10px rgba(187, 134, 252, 0.3);
+    }
+    
+    .step-title {
+        font-weight: 600;
+        font-size: 1.1rem;
+        margin-bottom: 0.3rem;
+    }
+    
+    .step-description {
+        color: var(--text-secondary);
+        font-size: 0.95rem;
+    }
+    
+    @keyframes pulse {
+        0% { transform: scale(1); }
+        50% { transform: scale(1.03); }
+        100% { transform: scale(1); }
+    }
+    
+    .animate-pulse {
+        animation: pulse 2s infinite;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+    # Header
+    st.markdown('<h1 class="main-title animate-pulse">Motion Capture Studio</h1>', unsafe_allow_html=True)
+    st.markdown('<p class="subtitle">Transform your videos into professional BVH motion files with cutting-edge AI</p>', unsafe_allow_html=True)
+
+    # Main content
+    st.markdown('<div class="card">', unsafe_allow_html=True)
+    
+    col1, col2 = st.columns([1, 1], gap="medium")
+
+    with col1:
+        st.markdown('<h3 class="section-title">Upload Your Video</h3>', unsafe_allow_html=True)
+        st.markdown('<div class="upload-container">', unsafe_allow_html=True)
+        uploaded_file = st.file_uploader(
+            "Drop your video here or click to browse",
+            type=['mp4', 'avi', 'mov'],
+            help="For best results, use clear full-body motion videos with good lighting",
+            key="file_uploader"
+        )
+        if uploaded_file:
+            st.session_state['uploaded_file'] = uploaded_file
+        st.markdown('</div>', unsafe_allow_html=True)
         
-        if st.button("Convert to BVH"):
-            with st.spinner("Processing video..."):
-                conversion_result = process_video(uploaded_file)
+        if not st.session_state.get('uploaded_file'):
+            st.markdown("""
+            <div class="status-indicator info">
+                💡 <strong>Pro Tip:</strong> Use MP4, AVI, or MOV files with clear, well-lit full-body motion
+            </div>
+            """, unsafe_allow_html=True)
+
+    with col2:
+        st.markdown('<h3 class="section-title">Video Preview</h3>', unsafe_allow_html=True)
+        if uploaded_file := st.session_state.get('uploaded_file', None):
+            st.markdown('<div class="video-container">', unsafe_allow_html=True)
+            st.video(uploaded_file)
+            st.markdown('</div>', unsafe_allow_html=True)
+        else:
+            st.markdown("""
+            <div style="height: 250px; display: flex; align-items: center; justify-content: center; 
+                 border: 2px dashed var(--border-color); border-radius: 15px; background-color: rgba(255, 255, 255, 0.05);">
+                <span style="color: var(--text-secondary); font-size: 1.2rem;">Your video preview will appear here</span>
+            </div>
+            """, unsafe_allow_html=True)
+
+    # Control Section (Button and Progress Bar)
+    if st.session_state.get('uploaded_file'):
+        st.markdown('<div class="control-section">', unsafe_allow_html=True)
+        if st.button("⚡ Start Motion Capture", key="convert_btn"):
+            with st.spinner("Processing your video..."):
+                progress_bar = st.progress(0.0)
+                progress_text = st.empty()
                 
-                if conversion_result:
-                    with open(conversion_result['bvh_file'], "rb") as f:
+                st.markdown('<div class="status-indicator info">🔄 Analyzing motion patterns...</div>', unsafe_allow_html=True)
+                
+                result = process_video(st.session_state['uploaded_file'])
+                
+                if result:
+                    st.markdown('<div class="status-indicator success">✅ Motion capture complete!</div>', unsafe_allow_html=True)
+                    with open(result['bvh_file'], "rb") as f:
                         st.download_button(
-                            label="Download BVH File",
+                            label="📥 Download BVH File",
                             data=f,
-                            file_name="output.bvh",
+                            file_name="motion_capture.bvh",
                             mime="application/octet-stream",
                             on_click=cleanup_immediate,
-                            args=(conversion_result['output_dir'],)
+                            args=(result['output_dir'],),
+                            key="download_btn"
                         )
-                    st.success("Conversion completed successfully! File will be deleted after download or in 30 minutes.")
+        st.markdown('</div>', unsafe_allow_html=True)
+
+    # Pipeline Info
+    st.markdown('<div class="separator"></div>', unsafe_allow_html=True)
+    st.markdown('<h3 class="section-title">Processing Pipeline</h3>', unsafe_allow_html=True)
+    st.markdown("""
+    <div class="pipeline-step">
+        <div class="step-number">1</div>
+        <div>
+            <div class="step-title">Pose Estimation</div>
+            <div class="step-description">AI detects and tracks human movements frame-by-frame</div>
+        </div>
+    </div>
+    <div class="pipeline-step">
+        <div class="step-number">2</div>
+        <div>
+            <div class="step-title">3D Conversion</div>
+            <div class="step-description">Converts 2D poses into 3D spatial data</div>
+        </div>
+    </div>
+    <div class="pipeline-step">
+        <div class="step-number">3</div>
+        <div>
+            <div class="step-title">BVH Generation</div>
+            <div class="step-description">Formats motion data into industry-standard BVH files</div>
+        </div>
+    </div>
+    """, unsafe_allow_html=True)
+
+    st.markdown('</div>', unsafe_allow_html=True)
     
+    # Footer
     st.markdown("""
-    ### Instructions
-    1. Upload a video file using the uploader above
-    2. Click 'Convert to BVH' to start the conversion process
-    3. Watch the progress bar as it processes
-    4. Download the resulting BVH file - folder will be deleted immediately after download or after 30 minutes
-    """)
+    <div class="footer">
+        © 2025 Motion Capture Studio | Powered by Streamlit & Advanced AI Technology
+    </div>
+    """, unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()