File size: 1,277 Bytes
52df484 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import os
from dotenv import load_dotenv
from openai import Client
import gradio as gr
# Load environment variables from .env file
load_dotenv()
# Get the value of the environment variable
api_key = os.getenv('OPENAI_API_KEY')
if not api_key:
raise ValueError("API key not found. Please set your OPENAI_API_KEY in the environment.")
# Initialize the OpenAI client
client = Client(api_key=api_key)
def chatbot(input):
"""Chatbot function using the OpenAI API"""
if input:
messages = [
{"role": "system", "content": "You are a helpful and kind AI Assistant."},
{"role": "user", "content": input},
]
response = client.chat.complete(
engine="text-davinci-003",
messages=messages,
max_tokens=150,
n=1,
stop=None,
temperature=0.7,
)
reply = response.choices[0].text
return reply
input_text = gr.Textbox(lines=7, label="Chat with AI")
output_text = gr.Textbox(label="Reply")
interface = gr.Interface(fn=chatbot, inputs=input_text, outputs=output_text, title="AI Chatbot",
description="Ask anything you want", theme="compact")
interface.launch(share=True)
|