Update app.py

app.py

@@ -1,22 +1,32 @@
+import streamlit as st
 import torch
 import torchvision.transforms as transforms
 from easymocap.apps.neuralbody import demo
 
-# Get the embed code for the YouTube video
-embed_code = get_embed_code("vjBrN18wiuE")
+# Create a text box to enter the YouTube video ID
+video_id = st.text_input("Enter the YouTube video ID:")
 
-# Download the video and extract the frames
-video = download_video(embed_code)
-frames = extract_frames(video)
+# Create a progress bar
+progress_bar = st.progress(0)
 
-# Convert the frames to tensors
-frames = torch.from_numpy(frames).float()
+# If the user enters a valid YouTube video ID, download the video and extract the frames
+if video_id:
+  embed_code = get_embed_code(video_id)
+  video = download_video(embed_code)
+  frames = extract_frames(video)
 
-# Create a 3D scene from the frames
-scene = create_scene(frames)
+  # Convert the frames to tensors
+  frames = torch.from_numpy(frames).float()
 
-# Use view synthesis to render a novel view of the scene
-novel_view = render_novel_view(scene)
+  # Create a 3D scene from the frames
+  scene = create_scene(frames)
+
+  # Use view synthesis to render a novel view of the scene
+  novel_view = render_novel_view(scene)
+
+  # Update the progress bar
+  progress_bar.progress(100)
+
+  # Show the novel view
+  st.image(novel_view)
 
-# Save the novel view
-save_image(novel_view, "novel_view.png")