Haneesh-dev/ChatGPT-Clone

1import streamlit as st2from langchain.llms import OpenAI3 4OPENAI_API_KEY = "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"5 6llm = OpenAI(openai_api_key=OPENAI_API_KEY)7 8st.title("ChatGPT Clone")9 10if "messages" not in st.session_state:11    st.session_state.messages = []12 13with st.chat_message("assistant"):14    st.write("Hello 👋")15 16for message in st.session_state.messages:17    with st.chat_message(message["role"]):18        st.write(message["content"])19 20prompt = st.chat_input("Enter your message")21 22if(prompt):23    with st.chat_message("user"):24        st.markdown(prompt)25    st.session_state.messages.append({"role": "user", "content": prompt})26 27    response = llm.predict(prompt)28 29    with st.chat_message("assistant"):30        st.markdown(response)31    st.session_state.messages.append({"role": "assistant", "content": response})32