79 lines
3.1 KiB
Python
79 lines
3.1 KiB
Python
import cv2
|
|
from video_cv_tracker import TemporalTracker
|
|
from youtube_tab_to_pdf import extract_unique_scroll, generate_long_image, generate_pdf, download_video, extract_frames
|
|
import sys
|
|
import os
|
|
from pathlib import Path
|
|
|
|
# Run verification specifically on Shintakarajima
|
|
url = "https://youtu.be/tJq1n8TofM0"
|
|
video_path = Path("output/サカナクション／新宝島(エレキギターTAB) 難易度★★★ sakanaction shintakarajima.mp4")
|
|
|
|
print("Extracting full video for final 142-measure verification...")
|
|
cap = cv2.VideoCapture(str(video_path))
|
|
|
|
# PRE-CALCULATE Dynamic Crop
|
|
# Just like extract_unique_scroll does automatically, we detect the white band.
|
|
ret, initial = cap.read()
|
|
scale = 1280 / initial.shape[1]
|
|
resized_init = cv2.resize(initial, (1280, int(initial.shape[0] * scale)))
|
|
|
|
from youtube_tab_to_pdf import _find_white_tab_strip
|
|
crop_top = 0
|
|
crop_bottom = resized_init.shape[0]
|
|
|
|
cap.set(cv2.CAP_PROP_POS_FRAMES, 500)
|
|
ret, check_frame = cap.read()
|
|
if ret:
|
|
resized_check = cv2.resize(check_frame, (1280, int(check_frame.shape[0] * scale)))
|
|
bounds = _find_white_tab_strip(resized_check)
|
|
if bounds:
|
|
crop_top, crop_bottom = bounds
|
|
# Preserve D.S. al Coda, ┌─ 1., ┌─ 2., and measure numbers drawn in the black abyss!
|
|
crop_top = max(0, crop_top - 60)
|
|
|
|
print(f"Dynamically Cropping to: Y={crop_top} to {crop_bottom}")
|
|
|
|
cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
|
|
frames = []
|
|
idx = 0
|
|
tracker = TemporalTracker(diff_threshold=0.05)
|
|
|
|
while True:
|
|
ret, frame = cap.read()
|
|
if not ret: break
|
|
|
|
frame_resized = cv2.resize(frame, (1280, int(frame.shape[0] * scale)))
|
|
clean_ribbon = frame_resized[crop_top:crop_bottom, :]
|
|
frames.append(clean_ribbon)
|
|
idx += 1
|
|
|
|
cap.release()
|
|
|
|
cv2.imwrite("C:/Users/Certes/.gemini/antigravity/brain/975cea00-dd68-4689-9ee3-f1a2408b4ee6/raw_frame_check.png", frames[30])
|
|
|
|
print(f"Extracted {len(frames)} frames. Running sequential page extraction...")
|
|
try:
|
|
final_chunks = extract_unique_scroll(frames)
|
|
print("DEBUG: final_chunks len =", len(final_chunks))
|
|
if final_chunks:
|
|
print("DEBUG: final_chunks[0].shape =", final_chunks[0].shape)
|
|
cv2.imwrite("C:/Users/Certes/.gemini/antigravity/brain/975cea00-dd68-4689-9ee3-f1a2408b4ee6/debug_chunk_0.png", final_chunks[0])
|
|
|
|
# Save the chunks to artifact directory to literally look at it
|
|
artifact_path = Path(os.environ.get('APPDATA', '')) / '..' / 'Local' / 'Google' / 'AndroidStudio2024.1' # Just using relative artifact manually? No, I'll save it to C:\Users\Certes\.gemini\antigravity\brain\975cea00-dd68-4689-9ee3-f1a2408b4ee6\
|
|
artifact_path = Path(r"C:\Users\Certes\.gemini\antigravity\brain\975cea00-dd68-4689-9ee3-f1a2408b4ee6")
|
|
output_png = artifact_path / "final_check_100_sec.png"
|
|
|
|
generate_long_image(final_chunks, output_png)
|
|
print(f"Saved successful verification image to: {output_png}")
|
|
|
|
if final_chunks:
|
|
generate_pdf(final_chunks, Path("output/shintakarajima_perfect.pdf"))
|
|
print("✨ Successfully generated output/shintakarajima_perfect.pdf ✨")
|
|
else:
|
|
print("Failed to produce rows.")
|
|
except Exception as e:
|
|
import traceback
|
|
traceback.print_exc()
|