added more dependencies

app.py

@@ -741,7 +741,7 @@ def run_openvla_inference(request: InferenceRequest) -> Tuple[Optional[str], str
         else:
             error_msg = f"❌ OpenVLA Error: {result.get('error', 'Unknown error')}\n\n{result.get('status_message', '')}"
             return None, error_msg
-            
+        
     except Exception as e:
         import traceback
         return None, f"❌ Worker communication error: {str(e)}\n\n{traceback.format_exc()}"
@@ -771,7 +771,7 @@ if HAS_OPENVLA:
         ),
         run_inference=run_openvla_inference,
     )
-else:
+        else:
     print("ℹ OpenVLA environment not found - OpenVLA model will not be available")
 
 
@@ -899,7 +899,7 @@ def create_gradio_interface():
             outputs=model_info,
             queue=False,
         )
-
+        
         # Event handler
         run_button.click(
             fn=run_model_inference,

environment_openpi.yml

@@ -75,6 +75,7 @@ dependencies:
     - moviepy==1.0.3
     - imageio-ffmpeg==0.6.0
     - opencv-python-headless==4.11.0.86
+    - av>=12.0.0
     # MuJoCo and robotics
     - mujoco==3.3.3
     - dm-control==1.0.31

inference_openpi.py

@@ -43,9 +43,9 @@ except ImportError as e:
 
 # Import OpenPI dependencies (only available in openpi_env)
 try:
-    # Import get_config function and AssetsConfig class directly
-    from openpi.training.config import get_config, AssetsConfig
-    from openpi.policies.policy_config import PIPolicy
+    # Import modules (same pattern as old app.py that worked)
+    from openpi.training import config as _config
+    from openpi.policies import policy_config as _policy_config
     print("✓ OpenPI config and policy modules imported successfully", file=sys.stderr, flush=True)
 except ImportError as e:
     print(f"✗ OpenPI config/policy import failed: {e}", file=sys.stderr, flush=True)
@@ -241,13 +241,13 @@ def load_pi0_policy(task_name: str, ckpt_path: str):
     if cache_key in _POLICY_CACHE:
         return _POLICY_CACHE[cache_key]
     
-    cfg = get_config("pi0_base_bimanual_droid_finetune")
-    bimanual_assets = AssetsConfig(
+    cfg = _config.get_config("pi0_base_bimanual_droid_finetune")
+    bimanual_assets = _config.AssetsConfig(
         assets_dir=f"{checkpoint_path}/assets/",
         asset_id=f"tan7271/{task_name}",
     )
     cfg = dataclasses.replace(cfg, data=dataclasses.replace(cfg.data, assets=bimanual_assets))
-    policy = PIPolicy.create_trained_policy(cfg, checkpoint_path)
+    policy = _policy_config.create_trained_policy(cfg, checkpoint_path)
     
     _POLICY_CACHE[cache_key] = policy
     return policy