Upload 2 files

main.py

@@ -0,0 +1,82 @@
+from streamlit_server_state import server_state, server_state_lock
+import streamlit as st
+
+if "messages" not in server_state:
+    server_state["messages"] = []
+
+if "NORMAL_PASSWORD_STR" not in server_state:
+    server_state["NORMAL_PASSWORD_STR"] = "123"
+    
+if "ADMIN_PASSWORD_STR" not in server_state:
+    server_state["ADMIN_PASSWORD_STR"] = "991152"
+
+if "LAST_IMAGE" not in server_state:
+    server_state["LAST_IMAGE"] = None
+
+def adminView():
+    st.write(f"Current Password: {server_state['NORMAL_PASSWORD_STR']}")
+    new_password = st.sidebar.text_input("New Password")
+    confirm_password = st.sidebar.text_input("Confirm Password")
+    change_btn = st.sidebar.button("Change")
+
+    if change_btn:
+        if new_password == confirm_password:
+            server_state["NORMAL_PASSWORD_STR"] = new_password
+            st.success("Password Changed")
+        else:
+            st.error("Passwords Don't Match")
+
+def displayMessages():
+    st.markdown("---")
+    st.header("Messages")
+    col1, col2 = st.columns(2)
+    message_str = ""
+
+    if len(server_state["messages"]) > 0:
+        for message in reversed(server_state["messages"]):
+            message_str += f"{message['MESSAGE']} by {message['NAME']}\n"
+        
+        col1.text_area("Chats", message_str, height=200)
+
+    if server_state["LAST_IMAGE"] is not None:
+        col2.image(server_state["LAST_IMAGE"]["IMAGE"], f"by - {server_state['LAST_IMAGE']['NAME']}")
+    
+def normalView():
+    name = st.sidebar.text_input("Name")
+    clear_chats = st.sidebar.button("Clear Chats")
+
+    col1, col2 = st.columns(2)
+    message = col1.text_area("Message",height=240)
+    send_btn = col1.button("Send")
+
+    camera_input = col2.camera_input("Take Pick")
+    send_image_btn = col2.button("Send Image")
+
+    if send_btn:
+        if message != "":
+            with server_state_lock["messages"]:
+                server_state["messages"] = server_state["messages"] + [{"NAME": name, "MESSAGE": message}]
+
+    
+    if send_image_btn:
+        if camera_input is not None:
+            with server_state_lock["LAST_IMAGE"]:
+                server_state["LAST_IMAGE"] = {"NAME": name, "IMAGE":camera_input}
+
+    
+    if clear_chats:
+        with server_state_lock["messages"]:
+            server_state["messages"] = []
+        with server_state_lock["LAST_IMAGE"]:
+            server_state["LAST_IMAGE"] = None
+    
+    displayMessages()
+
+
+#
+password_input = st.sidebar.text_input("Password")
+
+if password_input == server_state["NORMAL_PASSWORD_STR"]:
+    normalView()
+elif password_input == server_state["ADMIN_PASSWORD_STR"]:
+    adminView()