#!/usr/bin/python
# -*- coding: utf-8 -*-
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())