
    MiN	                     &   S SK Jr  S SKrS SKJrJr  S SKJr  S SKJ	r	  S SK
r
S SKrS SKJr  S SKr\" 5       rS rS\	R                   4S\\   S-  4S	 jjrS
 rS\	R                   4S\\   S-  4S jjrS\	R                   4S\\   S-  4S jjrS rg)    )ListN)open_pdfget_page_images)
get_logger)settings)Imagec                 d    [         R                  R                  U 5      R                  S5      S   $ )N.r   )ospathbasenamesplit)r   s    j/var/www/eduai.edurigo.com/storigo/production/storigo_env/lib/python3.13/site-packages/surya/input/load.pyget_name_from_pathr      s'    77D!'',Q//    
page_rangec                 f   [        U 5      n[        U5      nU(       a?  [        U Vs/ s H  nSUs=:*  =(       a    U:  Os  PM     sn5      (       d
   SU 35       eO[        [	        U5      5      n[        X1US9nUR                  5         U Vs/ s H  n[        U 5      PM     nnXh4$ s  snf s  snf )Nr   zInvalid page range: dpi)r   lenalllistranger   closer   )	pdf_pathr   r   doc	last_pagepageimages_namess	            r   load_pdfr"      s    
8
CCIjAjdA))	)jABB 	
":,/	
B %	*+
S#6FIIK3=>:a):E>= B ?s   B)B.c                 n    [         R                  " U 5      R                  S5      n[        U 5      nU/U/4$ )NRGB)r   openconvertr   )
image_pathimagenames      r   
load_imager*   $   s3    JJz"**51Ej)D7TF?r   c                     [         R                  " U 5      nU(       a  UR                  S:X  a
  [        XUS9$ [	        U 5      $ )Npdfr   )filetypeguess	extensionr"   r*   )
input_pathr   r   
input_types       r   load_from_filer2   *   s:     
+Jj**e3
C88*%%r   c                    [         R                  " U 5       Vs/ s H:  nUR                  S5      (       a  M  [         R                  R	                  X5      PM<     nnU Vs/ s H+  n[         R                  R                  U5      (       a  M)  UPM-     nn/ n/ nU H  n[        R                  " U5      n	U	(       a@  U	R                  S:X  a0  [        XUS9u  pUR                  U
5        UR                  U5        M`   [        U5      u  pUR                  U
5        UR                  U5        M     Xg4$ s  snf s  snf ! [        R                   a    [        R                  SU 35         M  f = f)Nr
   r,   r   zCould not load image )r   listdir
startswithr   joinisdirr-   r.   r/   r"   extendr*   PILUnidentifiedImageErrorloggerwarning)folder_pathr   r   
image_nameimage_pathsipr   r!   r   r/   r(   r)   s               r   load_from_folderrA   4   s3   
 **[11J$$S) 	.[-1  
 !,E"277==3D2KEFENN4(	,,5"4=KEMM% LL(.e$T"  =/
 F -- !6tf=>s(   D1#D1 (D6D6;/D;;,E+*E+c                     [        U S5       n[        R                  " U5      nS S S 5        U Vs/ s H  nWU   R                  5       PM     sn$ ! , (       d  f       N2= fs  snf )Nr)r%   jsonloadcopy)	lang_pathr!   f	lang_dictr)   s        r   load_lang_filerJ   Q   sM    	i	IIaL	 
/45utIdO  "u55 
	5s   AA!
A)typingr   r9   surya.input.processingr   r   surya.loggingr   surya.settingsr   r   r-   r   rD   r;   r   	IMAGE_DPIintr"   r*   r2   rA   rJ    r   r   <module>rR      s     
 < $ # 	   	0 7;@R@R 49t#3 " 049K9K& I,& 15(:L:L!#Y-:6r   