Chatbot.py

import streamlit as st
from openai import AzureOpenAI
from openai import OpenAI
from os import environ
import dotenv
from dotenv import load_dotenv
load_dotenv()  # Load environment variables from .env
# api_key = environ.get('AZURE_OPENAI_API_KEY')  # Safely get the API key

# print(api_key)
st.title("Awesome Chatbot")
st.caption("Powered by INFO-5940")   


if "messages" not in st.session_state:
    st.session_state["messages"] = [{"role": "assistant", "content": "Hello! How can I help you today?"}]

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

if prompt := st.chat_input():

    #client = OpenAI(api_key=environ['OPENAI_API_KEY'])
    # client = AzureOpenAI(
    #     azure_deployment="gpt-4o",
    #     temperature=0.2,
    #     api_version="2023-06-01-preview",
    #     max_tokens=None,
    #     timeout=None,
    #     max_retries=2,
    # )
        
    print(environ)
    client = AzureOpenAI(
    api_key= environ['AZURE_OPENAI_API_KEY'],
    api_version= "2023-03-15-preview",
    azure_endpoint=environ['AZURE_OPENAI_ENDPOINT'],
    azure_deployment=environ['AZURE_OPENAI_MODEL_DEPLOYMENT'],
    )
    st.session_state.messages.append({"role": "user", "content": prompt})
    st.chat_message("user").write(prompt)

    with st.chat_message("assistant"):
        stream = client.chat.completions.create(model="gpt-4o", 
                                                messages=st.session_state.messages,
                                                stream=True)
        response = st.write_stream(stream)

    st.session_state.messages.append({"role": "assistant", "content": response})