huggingface/Model_Cards_Writing_Tool
125
1a
2 � lcE � @ s� d dl Zd dlmZmZ d dlmZmZ d dlmZm Z m3Z4 d dlmZmZ d dl m Z m
Z
dd� Zdd � Zd5d � Zd d
� Zdd� Zdd� Zdd� ZdS )� N)�persist�load_widget_state)�CardData� ModelCard)� tag_checker� listToString� to_markdown)� Environment�FileSystemLoaderc C s" zt | � W dS Y dS 0 d S )NTF)�float)�value� r
�O/Users/ezi/Desktop/HF/Prompt_writing/live/Model_Cards_Writing_Tool/middleMan.py�is_float s6 r c ( C sX t td�dd�} | �tjj�}|jtjd tjd tjd tjd tjd tjd tjd7 tjd tjd tjd
tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd tjd! tjd" tjd# tjd$ tjd% tjd& tjd' tjd( tjd) d*�&S )+N�.T)�loader�8autoescape�9model_name� languages�model_description�Model_developers� Shared_by�license�Parent_Model_url�10Direct_Use�Downstream_UsezOut-of-Scope_Use�Model_Limits_n_Risks�Recommendations� Model_examin�Speeds_Sizes_Times�Model_hardware�11hours_used�Model_cloud_provider�Model_cloud_region�Model_c02_emitted� APA_citation�bibtex_citation�
training_Data�model_preprocessing� Model_specs�compute_infrastructure�technical_specs_software�Glossary� More_info� the_authors�Model_card_contact� Model_how_to�12github_url� paper_url�blog_url� Testing_Data�Factors�
Model_Results�Metrics)&�model_id�languageZthe_model_descriptionZ13developers� shared_byZ
model_licenseZparent_model_linkZ14direct_useZdownstream_useZout_of_scope_useZbias_risks_limitationsZbias_recommendationsZmodel_examinationZspeeds_sizes_timesZhardwarer! Zcloud_providerZ cloud_regionZ co2_emittedZcitation_bibtexZ citation_apaZ
training_dataZ
preprocessingZ model_specsr* ZsoftwareZglossaryZmore_informationZmodel_card_authorsZmodel_card_contactZget_started_codeZ repo_linkZ15paper_linkZ blog_linkZ testing_dataZtesting_factors�resultsZtesting_metrics)r r16 � get_template�st�
session_state�markdown_upload�render)�env�tempr
r
r �parse_into_jinja_markdown s6 (�rC c C s� d}t jjdkrh d�}n t jjdkr0h d�}nh d�}|D ]}|| kr<d}q<|dkrrt|| |�}d| d }|S t|| |�}|}|S d S )NT> Z Model_carbon�Model_card_authorsr, �17Model_Eval�18Model_uses�Model_details_text�19Model_cite> r/ r, ZModel_trainingr r0 rG ZCitation> r, rF r0 rG rE Fz.<details> <summary> Click to expand </summary>z20</details>)r= r> �21legal_view�researcher_view� out_text_out)�22page_state�
not_code_pullZ text_passedZnot_important_sectionZuser_view_collapser Z text_return�out_textr
r
r �23apply_view@ s 24 25 rO c C s | dkrt |�}|S |}|S d S )NT)Z26extract_it)rM rL rN r
r
r rK _ s27 rK c C s0 dt j_t jdd|t|�|d�}t|�t j_|S )NTzEnter some texti� )�heightr �key�help)r= r> � check_box� text_arear rL )rL � help_textrN Zvariable_output_promptr
r
r �
writingPromptg s � rV c C s | }t |||�}d�|�}|S )N� )r �join)�current_templateZ start_tagZend_tagZcurrent_Card_markdownZextracted_how_torN r
r
r �extract_section s 28rZ c C s d S )Nr
r
r
r
r �main� s r[ )� streamlitr= r r �29modelcardsr r �markdownTagExtractr r r �jinja2r r30 r rC rO rK rV rZ r[ r
r
r
r �<module> s -