Add application file

app.py

@@ -0,0 +1,143 @@
+import streamlit as st
+import subprocess
+import sys
+import os
+import time
+import shutil
+import tempfile
+
+def run_command(command, description, working_dir):
+    st.write(f"Running command: {' '.join(command)}")
+    try:
+        start_time = time.time()
+        process = subprocess.run(
+            command,
+            check=True,
+            cwd=working_dir,
+            capture_output=True,
+            text=True
+        )
+        end_time = time.time()
+        execution_time = end_time - start_time
+        st.write(f"{description} completed in {execution_time:.2f} seconds")
+        if process.stdout:
+            st.text(f"Output:\n{process.stdout}")
+        if process.stderr:
+            st.text(f"Warnings:\n{process.stderr}")
+        return True
+        
+    except subprocess.CalledProcessError as e:
+        st.error(f"Error in {description}: {e.stderr}")
+        return False
+    
+    except Exception as e:
+        st.error(f"Unexpected error in {description}: {str(e)}")
+        return False
+
+def process_video(video_file):
+    base_dir = os.path.dirname(os.path.abspath(__file__))
+    gen_skes_path = os.path.join(base_dir, "VideoToNPZ", "gen_skes.py")
+    convert_obj_path = os.path.join(base_dir, "convertNPZtoBVH", "conver_obj.py")
+    convert_bvh_path = os.path.join(base_dir, "convertNPZtoBVH", "conver_bvh.py")
+    
+    st.write(f"Base directory: {base_dir}")
+    st.write(f"gen_skes_path: {gen_skes_path}")
+    
+    for script_path in [gen_skes_path, convert_obj_path, convert_bvh_path]:
+        if not os.path.exists(script_path):
+            st.error(f"Required script not found: {script_path}")
+            return None
+    
+    with tempfile.TemporaryDirectory() as tmp_dir:
+        video_path = os.path.join(tmp_dir, "input_video.mp4")
+        with open(video_path, "wb") as f:
+            f.write(video_file.read())
+        
+        if not os.path.exists(video_path):
+            st.error(f"Video file not found at: {video_path}")
+            return None
+        st.write(f"Video file saved at: {video_path}, size: {os.path.getsize(video_path)} bytes")
+        
+        pipeline_steps = [
+            {
+                "command": [sys.executable, gen_skes_path, "-v", video_path],
+                "description": "Generating skeletons",
+                "working_dir": os.path.dirname(gen_skes_path)
+            },
+            {
+                "command": [sys.executable, convert_obj_path],
+                "description": "Converting to OBJ sequence",
+                "working_dir": os.path.dirname(convert_obj_path)
+            },
+            {
+                "command": [sys.executable, convert_bvh_path],
+                "description": "Converting to BVH",
+                "working_dir": os.path.dirname(convert_bvh_path)
+            }
+        ]
+        
+        successful = 0
+        failed = 0
+        progress_bar = st.progress(0)
+        status_text = st.empty()
+        
+        for i, step in enumerate(pipeline_steps):
+            status_text.text(f"Processing: {step['description']}...")
+            if run_command(step["command"], step["description"], step["working_dir"]):
+                successful += 1
+                if step["description"] == "Generating skeletons":
+                    npz_path = os.path.join(base_dir, "outputs", "npz", "input_video.npz")
+                    st.write(f"Expected NPZ file at: {npz_path}")
+                    if os.path.exists(npz_path):
+                        st.write(f"NPZ file found, size: {os.path.getsize(npz_path)} bytes")
+                    else:
+                        st.error("NPZ file not generated!")
+            else:
+                failed += 1
+            progress_bar.progress((i + 1) / len(pipeline_steps))
+        
+        status_text.text("Processing complete!")
+        
+        bvh_output_dir = os.path.join(os.path.dirname(convert_bvh_path), "../outputs/bvh")
+        bvh_file = os.path.join(bvh_output_dir, "output.bvh")
+        st.write(f"Checking for BVH file at: {bvh_file}")
+        
+        if os.path.exists(bvh_file) and successful == len(pipeline_steps):
+            return bvh_file
+        else:
+            st.error("Failed to generate BVH file")
+            return None
+
+def main():
+    st.title("Video to BVH Converter")
+    st.write("Upload a video file to convert it to BVH format")
+    
+    uploaded_file = st.file_uploader("Choose a video file", type=['mp4', 'avi', 'mov'])
+    
+    if uploaded_file is not None:
+        st.video(uploaded_file)
+        
+        if st.button("Convert to BVH"):
+            with st.spinner("Processing video... This may take a few minutes"):
+                bvh_file = process_video(uploaded_file)
+                
+                if bvh_file:
+                    with open(bvh_file, "rb") as f:
+                        st.download_button(
+                            label="Download BVH File",
+                            data=f,
+                            file_name="output.bvh",
+                            mime="application/octet-stream"
+                        )
+                    st.success("Conversion completed successfully!")
+    
+    st.markdown("""
+    ### Instructions
+    1. Upload a video file using the uploader above
+    2. Click 'Convert to BVH' to start the conversion process
+    3. Wait for processing to complete
+    4. Download the resulting BVH file using the provided button
+    """)
+
+if __name__ == "__main__":
+    main()