import openai
import os
from dotenv import load_dotenv, find_dotenv

_ = load_dotenv(find_dotenv()) # read local .env file

openai.api_key  = os.getenv('OPENAI_API_KEY')


def get_completion_from_messages(messages, 
                                 model="gpt-3.5-turbo", 
                                 temperature=0, 
                                 max_tokens=500):
    response = openai.chat.completions.create( # Use openai.chat.completions.create instead of openai.ChatCompletion.create
        model=model,
        messages=messages,
        temperature=temperature, 
        max_tokens=max_tokens,
    )
    return response.choices[0].message.content
import gradio as gr
def generate(input, slider):
    messages =  [
        {'role':'user',
         'content': f"{input}"},
    ]
    
    output = get_completion_from_messages(messages)
    return output

demo = gr.ChatInterface(
    fn=generate,
    chatbot=gr.Chatbot(height=300),
    textbox=gr.Textbox(placeholder="Ask me a yes or no question", container=False, scale=7),
    title="Your Helper CHAT Bot",
    description="Ask 'Yes Man' any question",
    theme="soft",
    examples=["Hello", "Am I cool?", "Are tomatoes vegetables?"],
    cache_examples=True,
    #retry_btn=None,
   #undo_btn="Delete Previous",
    #clear_btn="Clear",
).launch()