import gradio as gr
import os
import requests

# 如果你只打算通过 prompt 来定制机器人的行为，只需要修改这段 prompt 就够了。
prompt = """请将下述文字改成只用emoji来表达，不要输出任何其它字符
然后空两行，给出emoji表达的回复
########


"""


key = os.getenv("baixing_key")
url = "https://gpt.baixing.com/"

# 此处只是最简单的实现单轮对话，qid 和 uid 可以忽略
# 如果要实现多轮对话，可以用本地 sqlite 等方式存储数据，并根据 uid 找到对话历史。
def chat(p, qid, uid): 
    result = requests.get(url, params={"p": prompt + p, "k": key}).json()['data']
    return ["text", result]

iface = gr.Interface(fn=chat, 
                     inputs=["text", "text", "text"], 
                     outputs=["text", "text"],
                     description="""这是一个极其简单的示范程序，把你的问话翻译成 emoji 并只用 emoji 来回答。
[参考文档](https://huggingface.co/spaces/baixing/hackathon_test/blob/main/bot-api.md)  [对话测试](https://huggingface.co/spaces/baixing/hackathon_test)  [Q & A](https://huggingface.co/spaces/baixing/hackathon_test/blob/main/qna.md)
                     """
                    )
iface.launch()