import os
import openai
import gradio as gr

# openai.api_key = "sk-"  # Replace with your key

OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
USER=os.getenv('USER')
PASS=os.getenv('PASS')

prompt = "您是回答所有用户查询的得力助手,所有问题请用中文回答"

def predict(message, history):
    history_openai_format = []
    for human, assistant in history:
        history_openai_format.append({"role": "system", "content": prompt })
        history_openai_format.append({"role": "user", "content": human })
        history_openai_format.append({"role": "assistant", "content":assistant})
    history_openai_format.append({"role": "user", "content": message})

    response = openai.ChatCompletion.create(
        model='gpt-3.5-turbo',
        messages= history_openai_format,         
        temperature=1.0,
        stream=True
    )

    partial_message = ""
    for chunk in response:
        if len(chunk['choices'][0]['delta']) != 0:
            partial_message = partial_message + chunk['choices'][0]['delta']['content']
            yield partial_message    
   
gr.ChatInterface(predict,
    textbox=gr.Textbox(placeholder="请输入您的问题", container=False, scale=7),
    title="欢迎使用AI机器人",
    examples=["你是谁?", "Am I cool?", "Are tomatoes vegetables?"],
                description="请避免输入有违公序良俗的问题，模型可能无法回答不合适的问题",)
.queue().launch(auth=(USER, PASS))