|
|
|
|
|
import os |
|
import sys |
|
import cohere |
|
from streamlit.web import cli as stcli |
|
from streamlit import runtime |
|
import streamlit as st |
|
|
|
COHERE_CLIENT=st.secrets["COHERE_CLIENT"] |
|
COHERE_MODEL=st.secrets["COHERE_MODEL"] |
|
BASE_PROMPT="Generate Mandarin by HSK level and English keywords\nenglish: light\nhsk: 1\nmandarin: 所有的灯都闪了一会儿。\npinyin: suǒ yǒu de dēng dōu shǎn le yī huìr。\n--\nenglish: window\nhsk: 1\nmandarin: 所有的窗户都用砖堵住了。\npinyin: suǒ yǒu de chuāng hu dōu yòng zhuān dǔ zhù le。\n--\nenglish: airplane\nhsk: 1\nmandarin: 一架飞机在机场着陆。\npinyin: yī jià fēi jī zài jī chǎng zhuó lù。\n--\nenglish: styles\nhsk: 2\nmandarin: 这些式样均可改动以适应个人不同的爱好.\npinyin: zhè xie shì yàng jūn kě gǎi dòng yǐ shì yìng gè rén bù tóng de ài hào.\n--\nenglish: food.\nhsk: 2\nmandarin: 他以自己的名字命名的冷冻食品成了名牌。\npinyin: tā yǐ zì jǐ de míng zi mìng míng de lěng dòng shí pǐn chéng le míng pái。\n--\nenglish: glasses.\nhsk: 2\nmandarin: 印刷字体太小，我不带眼镜就看不清。\npinyin: yìn shuà zì tǐ tài xiǎo， wǒ bù dài yǎn jìng jiù kàn bù qīng。\n--\nenglish: practice\nhsk: 3\nmandarin: 实践与理论一样重要，但是我们易于珍视后者而轻视前者。\npinyin: shí jiàn yù lǐ lùn yí yàng zhòng yào， dàn shì wǒ men yì yú zhēn shì hǒu zhě ér qīng shì qián zhě。\n--\nenglish: sluggish\nhsk: 3\nmandarin: 很多队员渐渐变得动作迟缓了。这支队所需的是一些新鲜血液。\npinyin: hěn duō duì yuán jiàn jiàn biàn de dòng zuò chí huǎn le。 zhè zhī duì suǒ xū de shì yī xiē xīn xiān xuè yè。\n--\nenglish: medicine\nhsk: 3\nmandarin: 医术在于，当大自然为病者医治疾病的一段期间内，逗得病人高兴。\npinyin: yī zhú zài yú， dàng dà zì rán wèi bìng zhě yī zhì jí bìng de yī duàn qī jiān nèi， dòu dé bìng rén gāo xìng。\n--\n--\nenglish: princess\nhsk: 4\nmandarin: 公主光彩照人地穿着一身雪白的新礼服来了。\npinyin: gōng zhǔ guāng cǎi zhào rén de chuān zhuó yī shēn xuě bái de xīn lǐ fú lái le。\n--\nenglish: parking space\nhsk: 4\nmandarin: 我认为他侵占我的停车位肯定是有意和我作对。\npinyin: wǒ rèn wéi tā qīn zhàn wǒ de tíng chē wèi kěn dìng shì yǒu yì huò wǒ zuò duì。\n--\nenglish: discrepancy\nhsk: 4\nmandarin: 这件事的两种说法有很大出入.\npinyin: zhè jiàn shì de liǎng zhòng shuō fa yǒu hěn dà chū rù.\n--\n" |
|
|
|
|
|
def generate_texts(keyword, hsk): |
|
print(f"keyword: {keyword} and hsk: {hsk}") |
|
co = cohere.Client(COHERE_CLIENT) |
|
response = co.generate( |
|
model=COHERE_MODEL, |
|
prompt=BASE_PROMPT + "english: {keyword}\nhsk: {hsk}\nmandarin:".format(keyword=keyword, hsk=hsk), |
|
max_tokens=200, |
|
temperature=0.5, |
|
k=0, |
|
p=0.75, |
|
frequency_penalty=0.5, |
|
presence_penalty=0.5, |
|
stop_sequences=["--"], |
|
return_likelihoods='NONE') |
|
|
|
print('Prediction: {}'.format(response.generations[0].text)) |
|
return response.generations[0].text |
|
|
|
|
|
def main(): |
|
st.title("Chinese Sentence Generator with English Keyword and HSK Level") |
|
|
|
form = st.form(key="user_settings") |
|
with form: |
|
en_keyword = st.text_input("English Keyword", |
|
placeholder="Enter an English keyword, eg. home, key, etc", |
|
key = "en_keyword") |
|
hsk_level = st.select_slider("HSK Level", |
|
options=["1", "2", "3", "4"], |
|
key = "hsk_level") |
|
|
|
generate_button = form.form_submit_button("Generate Mandarin Sentence and Pinyin") |
|
|
|
if generate_button: |
|
cohere_txts = generate_texts(en_keyword, hsk_level) |
|
st.markdown("##### " + "Generated Mandarin Sentence: ") |
|
st.write(cohere_txts) |
|
|
|
st.write("Check out the prediction on [Google Translate](https://www.google.com/search?q=google+translate&oq=google+translate&aqs=chrome.0.69i59j69i64j0i271l2j69i60.5320j0j1&sourceid=chrome&ie=UTF-8)") |
|
|
|
if __name__ == "__main__": |
|
if runtime.exists(): |
|
main() |
|
else: |
|
sys.argv = ["streamlit", "run", sys.argv[0]] |
|
sys.exit(stcli.main()) |
