#compdef ai_research_os.py

_airos() {
    local -a opts
    opts=(
        '--root[Root folder for your research OS]:folder:_files -/'
        '--category[Folder under root to place P-Note]:folder:( 00-Radar 01-Foundations 02-Models 03-Applications 04-Evaluation 05-Tools 06-Theory 07-Architecture 08-Training 09-Data 10-Applications 11-Future-Directions )'
        '--tags[Comma-separated tags]:tags:'
        '--concept-dir[Folder under root to place C-Notes]:folder:'
        '--comparison-dir[Folder under root to place M-Notes]:folder:'
        '--max-pages[Max PDF pages to extract]:number:'
        '--pdf[Path to a local PDF]:file:_files -g "*.pdf"'
        '--ocr[Enable OCR fallback]'
        '--ocr-lang[Tesseract language]:lang:'
        '--ocr-zoom[OCR render zoom]:zoom:'
        '--no-pdfminer[Disable pdfminer fallback]'
        '--structured[Use structured PDF extraction]'
        '--ai[Use AI to draft-fill P-Note sections]'
        '--ai-cnote[AI-fill all C-Notes from existing P-Notes]'
        '--ai-max-papers[Max P-notes to feed per C-note]:number:'
        '--api-key[LLM API key]:key:'
        '--model[LLM model name]:model:'
        '--base-url[OpenAI-compatible base url]:url:'
        '--ai-max-chars[Max chars of extracted text sent to AI]:number:'
        '*:arXiv URL/ID or DOI:'
    )
    _describe 'options' opts
} && _airos
