o "e@s$ddlZddlZddlZddZdS)Nc Ks|d}|d}g}g}t|dD]L\}}tjdr'|dd|d} n|d} d| d } |d d vr9d nd } |t| d|dpEd| |d|doZ|dupZ||vpZ|||vdq|rft|} nt|d} |ddkrv| |d<|ddurd|d<|d} |ddkrd}n t| |d}|ddkrn|dkrt t ||dD]\}}|t j di|qWdn1swYnz||dkrt t ||dD]\}}|t j di|qWdn1swYnF|dkr~t -tt ||dD]\}\}}|| dkr%q|t j di|qWdn 1s=wYt -tt ||dD]\}\}}|| dkr_qP|t j di|qPWdn 1swwYn|dkrBt /tt ||dD]\}\}}|d| dkrq|t j di|qWdn 1swYt 8tt ||dD]&\}\}}|d| dks|d| dkrq|t j di|qWdn 1swYt /tt ||dD]\}\}}|d| dkr#q|t j di|qWdn 1s;wYn|dkrIt /tt ||dD]\}\}}|d| dkrfqU|t j di|qUWdn 1s~wYt 8tt ||dD]&\}\}}|d| dks|d| dkrq|t j di|qWdn 1swYt 8tt ||dD]&\}\}}|d| dks|d| dkrq|t j di|qWdn 1swYt /tt ||dD]\}\}}|d| dkr,q|t j di|qWdn 1sDwYt %t j ||d |dp[dd}t j |d|dphdd}Wdn 1svwY|||fS)Nvisible_models all_models model_statesZDEBUG_MODEL_LOCK base_modelz : inference_serverzh2oGPT [] gradio_size)smalllargemediumZ chatsmallheightishow_copy_button model_lock)label elem_classesr min_widthrvisiblemodel_lock_columnsr)rrrF) enumerateosenvirongetappenddictlenmathceilgrRowzipZChatbot) output_label0output_label0_model2kwargsrr text_outputsZ chat_kwargsZmodel_state_lockiZmodel_state_lock model_nameZ output_labelrZ len_visiblencolsnrowsZ chat_kwargs1Zmii text_output text_output2rr,/home/jon/h2ogpt/gradio_utils/prompt_form.py make_chatbotss                    $    $ $    r2)rr#gradior%r2rrrr1s