pdf2epub 0.1.2 → 0.1.3

package/vendor/pdf2epub/main.py

@@ -1,118 +0,0 @@
-#!/usr/bin/env python3
-import argparse
-import sys
-from pathlib import Path
-import modules.pdf2md as pdf2md
-import modules.mark2epub as mark2epub
-import torch
-                
-
-
-def main():
-    if torch.cuda.is_available():
-        print("CUDA is available. Using GPU for processing.")
-    elif torch.backends.mps.is_available():
-        print("MPS is available. Using Apple Silicon for processing.")
-    else:
-        print("CUDA is not available. Using CPU for processing.")
-        
-    parser = argparse.ArgumentParser(
-        description='Convert PDF files to EPUB format via Markdown'
-    )
-    parser.add_argument(
-        'input_path',
-        nargs='?',
-        type=str,
-        help='Path to input PDF file or directory (default: ./input/*.pdf)'
-    )
-    parser.add_argument(
-        'output_path',
-        nargs='?',
-        type=str,
-        help='Path to output directory (default: directory named after PDF)'
-    )
-    parser.add_argument(
-        '--max-pages',
-        type=int,
-        default=None,
-        help='Maximum number of pages to process'
-    )
-    parser.add_argument(
-        '--start-page',
-        type=int,
-        default=None,
-        help='Page number to start from'
-    )
-    parser.add_argument(
-        '--skip-epub',
-        action='store_true',
-        help='Skip EPUB generation, only create markdown'
-    )
-    parser.add_argument(
-        '--skip-md',
-        action='store_true',
-        help='Skip markdown generation, use existing markdown files'
-    )
-    
-    args = parser.parse_args()
-    
-    # Get input path
-    input_path = Path(args.input_path) if args.input_path else pdf2md.get_default_input_dir()
-    
-    # Get queue of PDFs to process
-    queue = pdf2md.add_pdfs_to_queue(input_path)
-    print(f"Found {len(queue)} PDF files to process")
-    
-    # Process each PDF
-    failed = []
-    for pdf_path in queue:
-        print(f"\nProcessing: {pdf_path.name}")
-        
-        # Get output directory for this PDF
-        if args.output_path:
-            output_path = Path(args.output_path)
-            markdown_dir = output_path / pdf_path.stem
-        else:
-            markdown_dir = pdf2md.get_default_output_dir(pdf_path)
-            output_path = markdown_dir.parent
-            
-        try:
-            # Check if markdown directory exists when skipping MD generation
-            if args.skip_md:
-                if not markdown_dir.exists():
-                    print(f"Error: Markdown directory not found: {markdown_dir}", file=sys.stderr)
-                    failed.append(pdf_path.name)
-                    continue
-                print(f"Using existing markdown files from: {markdown_dir}")
-                
-            # Convert PDF to Markdown unless skipped
-            if not args.skip_md:
-                print("Converting PDF to Markdown...")
-                pdf2md.convert_pdf(
-                    str(pdf_path),
-                    markdown_dir,
-                    args.max_pages,
-                    args.start_page,
-                )
-            
-            # Convert Markdown to EPUB unless skipped
-            if not args.skip_epub:
-                print("Converting Markdown to EPUB...")
-                mark2epub.convert_to_epub(markdown_dir, output_path)
-                
-        except Exception as e:
-            print(f"Error processing {pdf_path.name}: {str(e)}", file=sys.stderr)
-            failed.append(pdf_path.name)
-            continue
-
-    if failed:
-        print(
-            f"\nFailed to process {len(failed)} of {len(queue)} PDF file(s):",
-            file=sys.stderr,
-        )
-        for name in failed:
-            print(f"  - {name}", file=sys.stderr)
-        sys.exit(1)
-
-if __name__ == '__main__':
-    main()