#!/bin/bash

# Download Videos from NotebookLM - Working Script

STORAGE_PATH="/home/ashraffarid2010/jadwaai.com/nlm_sessions/account_10/storage_state.json"
NOTEBOOK_ID="0f38d000-7497-442c-9b67-8773b02a558b"
OUTPUT_DIR="/home/ashraffarid2010/jadwaai.com/app/static/uploads/notebooks/35"

echo "================================================================================"
echo "DOWNLOAD VIDEOS FROM NOTEBOOKLM"
echo "================================================================================"
echo "Storage: $STORAGE_PATH"
echo "Notebook ID: $NOTEBOOK_ID"
echo "Output Directory: $OUTPUT_DIR"
echo ""

# Ensure output directory exists
mkdir -p "$OUTPUT_DIR"

# Change to directory where we want to download
cd "$OUTPUT_DIR"

# Set NOTEBOOKLM_HOME and export it
export NOTEBOOKLM_HOME="$(dirname "$STORAGE_PATH")"

echo "Step 1: Setting notebook context..."
notebooklm --storage "$STORAGE_PATH" use "$NOTEBOOK_ID"

echo ""
echo "Step 2: Downloading all videos..."
notebooklm --storage "$STORAGE_PATH" download video --all . --force

echo ""
echo "Step 3: Checking downloaded files..."
echo "Video files in $OUTPUT_DIR:"
ls -lah *.mp4 2>/dev/null || echo "No video files found"

echo ""
echo "Step 4: Renaming files to match expected format..."
# Rename files to match expected naming convention
count=1
for video in *.mp4; do
    if [ -f "$video" ]; then
        new_name="video_${count}.mp4"
        echo "Renaming: $video -> $new_name"
        mv "$video" "$new_name"
        count=$((count + 1))
    fi
done

echo ""
echo "Final video files:"
ls -lah video_*.mp4 2>/dev/null || echo "No video files found"

echo ""
echo "================================================================================"
echo "DOWNLOAD COMPLETE"
echo "================================================================================"