Update app.py

app.py

@@ -39,7 +39,9 @@ def load_data():
         st.session_state.users = []
 
 # Load data at the start
-load_data()
+if 'data_loaded' not in st.session_state:
+    load_data()
+    st.session_state.data_loaded = True
 
 # Function to get or create user
 def get_or_create_user():
@@ -100,7 +102,7 @@ with st.sidebar:
                     st.session_state.current_user['name'] = new_name
                     save_data()
                     st.success(f"Name updated to {new_name}")
-                    break
+                    st.rerun()
     
     st.title("Active Users")
     for user in st.session_state.users:
@@ -110,47 +112,42 @@ with st.sidebar:
 st.title("Personalized Real-Time Chat with ArXiv Search")
 
 # Display chat messages
-chat_container = st.container()
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
 # Input for new message
-new_message = st.text_input("Type your message or ArXiv search query:")
-if st.button("Send / Search"):
-    if new_message:
-        timestamp = datetime.now(pytz.utc).strftime('%Y-%m-%d %H:%M:%S %Z')
-        
-        # Check if it's an ArXiv search query
-        if new_message.lower().startswith("arxiv:"):
-            query = new_message[6:].strip()
-            search_results = search_arxiv(query)
-            st.session_state.messages.append({
-                'user': 'ArXiv Bot',
-                'message': f"Search results for '{query}':\n\n{search_results}",
-                'timestamp': timestamp
-            })
-        else:
-            st.session_state.messages.append({
-                'user': st.session_state.current_user['name'],
-                'message': new_message,
-                'timestamp': timestamp
-            })
-        
-        save_data()
-        st.experimental_rerun()
-
-# Function to display chat messages
-def display_messages():
-    for msg in st.session_state.messages:
-        with chat_container.container():
-            st.write(f"**{msg['user']}** ({msg['timestamp']}): {msg['message']}")
+if prompt := st.chat_input("Type your message or ArXiv search query:"):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    # Check if it's an ArXiv search query
+    if prompt.lower().startswith("arxiv:"):
+        query = prompt[6:].strip()
+        with st.chat_message("assistant"):
+            with st.spinner("Searching ArXiv..."):
+                search_results = search_arxiv(query)
+                st.markdown(f"Search results for '{query}':\n\n{search_results}")
+        st.session_state.messages.append({"role": "assistant", "content": f"Search results for '{query}':\n\n{search_results}"})
+    else:
+        # Here you would typically send the message to your chat model and get a response
+        # For this example, we'll just echo the message
+        with st.chat_message("assistant"):
+            st.markdown(f"You said: {prompt}")
+        st.session_state.messages.append({"role": "assistant", "content": f"You said: {prompt}"})
 
-# Display messages
-display_messages()
+    save_data()
+    st.rerun()
 
 # Polling for updates
 if st.button("Refresh Chat"):
-    load_data()
-    st.experimental_rerun()
+    st.rerun()
+
+# Auto-refresh
+if 'last_refresh' not in st.session_state:
+    st.session_state.last_refresh = time.time()
 
-# Auto-refresh (note: this will refresh the entire app)
-time.sleep(5)
-st.experimental_rerun()
+if time.time() - st.session_state.last_refresh > 5:  # Refresh every 5 seconds
+    st.session_state.last_refresh = time.time()
+    st.rerun()