a
     ɝh                     @   s   d dl Z d dlZd dlZd dlZd dlZdd Zdd Z	dd Z
dd	 Zed
krdZdZe
e\ZZeee eeeD ]\ZZeee qxdS )    Nc              
   C   s   zNddl }ddl}|j|d}|j| rL|| | td|  W dS W n4 ty } ztd|  W Y d}~dS d}~0 0 dS )z(
    Copy model file to target path
    r   Npre-trained.pth.tarzModel copied to TzError copying model: F)shutilospathjoinexistscopy2print	Exception)
model_path	sour_pathr   r   target_pathe r   7/var/www/html/AI4Kappa/streamlit_scripts/chang_model.py
copy_model   s    
r   c              
   C   sv   z>ddl }|j| d}|j|r<|| td|  W n2 typ } ztd|  W Y d}~n
d}~0 0 dS )z-
    Delete used pre-trained model files
    r   Nr   Removed zError removing model: )r   r   r   r   remover	   r
   )r   r   r   r   r   r   r   clean_model   s    
r   c                 C   sZ   ddl }ddl}||j| d}g }|D ]$} |j| dd }|| q,||fS )z(
    Get model paths and name lists
    r   Nz*-pre-trained.pth.tarz-pre-trained.pth.tar)r   globr   r   basenamesplitappend)r   r   r   model_path_listmodel_name_list
model_namer   r   r   get_model_path)   s    r   c              
      sP  ddl }ddl}ddl z|j| dd}dd|g|_|d  fdd|d< |d|| ||< |jdddd	gf }|jdd
d t	d| d t	|
  z( j| rȈ |  t	d|   W n: ty } z t	d|  d|  W Y d}~n
d}~0 0 |W S  tyJ } z&t	dt|  | W  Y d}~S d}~0 0 dS )z}
    Get prediction results dataframe and convert the third column to powers of 10
    Delete the CSV file after reading
    r   N)headerIDZRANDc                    s    j | d S )Nr   )r   splitext)xr   r   r   <lambda>E       z#get_pre_dataframe.<locals>.<lambda>
      T)inplacezProcessed data for :r   zError removing z: zError in get_pre_dataframe: )pandasnumpyr   read_csvcolumnsapplypoweriloc	set_indexr	   headr   r   r   r
   str	DataFrame)results_csv_pathr   pdnptest_resultsZtest_results_pr   r   r!   r   get_pre_dataframe6   s,    
*r7   __main__z(D:\pycharm\Thermo_Conductivity_APP\modelz"D:\pycharm\Thermo_Conductivity_APP)r   r   r(   r4   	streamlitstr)   r5   r   r   r   r7   __name__r   new_pathr   r   r	   zipr   r   r   r   r   r   <module>   s   )
