AI_App / app.py
resolverkatla's picture
.
b409e48
raw
history blame
894 Bytes
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()