Update app.py

app.py

@@ -1,4 +1,4 @@
-# ------------------ 导入库 ------------------
+# ------------------ Import Libraries ------------------
 import dash
 from dash import dcc, html, Input, Output, State, no_update
 import plotly.graph_objects as go
@@ -14,7 +14,7 @@ import os
 from urllib.parse import urljoin
 import subprocess
 
-# ------------------ 数据下载与处理 ------------------
+# ------------------ Data Download and Processing ------------------
 class RemoteDatasetLoader:
     def __init__(self, repo_id: str, timeout: int = 30):
         self.repo_id = repo_id
@@ -40,7 +40,7 @@ class RemoteDatasetLoader:
     def _is_valid_mp4(self, file_path):
         if not os.path.exists(file_path) or os.path.getsize(file_path) < 1024 * 100:
             return False
-        # 用ffprobe检查是否为有效mp4
+        # Use ffprobe to check if it is a valid mp4
         try:
             result = subprocess.run([
                 'ffprobe', '-v', 'error', '-select_streams', 'v:0',
@@ -55,9 +55,9 @@ class RemoteDatasetLoader:
     def _download_video(self, video_url: str, save_path: str) -> str:
         response = requests.get(video_url, timeout=self.timeout, stream=True)
         response.raise_for_status()
-        # 检查Content-Type
+        # Check Content-Type
         if 'video' not in response.headers.get('Content-Type', ''):
-            raise ValueError(f"URL {video_url} 返回的不是视频内容，Content-Type: {response.headers.get('Content-Type')}")
+            raise ValueError(f"URL {video_url} does not return video content, Content-Type: {response.headers.get('Content-Type')}")
         os.makedirs(os.path.dirname(save_path), exist_ok=True)
         with open(save_path, 'wb') as f:
             for chunk in response.iter_content(chunk_size=8192):
@@ -68,7 +68,7 @@ class RemoteDatasetLoader:
                           video_keys=None,
                           download_dir=None):
         dataset_info = self._get_dataset_info()
-        self._get_episode_info(episode_id)  # 检查episode是否存在
+        self._get_episode_info(episode_id)  # Check if episode exists
 
         if download_dir is None:
             download_dir = tempfile.mkdtemp(prefix="lerobot_videos_")
@@ -81,8 +81,8 @@ class RemoteDatasetLoader:
         video_paths = []
         chunks_size = dataset_info.get("chunks_size", 1000)
 
-        # 创建repo特定的子目录
-        repo_name = self.repo_id.replace('/', '_')  # 将 / 替换为 _ 避免路径问题
+        # Create repo-specific subdirectory
+        repo_name = self.repo_id.replace('/', '_')  # Replace / with _ to avoid path issues
         repo_dir = os.path.join(download_dir, repo_name)
         os.makedirs(repo_dir, exist_ok=True)
 
@@ -94,7 +94,7 @@ class RemoteDatasetLoader:
             )
             video_filename = f"episode_{episode_id}_{video_key}.mp4"
             local_path = os.path.join(repo_dir, video_filename)
-            # 优先加载本地有效mp4
+            # Prefer loading local valid mp4
             if self._is_valid_mp4(local_path):
                 print(f"Local valid video found: {local_path}")
                 video_paths.append(local_path)
@@ -138,7 +138,7 @@ def get_video_codec_info(video_path):
                 if stream.get('codec_type') == 'video':
                     return stream.get('codec_name', 'unknown')
     except Exception as e:
-        print(f"获取视频编码信息失败: {e}")
+        print(f"Failed to get video codec info: {e}")
     return 'unknown'
 
 def reencode_video_to_h264(input_path, output_path=None, quality='medium'):
@@ -163,21 +163,21 @@ def reencode_video_to_h264(input_path, output_path=None, quality='medium'):
         if result.returncode == 0:
             return output_path
         else:
-            print(f"重编码失败: {result.stderr}")
+            print(f"Re-encoding failed: {result.stderr}")
             return input_path
     except subprocess.TimeoutExpired:
-        print("重编码超时")
+        print("Re-encoding timeout")
         return input_path
     except Exception as e:
-        print(f"重编码异常: {e}")
+        print(f"Re-encoding exception: {e}")
         return input_path
 
 def process_video_for_compatibility(video_path):
     if not os.path.exists(video_path):
-        print(f"视频文件不存在: {video_path}")
+        print(f"Video file does not exist: {video_path}")
         return video_path
     if not check_ffmpeg_available():
-        print("ffmpeg不可用，跳过重编码")
+        print("ffmpeg not available, skipping re-encoding")
         return video_path
     codec = get_video_codec_info(video_path)
     if codec in ['av01', 'av1', 'vp9', 'vp8'] or codec == 'unknown':
@@ -185,7 +185,7 @@ def process_video_for_compatibility(video_path):
         if os.path.exists(reencoded_path) and os.path.getsize(reencoded_path) > 1024:
             return reencoded_path
         else:
-            print("重编码失败，使用原始文件")
+            print("Re-encoding failed, using original file")
             return video_path
     else:
         return video_path
