Spaces:
Sleeping
Sleeping
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() | |