app.py

import streamlit as st
import requests
import time

API_KEY = st.secrets.HugApiKey.key
API_URL = "https://api-inference.huggingface.co/models/huranokuma/es"
headers = {"Authorization": "Bearer "+API_KEY}

st.set_page_config(
     page_title="ESを書くAI",
     page_icon="🤖",
     initial_sidebar_state="expanded",
 )

def query(payload):
	response = requests.post(API_URL, headers=headers, json=payload)
	return response.json()

def main():
  st.title("AIによる自動ES作成")  

  min_length = st.slider(label='最小文字数(最大トークン数)',
                  min_value=50,
                  max_value= 500,
                  value=100,
                  )
  
  top_p = st.number_input(label='top p : (上位 P %の単語から選んできます)',
                  min_value=0.00,
                  max_value=1.00,
                  value=0.95
  )

  top_k = st.slider(label='top k : (上位k個の文章を保持します)',
                min_value=1,
                max_value=1000,
                value=500
)
  temperature = st.number_input(label='temperature : (高いほどランダム性が上がり、低いほど同じ結果が出力されます)',
                min_value=0.01,
                max_value=100.0,
                value=1.00
)
  

  prompt_text = st.text_area(
        label='ESの書き出しの文章', 
        value='私が御社を志望した理由は'
  )

  progress_num = 0
  status_text = st.empty()
  progress_bar = st.progress(progress_num)
  process_text = st.empty()

  process_text.text("ESの書き出しや、会社の質問を入力してください。それに続く文章を生成します。")

  if st.button('文章生成'):

    progress_num = 30 
    progress_bar.progress(progress_num)
    status_text.text(f'Progress: {progress_num}%')
    process_text.text("文章を生成しています...これには時間がかかるかもしれません。")
    start = time.time()

    # APIを使ってHuggingfaceから文章を取ってくる。
    output = query({"inputs": prompt_text,
                "parameters": {
                               "max_length":500,
                               "min_length":min_length,
                               "top_p":top_p,
                               "top_k":top_k,
                               "temperature":temperature,
                               },
                "options":{
                    "wait_for_model": True,
                }
                })

    process_text.text("ESの生成が終了しました。")
    elapsed_time =round(time.time()-start,2)
    st.info(f'生成結果 : 経過時間{elapsed_time}秒')
    progress_num = 100
    status_text.text(f'Progress: {progress_num}%')
    st.write(output[0]['generated_text'])
    progress_bar.progress(progress_num)

if __name__ == '__main__':
  main()