Spaces:
Sleeping
Sleeping
| # import openai | |
| import streamlit as st | |
| from Models import HFAgent | |
| def initMessages(): | |
| if "messages" not in st.session_state: | |
| st.session_state["messages"] = [{"role": "assistant", "content": "Hello, Sir. How can I assist you today, Sir?"}] | |
| def showMessages(): | |
| for msg in st.session_state.messages: | |
| st.chat_message(msg["role"]).write(msg["content"]) | |
| # def sidebar(): | |
| # with st.sidebar: | |
| # st.title("Settings") | |
| def appendMessage(role, content): | |
| st.session_state.messages.append({"role": role, "content": content}) | |
| def writeMessage(role, content): | |
| st.chat_message(role).write(content) | |
| def generateHFResponse(): | |
| if "agent" not in st.session_state: | |
| st.session_state["agent"] = HFAgent.HFAgent() | |
| agent = st.session_state.agent | |
| new_messages = agent(st.session_state.messages) | |
| return new_messages[-1] | |
| def app(): | |
| # sidebar() | |
| st.title("🗨️ Chatbot with Stark Model") | |
| initMessages() | |
| showMessages() | |
| if prompt := st.chat_input(): | |
| appendMessage("user", prompt) | |
| writeMessage("user", prompt) | |
| msg = generateHFResponse() | |
| appendMessage(msg["role"], msg["content"]) | |
| writeMessage(msg["role"], msg["content"]) | |
| if __name__ == "__main__": | |
| app() |