| import gradio as gr | |
| from openai import OpenAI | |
| with gr.Blocks() as demo: | |
| def Chat_bot(prompt, history, api_key): | |
| if api_key == '': | |
| output= "Please provide API Key" | |
| else: | |
| try: | |
| messages = [{"role": "system", "content": "you are an English teaching chatbot who replies everything in both English and Korean. but each line should be in English and Korean."}, {"role": "user", "content": prompt}] | |
| client = OpenAI(api_key= api_key ) | |
| response = client.chat.completions.create( | |
| model="ft:gpt-3.5-turbo-0613:personal::8fjNPFEp", | |
| messages=messages, | |
| temperature=0 ) | |
| output = response.choices[0].message.content | |
| except : | |
| output = "Please check your API-Key and try again" | |
| return output | |
| gr.Markdown( | |
| """ | |
| # Fine-tuned English training ChatGPT model. | |
| <img src = "https://upload.wikimedia.org/wikipedia/commons/4/4d/OpenAI_Logo.svg" width=300px> | |
| # Please provide API-Key and start the class | |
| """) | |
| gr.ChatInterface(Chat_bot, | |
| additional_inputs=[ | |
| gr.Textbox( type = 'password', label="Enter your API-Key", placeholder="API-Key", lines=1) | |
| ] | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() | |