File size: 668 Bytes
d35c879
b307974
d35c879
 
b307974
 
 
d35c879
b307974
 
 
 
 
 
 
 
 
 
d35c879
b307974
 
d35c879
 
b307974
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
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()