Spaces:
Sleeping
Sleeping
import openai | |
import gradio as gr | |
import os | |
def chat_with_openai(prompt, history=[]): | |
messages = [{"role": "system", "content": "You are a helpful assistant."}] | |
for user_msg, bot_msg in history: | |
messages.append({"role": "user", "content": user_msg}) | |
messages.append({"role": "assistant", "content": bot_msg}) | |
messages.append({"role": "user", "content": prompt}) | |
response = openai.ChatCompletion.create( | |
model="gpt-4", # Change to "gpt-4" if needed | |
messages=messages | |
) | |
return response["choices"][0]["message"]["content"] | |
def chatbot_interface(user_input, chat_history): | |
response = chat_with_openai(user_input, chat_history) | |
chat_history.append((user_input, response)) | |
return response, chat_history | |
# Create a Gradio interface | |
demo = gr.ChatInterface(fn=chatbot_interface) | |
if __name__ == "__main__": | |
demo.launch() | |