o
    WiÀ  ã                   @   s.   d dl mZmZ d dlZdedefdd„ZdS )é    )ÚKEYWORDSÚ
best_labelNÚtextÚreturnc              
      sÜ   | pd  ¡ }dd„ tD ƒ‰ dd„ | ¡ D ƒ}t ¡ D ]\}}|d d… D ]}ˆ |  t||ƒtdt|ƒƒ 7  < q$qt d|¡rHˆ d	  d
7  < t d|¡rVˆ d  d7  < t d|¡rdˆ d  d7  < tˆ ‡ fdd„dS )NÚ c                 S   s   i | ]}|d “qS )r   © )Ú.0Úkr   r   ú4/var/www/html/id_ocr_project_final/utils/classify.pyÚ
<dictcomp>   s    z!classify_text.<locals>.<dictcomp>c                 S   s   g | ]}|  ¡ r|‘qS r   )Ústrip)r   Úlr   r   r
   Ú
<listcomp>   s    z!classify_text.<locals>.<listcomp>é(   é   z\b(\d{4}[ -]?\d{4}[ -]?\d{4})\bÚaadhaaré   z\b(epic|elector|voter)\bÚvoteré   uC   \b(khata|khatian|à¦–à¦¤|à¦–à¦¤à¦¿à¦¯à¦¼à¦¾à¦¨|dag|à¦¦à¦¾à¦—|plot)\bÚlandc                    s   ˆ |  S )Nr   )r	   ©Úscoresr   r
   Ú<lambda>   s    zclassify_text.<locals>.<lambda>)Úkey)	Úlowerr   Ú
splitlinesÚitemsr   ÚmaxÚlenÚreÚsearch)r   ÚtÚlinesr	   Úlabelsr   r   r   r
   Úclassify_text   s   &ÿr$   )Ú
utils.textr   r   r   Ústrr$   r   r   r   r
   Ú<module>   s    