Code Editor for app.py

import streamlit as st
import openai

# Set your OpenAI API key
openai.api_key = 'YOUR_OPENAI_API_KEY'

def generate_response(prompt):
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[{"role": "user", "content": prompt}]
    )
    return response['choices'][0]['message']['content']

# Streamlit App
st.title("ChatGPT Chatbot")

# Create a container to hold chat messages
if 'messages' not in st.session_state:
    st.session_state.messages = []

# Display messages
for msg in st.session_state.messages:
    st.chat_message(msg['role']).write(msg['content'])

# Input from user
user_input = st.chat_input("Type your message...")
if user_input:
    # Append user message to the messages list
    st.session_state.messages.append({"role": "user", "content": user_input})

    # Generate response from ChatGPT
    bot_response = generate_response(user_input)
    
    # Append bot response to the messages list
    st.session_state.messages.append({"role": "assistant", "content": bot_response})