@@ -202,15 +202,15 @@ def load_remote_dataset(repo_id: str,
         processed_video_paths.append(processed_path)
     return processed_video_paths, df
 
-# ------------------ Dash 初始化 ------------------
+# ------------------ Dash Initialization ------------------
 app = dash.Dash(__name__, suppress_callback_exceptions=True)
 server = app.server
 
-# ------------------ 页面布局 ------------------
+# ------------------ Page Layout ------------------
 app.layout = html.Div([
     # Header with gradient background
     html.Div([
-        html.H1("Robot Data Visualization", 
+        html.H1("Keyframe Identification", 
                 style={
                     "textAlign": "center", 
                     "marginBottom": "10px",
@@ -340,7 +340,7 @@ app.layout = html.Div([
     "padding": "0"
 })
 
-# ------------------ 数据加载回调 ------------------
+# ------------------ Data Loading Callback ------------------
 @app.callback(
     Output("store-data", "data"),
     Input("btn-load", "n_clicks"),
@@ -367,7 +367,7 @@ def load_data_callback(n_clicks, repo_id, episode_id):
         print(f"Data loading error: {e}")
         return {}
 
-# ------------------ 主内容渲染回调 ------------------
+# ------------------ Main Content Rendering Callback ------------------
 @app.callback(
     Output("main-content", "children"),
     Input("store-data", "data")
@@ -389,7 +389,7 @@ def update_main_content(data):
     rows = []
     for i, joint in enumerate(columns):
         rows.append(html.Div([
-            # 关节图 - 左侧50%
+            # Joint Graph - Left 50%
             html.Div([
                 dcc.Graph(id=f"graph-{i}")
             ], style={
@@ -401,7 +401,7 @@ def update_main_content(data):
                 "border": "1px solid #e9ecef",
                 "marginRight": "2%"
             }),
-            # 视频区域 - 右侧48%
+            # Video Area - Right 48%
             html.Div([
                 html.Img(id=f"video1-{i}", style={
                     "width": "49%", 
@@ -435,7 +435,7 @@ def update_main_content(data):
         }))
     return html.Div(rows)
 
-# ------------------ 阴影与高亮工具函数 ------------------
+# ------------------ Shadow and Highlight Utility Functions ------------------
 def find_intervals(mask):
     intervals = []
     start = None
@@ -509,7 +509,7 @@ def generate_joint_graph(joint_name, idx, action_df, delta_t, time_for_plot, all
             "x1": shadow['end_time'],
             "y0": 0,
             "y1": 1,
-            "fillcolor": "#ef4444",  # 固定红色
+            "fillcolor": "#ef4444",  # Fixed red
             "opacity": 0.4,
             "line": {"width": 0}
         })
@@ -558,7 +558,7 @@ def generate_joint_graph(joint_name, idx, action_df, delta_t, time_for_plot, all
         )
     }
 
-# ------------------ 图表更新回调 ------------------
+# ------------------ Chart Update Callback ------------------
 @app.callback(
     [Output(f"graph-{i}", "figure") for i in range(6)],
     [Input("store-data", "data")],
@@ -578,18 +578,18 @@ def update_all_graphs(data):
     for joint in columns:
         all_shadows[joint] = get_shadow_info(joint, action_df, delta_t, time_for_plot)
 
-    # 生成所有图表，无高亮逻辑
+    # Generate all charts, no highlight logic
     return [
         generate_joint_graph(joint, i, action_df, delta_t, time_for_plot, all_shadows)
         for i, joint in enumerate(columns)
     ]
 
-# ------------------ 视频帧提取函数 ------------------
+# ------------------ Video Frame Extraction Function ------------------
 def get_video_frame(video_path, time_in_seconds):
     try:
         cap = cv2.VideoCapture(video_path)
         if not cap.isOpened():
-            print(f"❌ 无法打开视频: {video_path}")
+            print(f"❌ Cannot open video: {video_path}")
             return None
         fps = cap.get(cv2.CAP_PROP_FPS)
         if fps <= 0:
@@ -612,10 +612,10 @@ def get_video_frame(video_path, time_in_seconds):
         else:
             return None
     except Exception as e:
-        print(f"❌ 提取视频帧异常: {e}")
+        print(f"❌ Exception extracting video frame: {e}")
         return None
 
-# ------------------ 视频帧回调 ------------------
+# ------------------ Video Frame Callback ------------------
 for i in range(6):
     @app.callback(
         Output(f"video1-{i}", "src"),
@@ -633,13 +633,13 @@ for i in range(6):
         time_for_plot = timestamps[1:]
         video_paths = data["video_paths"]
         
-        # 确定要显示的时间点
-        display_time = 0.0  # 默认显示开始时间
+        # Determine the time point to display
+        display_time = 0.0  # Default to start time
         if hover_data and "points" in hover_data and len(hover_data["points"]) > 0:
-            # 如果有hover数据，使用hover时间
+            # If there is hover data, use hover time
             display_time = float(hover_data["points"][0]["x"])
         elif len(time_for_plot) > 0:
-            # 如果没有hover数据，使用时间轴开始时间
+            # If no hover data, use the start time of the timeline
             display_time = time_for_plot[0]
         
         try:
@@ -653,6 +653,6 @@ for i in range(6):
             print(f"update_video_frames callback error: {e}")
             return no_update, no_update
 
-# ------------------ 启动应用 ------------------
+# ------------------ Start Application ------------------
 if __name__ == "__main__":
     app.run(debug=True, host='0.0.0.0', port=7860)