BhaskarChatBot / app.py
Bhaskar2611's picture
Update app.py
b307974 verified
raw
history blame
668 Bytes
import os
import openai
import gradio as gr
# Set OpenAI API Key
OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
openai.api_key = OPENAI_API_KEY
def get_text_response(user_message, history):
# Call OpenAI GPT model
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": user_message},
]
)
return response['choices'][0]['message']['content']
# Create a Gradio chat interface
demo = gr.Interface(fn=get_text_response, inputs="text", outputs="text")
if __name__ == "__main__":
demo.launch()