|
|
|
|
|
import os
|
|
|
import glob
|
|
|
from app import load_and_render, run_analysis, DATA_DIR
|
|
|
|
|
|
def verify():
|
|
|
print("Verifying data availability...")
|
|
|
candidates = glob.glob(os.path.join(DATA_DIR, "*.pkl"))
|
|
|
if not candidates:
|
|
|
print("No candidates found in data directory!")
|
|
|
return
|
|
|
|
|
|
|
|
|
candidate_file = "p_005.pkl"
|
|
|
|
|
|
print(f"Testing with candidate: {candidate_file}")
|
|
|
|
|
|
|
|
|
print("\n--- Testing load_and_render ---")
|
|
|
video_path = load_and_render(candidate_file)
|
|
|
|
|
|
if not video_path or not os.path.exists(video_path):
|
|
|
print(f"FAILED: Video rendering failed for {candidate_file}")
|
|
|
return
|
|
|
print(f"SUCCESS: Video rendered as {video_path}")
|
|
|
|
|
|
|
|
|
print("\n--- Testing run_analysis ---")
|
|
|
output_video, output_json = run_analysis(candidate_file, video_path)
|
|
|
|
|
|
if not output_video:
|
|
|
print(f"FAILED: Analysis failed. Error: {output_json}")
|
|
|
else:
|
|
|
print(f"SUCCESS: Analysis complete.")
|
|
|
print(f"Output Video: {output_video}")
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
verify()
|
|
|
|