import openai
import gradio as gr

# Getting responses using the OpenAI API
def response_chatgpt(api_key, message):
  # OPENAI API KEY
  openai.api_key = api_key
  prompt = (f"{message}")
  response = openai.Completion.create(
      engine="text-davinci-003",
      prompt=prompt,
      max_tokens=1024
  )
  # Displaying the answer on the screen
  result = response["choices"][0]["text"]
  return result
      
# User input
chatbot = gr.Interface(
    fn=response_chatgpt, 
    inputs=["text", "text"],
    outputs="text",
)
chatbot.launch()