|
|
import openai |
|
|
import streamlit as st |
|
|
from PIL import Image |
|
|
|
|
|
st.title("SunnyBot") |
|
|
|
|
|
|
|
|
openai.api_key = st.secrets["OPENAI_API_KEY"] |
|
|
grounding = st.secrets["GROUNDING"] |
|
|
|
|
|
image = Image.open('logo.png') |
|
|
|
|
|
st.image(image) |
|
|
|
|
|
if "openai_model" not in st.session_state: |
|
|
st.session_state["openai_model"] = "gpt-4o-mini" |
|
|
|
|
|
if "messages" not in st.session_state: |
|
|
st.session_state.messages = [] |
|
|
|
|
|
for message in st.session_state.messages: |
|
|
with st.chat_message(message["role"]): |
|
|
st.markdown(message["content"]) |
|
|
|
|
|
if prompt := st.chat_input("How can I help you today?"): |
|
|
st.session_state.messages.append({"role": "user", "content": prompt}) |
|
|
with st.chat_message("user"): |
|
|
st.markdown(prompt) |
|
|
|
|
|
with st.chat_message("assistant"): |
|
|
message_placeholder = st.empty() |
|
|
full_response = "" |
|
|
|
|
|
response = openai.responses.create( |
|
|
model=st.session_state["openai_model"], |
|
|
instructions = grounding, |
|
|
input=[ |
|
|
{"role": m["role"], "content": m["content"]} |
|
|
for m in st.session_state.messages |
|
|
], |
|
|
stream=True, |
|
|
) |
|
|
|
|
|
for event in response: |
|
|
if event.type == 'response.output_text.delta': |
|
|
full_response += event.delta.replace('\\$','$').replace('$','\\$') |
|
|
message_placeholder.markdown(full_response + "▌") |
|
|
|
|
|
message_placeholder.markdown(full_response) |
|
|
print(full_response) |
|
|
|
|
|
st.session_state.messages.append({"role": "assistant", "content": full_response}) |
