Fix UI response display and extend session timeout

app.py

@@ -269,7 +269,7 @@ def validate_user_input(text):
     
     return True, text.strip()
 
-# Chat input - COMPLETELY REWRITTEN VERSION
+# Chat input - FIXED RESPONSE DISPLAY VERSION
 user_input = st.chat_input("Type your message here...", disabled=st.session_state.is_processing)
 
 # Process message when received
@@ -294,86 +294,82 @@ if user_input and not st.session_state.is_processing:
         "timestamp": timestamp
     })
     
-    # Process AI response
-    with st.chat_message("assistant"):
-        status_placeholder = st.empty()
-        response_placeholder = st.empty()
+    # Show processing status
+    status_container = st.empty()
+    response_container = st.empty()
+    
+    try:
+        status_container.info("🔄 Processing your request...")
+        
+        # Get conversation history from session
+        user_session = session_manager.get_session("default_user")
+        conversation_history = user_session.get("conversation", []).copy()
+        
+        # Add the current user message to history for context
+        conversation_history.append({"role": "user", "content": clean_input})
+        
+        # Try Ollama first
+        status_container.info("🦙 Contacting Ollama...")
         
         try:
-            status_placeholder.info("🔄 Processing your request...")
-            
-            # Get conversation history from session
-            user_session = session_manager.get_session("default_user")
-            conversation_history = user_session.get("conversation", []).copy()
-            
-            # Log conversation history for debugging
-            logger.info(f"Conversation history length: {len(conversation_history)}")
-            
-            # Add the current user message to history for context
-            conversation_history.append({"role": "user", "content": clean_input})
-            
-            # Try Ollama first
-            status_placeholder.info("🦙 Contacting Ollama...")
-            logger.info("Attempting Ollama connection...")
+            # Use the OllamaProvider directly with proper configuration
+            ollama_provider = OllamaProvider(st.session_state.selected_model)
+            ai_response = ollama_provider.generate(clean_input, conversation_history)
             
-            try:
-                # Use the OllamaProvider directly with proper configuration
-                ollama_provider = OllamaProvider(st.session_state.selected_model)
-                logger.info(f"Ollama provider created with model: {st.session_state.selected_model}")
-                logger.info(f"Ollama host: {ollama_provider.host}")
-                
-                ai_response = ollama_provider.generate(clean_input, conversation_history)
-                logger.info(f"Ollama response received: {type(ai_response)} - {str(ai_response)[:100] if ai_response else 'None'}")
+            if ai_response and ai_response.strip():
+                # DISPLAY THE RESPONSE IN THE CONTAINER (not using placeholders)
+                with st.chat_message("assistant"):
+                    st.markdown(ai_response)
+                status_container.success("✅ Response received!")
+            else:
+                # DISPLAY ERROR RESPONSE
+                with st.chat_message("assistant"):
+                    st.warning("⚠️ Received empty response from Ollama")
+                ai_response = "I received your message but couldn't generate a proper response."
                 
-                if ai_response and ai_response.strip():
-                    response_placeholder.markdown(ai_response)
-                    status_placeholder.success("✅ Response received!")
-                else:
-                    status_placeholder.warning("⚠️ Empty response from Ollama")
-                    ai_response = "I received your message but couldn't generate a proper response. Please try again."
-                    response_placeholder.markdown(ai_response)
-                    
-            except Exception as ollama_error:
-                error_message = str(ollama_error)
-                logger.error(f"Ollama error: {error_message}")
-                status_placeholder.error(f"❌ Ollama error: {error_message[:100]}...")
-                ai_response = f"Error contacting Ollama: {error_message[:100]}..."
-                response_placeholder.error(ai_response)
-            
-            # Save response to session and message history
-            if ai_response:
+        except Exception as ollama_error:
+            error_message = str(ollama_error)
+            # DISPLAY ERROR
+            with st.chat_message("assistant"):
+                st.error(f"❌ Ollama error: {error_message[:100]}...")
+            ai_response = f"Error: {error_message[:100]}..."
+        
+        # Save response to session and message history
+        if ai_response:
+            try:
                 # Update conversation history in session
-                try:
-                    conversation = user_session.get("conversation", []).copy()
-                    conversation.append({"role": "user", "content": clean_input})
-                    conversation.append({"role": "assistant", "content": str(ai_response)})
-                    
-                    # Update session with new conversation
-                    update_result = session_manager.update_session("default_user", {"conversation": conversation})
-                    logger.info(f"Session update result: {update_result}")
-                    
-                except Exception as session_error:
-                    logger.error(f"Session update error: {session_error}")
+                conversation = user_session.get("conversation", []).copy()
+                conversation.append({"role": "user", "content": clean_input})
+                conversation.append({"role": "assistant", "content": str(ai_response)})
                 
-                # Add to message history
+                # Update session with new conversation
+                update_result = session_manager.update_session("default_user", {"conversation": conversation})
+                
+                # Add to message history (this was missing!)
                 st.session_state.messages.append({
                     "role": "assistant",
                     "content": str(ai_response),
                     "timestamp": timestamp
                 })
                 
-        except Exception as e:
-            error_msg = f"System error: {str(e)}"
-            logger.error(f"Chat processing error: {error_msg}")
-            response_placeholder.error(error_msg)
-            st.session_state.messages.append({
-                "role": "assistant",
-                "content": error_msg,
-                "timestamp": timestamp
-            })
-        finally:
-            st.session_state.is_processing = False
-            st.experimental_rerun()
+            except Exception as session_error:
+                logger.error(f"Session update error: {session_error}")
+                
+    except Exception as e:
+        error_msg = f"System error: {str(e)}"
+        logger.error(f"Chat processing error: {error_msg}")
+        # DISPLAY SYSTEM ERROR
+        with st.chat_message("assistant"):
+            st.error(error_msg)
+        st.session_state.messages.append({
+            "role": "assistant",
+            "content": error_msg,
+            "timestamp": timestamp
+        })
+    finally:
+        st.session_state.is_processing = False
+        time.sleep(0.1)  # Small delay to ensure UI updates
+        st.experimental_rerun()
 
 # Add evaluation dashboard tab (separate from chat interface) - ONLY ABOUT TAB NOW
 st.divider()
@@ -420,3 +416,20 @@ if user_input and user_input.lower().strip() in ["tell me a story", "tell me a c
     })
     st.session_state.is_processing = False
     st.experimental_rerun()
+
+# Simple test to verify the fix works
+def test_response_display():
+    """Test function to verify response display works"""
+    test_response = "This is a test response to verify the display fix is working correctly."
+    with st.chat_message("assistant"):
+        st.markdown(test_response)
+    st.session_state.messages.append({
+        "role": "assistant",
+        "content": test_response,
+        "timestamp": datetime.now().strftime("%H:%M:%S")
+    })
+
+# Add a test button in sidebar:
+with st.sidebar:
+    if st.button("Test Response Display"):
+        test_response_display()