Create app.py

app.py

@@ -0,0 +1,45 @@
+import random
+import streamlit as st
+
+assistant_icon_path = "chat_icon/assistant.ico"
+user_icon_path = "chat_icon/user01.png"
+
+# 세션 상태에서 approval_state 및 messages 초기화
+if "approval_state" not in st.session_state:
+    st.session_state.approval_state = "require"
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    with st.chat_message(message["role"], avatar=message["avatar"]):
+        st.markdown(message["content"])
+
+# Start with an assistant message
+if not st.session_state.messages:
+    st.chat_message("assistant", avatar=assistant_icon_path).markdown("type yes to start")
+    st.session_state.messages.append({"role": "assistant", "avatar":assistant_icon_path, "content": "type yes to start"})
+
+if prompt := st.chat_input():
+    if st.session_state.approval_state == "require":
+        st.chat_message("user", avatar=user_icon_path).markdown(prompt)
+        st.session_state.messages.append({"role": "user", "avatar":user_icon_path,"content": prompt})
+        if prompt == "yes":
+            st.session_state.approval_state = "approved"
+            st.chat_message("assistant", avatar=assistant_icon_path).markdown("token approved now you can use ai-tom chatbot")
+            st.session_state.messages.append({"role": "assistant", "avatar":assistant_icon_path, "content": "token approved now you can use ai-tom chatbot"})
+        else:
+            st.chat_message("assistant", avatar=assistant_icon_path).markdown("wrong token try again")
+            st.session_state.messages.append({"role": "assistant", "avatar":assistant_icon_path, "content": "wrong token try again"})
+    elif st.session_state.approval_state == "approved":
+        # Display user message in chat message container
+        st.chat_message("user", avatar=user_icon_path).markdown(prompt)
+        st.session_state.messages.append({"role": "user", "avatar":user_icon_path, "content": prompt})
+        
+        response = f"Echo: {prompt}"
+        # Display assistant response in chat message container
+        with st.chat_message("assistant", avatar=assistant_icon_path):
+            st.markdown(response)
+        # Add assistant response to chat history
+        st.session_state.messages.append({"role": "assistant", "avatar":assistant_icon_path, "content": response})
+