Fix UI response display issues, increase session timeout, and improve error handling

app.py

@@ -265,7 +265,7 @@ if user_input and not st.session_state.is_processing:
     else:
         st.session_state.is_processing = True
         
-        # Display user message
+        # Display user message immediately
         with st.chat_message("user"):
             st.markdown(validated_input)
             
@@ -430,42 +430,40 @@ if user_input and not st.session_state.is_processing:
                         user_session["conversation"] = conversation
                         session_manager.update_session("default_user", user_session)
                         
-                        # Add to message history - ensure proper format
-                        st.session_state.messages.append({
+                        # Add to message history with proper display
+                        message_data = {
                             "role": "assistant",
                             "content": ai_response,
                             "timestamp": datetime.now().strftime("%H:%M:%S")
-                        })
+                        }
+                        
+                        st.session_state.messages.append(message_data)
                         
-                        # Add feedback buttons
-                        st.divider()
-                        col1, col2 = st.columns(2)
-                        with col1:
-                            if st.button("👍 Helpful", key=f"helpful_{len(st.session_state.messages)}"):
-                                st.success("Thanks for your feedback!")
-                        with col2:
-                            if st.button("👎 Not Helpful", key=f"not_helpful_{len(st.session_state.messages)}"):
-                                st.success("Thanks for your feedback!")
+                        # Force display update
+                        with st.chat_message("assistant"):
+                            st.markdown(ai_response)
                     else:
+                        error_msg = "Sorry, I couldn't process your request. Please try again."
                         st.session_state.messages.append({
                             "role": "assistant",
-                            "content": "Sorry, I couldn't process your request. Please try again.",
+                            "content": error_msg,
                             "timestamp": datetime.now().strftime("%H:%M:%S")
                         })
+                        with st.chat_message("assistant"):
+                            st.markdown(error_msg)
                         
             except Exception as e:
                 user_msg = translate_error(e)
                 response_placeholder.error(f"⚠️ {user_msg}")
                 st.session_state.messages.append({
-                    "role": "assistant",
+                    "role": "assistant", 
                     "content": f"⚠️ {user_msg}",
                     "timestamp": datetime.now().strftime("%H:%M:%S")
                 })
-                
-        # Moved finally block to proper location
-        st.session_state.is_processing = False
-        time.sleep(0.5)  # Brief pause
-        st.experimental_rerun()
+            finally:
+                st.session_state.is_processing = False
+                # Force UI update
+                st.experimental_rerun()
 
 # Add evaluation dashboard tab (separate from chat interface) - ONLY ABOUT TAB NOW
 st.divider()
@@ -500,8 +498,8 @@ with tab1:
     - Performance metrics and analytics
     """)
 
-# Add special command handling
-if user_input and user_input.lower().strip() in ["hello", "hi", "hey"]:
+# Add special command handling for stories
+if user_input and user_input.lower().strip() in ["tell me a story", "tell me a cosmic cat story", "story", "cosmic story"]:
     with st.chat_message("assistant"):
         story = personality.get_space_story()
         st.markdown(f"### 🐱 Cosmic Kitten Story:\n\n{story}")
@@ -511,3 +509,4 @@ if user_input and user_input.lower().strip() in ["hello", "hi", "hey"]:
         "source": "space_story",
         "timestamp": datetime.now().strftime("%H:%M:%S")
     })
+    st.experimental_rerun()  # Add this to refresh the UI

core/session.py

@@ -12,10 +12,10 @@ logger = logging.getLogger(__name__)
 class SessionManager:
     """Manages user sessions and conversation context with optimized operations"""
     
-    def __init__(self, session_timeout: int = 3600):
+    def __init__(self, session_timeout: int = 7200):  # Increased from 3600 to 7200 (2 hours)
         """Initialize session manager
         Args:
-            session_timeout: Session timeout in seconds (default: 1 hour)
+            session_timeout: Session timeout in seconds (default: 2 hours)
         """
         self.session_timeout = session_timeout