Update to purple theme - sync changes from test-playground

app.py

@@ -7,7 +7,7 @@ from urllib.parse import urlencode
 import requests
 import streamlit as st
 
-from gateway_client import delete_profile, ingest_and_rewrite
+from gateway_client import delete_profile, ingest_and_rewrite, get_memories, ingest_memories
 from llm import chat, set_model
 from model_config import MODEL_CHOICES, MODEL_TO_PROVIDER, MODEL_DISPLAY_NAMES
 
@@ -116,13 +116,15 @@ def delete_session(session_name: str) -> bool:
 
 
 def rewrite_message(
-    msg: str, persona_name: str, show_rationale: bool
+    msg: str, persona_name: str, show_rationale: bool, use_memory: bool = True
 ) -> str:
-    if persona_name.lower() == "control":
+    # If memory is disabled or Control persona, don't use memory
+    if not use_memory or persona_name.lower() == "control":
         rewritten_msg = msg
         if show_rationale:
             rewritten_msg += " At the beginning of your response, please say the following in ITALIC: 'Persona Rationale: No personalization applied.'. Begin your answer on the next line."
         return rewritten_msg
+    
     try:
         rewritten_msg = ingest_and_rewrite(
             user_id=persona_name, query=msg
@@ -165,7 +167,7 @@ HEADER_STYLE = """
     border-radius: 0;
 }
 .memmachine-header-links .powered-by {
-    color: #0a6cff;
+    color: #667eea;
     font-weight: 700;
     font-size: 16px;
     margin-right: 6px;
@@ -449,7 +451,7 @@ with st.sidebar:
                 key="hf_token_input",
                 type="password",
                 placeholder="hf_xxxxxxxxxxxxxxxxxxxxx",
-                help="Get your token from: https://huggingface.co/settings/tokens"
+                help="❓ Create a Read token: https://huggingface.co/settings/tokens"
             )
             
             if st.button("Authenticate", use_container_width=True, type="primary"):
@@ -459,7 +461,20 @@ with st.sidebar:
                     if is_valid and username:
                         st.session_state.hf_authenticated_user = username
                         st.session_state.hf_token = token_input.strip()  # Store for future use
-                        st.success(f"✅ Authenticated as **{username}**")
+                        # Use custom purple styling instead of green success message
+                        st.markdown(f"""
+                        <div style="
+                            background-color: rgba(102, 126, 234, 0.1);
+                            border-left: 4px solid #667eea;
+                            padding: 0.75rem 1rem;
+                            border-radius: 0.25rem;
+                            margin-bottom: 0.5rem;
+                        ">
+                            <div style="color: #667eea; font-weight: 500;">
+                                ✅ Authenticated as <strong>{username}</strong>
+                            </div>
+                        </div>
+                        """, unsafe_allow_html=True)
                         st.rerun()
                     else:
                         error_display = error_msg if error_msg else "Invalid token. Please check your Hugging Face access token."
@@ -471,7 +486,20 @@ with st.sidebar:
         else:
             # User is authenticated - lock to their username
             persona_name = st.session_state.hf_authenticated_user
-            st.success(f"🔐 Authenticated as: **{persona_name}**")
+            # Use custom purple styling instead of green success message
+            st.markdown(f"""
+            <div style="
+                background-color: rgba(102, 126, 234, 0.1);
+                border-left: 4px solid #667eea;
+                padding: 0.75rem 1rem;
+                border-radius: 0.25rem;
+                margin-bottom: 0.5rem;
+            ">
+                <div style="color: #667eea; font-weight: 500;">
+                    🔐 Authenticated as: <strong>{persona_name}</strong>
+                </div>
+            </div>
+            """, unsafe_allow_html=True)
             st.caption("Your memories are secured to your account only.")
             if st.button("🔓 Sign Out", use_container_width=True):
                 del st.session_state.hf_authenticated_user
@@ -495,7 +523,48 @@ with st.sidebar:
             custom_persona.strip() if custom_persona.strip() else selected_persona
         )
 
-    compare_personas = st.checkbox("Compare without MemMachine")
+    # Memory toggle - default enabled
+    if "memmachine_enabled" not in st.session_state:
+        st.session_state.memmachine_enabled = True
+    if "compare_personas" not in st.session_state:
+        st.session_state.compare_personas = True
+    
+    memmachine_enabled = st.checkbox(
+        "Enable MemMachine",
+        value=st.session_state.memmachine_enabled,
+        help="Enable MemMachine's persistent memory system. When unchecked, the AI will respond without memory (Control Persona mode)."
+    )
+    st.session_state.memmachine_enabled = memmachine_enabled
+    
+    if memmachine_enabled:
+        # Enhanced "Compare with control persona" section with cool styling
+        st.markdown("""
+        <div style="
+            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+            padding: 1rem;
+            border-radius: 0.5rem;
+            margin: 0.5rem 0;
+            border: 2px solid rgba(255, 255, 255, 0.2);
+            box-shadow: 0 4px 15px rgba(102, 126, 234, 0.3);
+        ">
+            <div style="display: flex; align-items: center; gap: 0.5rem; color: white;">
+                <span style="font-size: 1.5rem;">⚖️</span>
+                <div>
+                    <div style="font-weight: 600; font-size: 1rem;">Side-by-Side Comparison with Control Persona</div>
+                    <div style="font-size: 0.85rem; opacity: 0.9;">Compare MemMachine responses vs Control Persona (no memory)</div>
+                </div>
+            </div>
+        </div>
+        """, unsafe_allow_html=True)
+        
+        compare_personas = st.checkbox(
+            "🔄 Compare with control persona",
+            value=st.session_state.compare_personas,
+            help="Enable side-by-side comparison to see how MemMachine's persistent memory enhances responses compared to the control persona (no memory)"
+        )
+        st.session_state.compare_personas = compare_personas
+    else:
+        compare_personas = False
     show_rationale = st.checkbox("Show Persona Rationale")
 
     st.divider()
@@ -560,12 +629,117 @@ def typewriter_effect(text: str, speed: float = 0.02):
             yield " " + word
         time.sleep(speed)
 
+# ──────────────────────────────────────────────────────────────
+# Load Previous Memories Section (Import External Memories)
+# ──────────────────────────────────────────────────────────────
+if "memories_preview" not in st.session_state:
+    st.session_state.memories_preview = None
+if "imported_memories_text" not in st.session_state:
+    st.session_state.imported_memories_text = ""
+
+# Add expandable section for importing memories
+with st.expander("📋 Load Previous Memories (Import from ChatGPT, etc.)", expanded=False):
+    st.markdown("**Paste your conversation history or memories from external sources (e.g., ChatGPT, other AI chats)**")
+    
+    # Text area for pasting memories
+    imported_text = st.text_area(
+        "Paste your memories/conversations here",
+        value=st.session_state.imported_memories_text,
+        height=200,
+        placeholder="Example:\nUser: What is machine learning?\nAssistant: Machine learning is...\n\nUser: Can you explain neural networks?\nAssistant: Neural networks are...",
+        help="Paste any conversation history, notes, or context you want the AI to remember. These will be ingested into MemMachine's memory system and available for future conversations.",
+        key="import_memories_textarea"
+    )
+    
+    # File upload option
+    uploaded_file = st.file_uploader(
+        "Or upload a text file",
+        type=['txt', 'md', 'json'],
+        help="Upload a text file containing your conversation history or memories"
+    )
+    
+    if uploaded_file is not None:
+        try:
+            # Read file content
+            if uploaded_file.type == "application/json":
+                import json
+                file_content = json.loads(uploaded_file.read().decode("utf-8"))
+                imported_text = str(file_content)
+            else:
+                imported_text = uploaded_file.read().decode("utf-8")
+            st.session_state.imported_memories_text = imported_text
+            st.success("File loaded successfully!")
+        except Exception as e:
+            st.error(f"Error reading file: {e}")
+    
+    col1, col2 = st.columns(2)
+    
+    with col1:
+        if st.button("👁️ Preview", use_container_width=True, key="preview_memories"):
+            if imported_text and imported_text.strip():
+                st.session_state.memories_preview = imported_text
+                st.session_state.imported_memories_text = imported_text
+                st.rerun()
+            else:
+                st.warning("Please paste or upload some memories first.")
+    
+    with col2:
+        if st.button("💉 Ingest into MemMachine", use_container_width=True, key="inject_memories_direct"):
+            if imported_text and imported_text.strip():
+                if persona_name and persona_name != "Control":
+                    with st.spinner("Ingesting memories into MemMachine..."):
+                        success = ingest_memories(persona_name, imported_text)
+                        if success:
+                            st.session_state.imported_memories_text = imported_text
+                            st.success("✅ Memories successfully ingested into MemMachine! They are now part of your memory system.")
+                        else:
+                            st.error("❌ Failed to ingest memories. Please try again.")
+                else:
+                    st.warning("Please authenticate or select a persona to ingest memories.")
+                st.rerun()
+            else:
+                st.warning("Please paste or upload some memories first.")
+
+# Show preview if memories are loaded
+if st.session_state.memories_preview:
+    with st.expander("📋 Preview Imported Memories", expanded=True):
+        memories = st.session_state.memories_preview
+        preview_text = str(memories)[:2000]  # Show first 2000 chars
+        
+        if preview_text:
+            st.text_area("Memories Preview", preview_text, height=200, disabled=True, key="memories_preview_text")
+            st.caption(f"Total length: {len(str(memories))} characters")
+            
+            col1, col2 = st.columns(2)
+            with col1:
+                if st.button("💉 Ingest into MemMachine", use_container_width=True, key="inject_memories_from_preview"):
+                    if persona_name and persona_name != "Control":
+                        with st.spinner("Ingesting memories into MemMachine..."):
+                            success = ingest_memories(persona_name, str(st.session_state.memories_preview))
+                            if success:
+                                st.success("✅ Memories successfully ingested into MemMachine! They are now part of your memory system.")
+                            else:
+                                st.error("❌ Failed to ingest memories. Please try again.")
+                    else:
+                        st.warning("Please authenticate or select a persona to ingest memories.")
+                    st.rerun()
+            with col2:
+                if st.button("🗑️ Clear", use_container_width=True, key="clear_memories_preview"):
+                    st.session_state.memories_preview = None
+                    st.session_state.imported_memories_text = ""
+                    st.rerun()
+        else:
+            st.info("No memories to preview.")
+            st.session_state.memories_preview = None
+
 msg = st.chat_input("Type your message…")
 if msg:
     st.session_state.history.append({"role": "user", "content": msg})
-    if compare_personas:
+    memmachine_enabled = st.session_state.get("memmachine_enabled", True)
+    
+    if compare_personas and memmachine_enabled:
         all_answers = {}
-        rewritten_msg = rewrite_message(msg, persona_name, show_rationale)
+        rewritten_msg = rewrite_message(msg, persona_name, show_rationale, use_memory=True)
         msgs = clean_history(st.session_state.history, persona_name)
         msgs = append_user_turn(msgs, rewritten_msg)
         try:
@@ -575,7 +749,7 @@ if msg:
             st.error(f"❌ {str(e)}")
             st.stop()
 
-        rewritten_msg_control = rewrite_message(msg, "Control", show_rationale)
+        rewritten_msg_control = rewrite_message(msg, "Control", show_rationale, use_memory=False)
         msgs_control = clean_history(st.session_state.history, "Control")
         msgs_control = append_user_turn(msgs_control, rewritten_msg_control)
         try:
@@ -589,12 +763,13 @@ if msg:
             {"role": "assistant_all", "axis": "role", "content": all_answers, "is_new": True}
         )
     else:
-        rewritten_msg = rewrite_message(msg, persona_name, show_rationale)
+        # Use memory only if memmachine_enabled is True
+        rewritten_msg = rewrite_message(msg, persona_name, show_rationale, use_memory=memmachine_enabled)
         msgs = clean_history(st.session_state.history, persona_name)
         msgs = append_user_turn(msgs, rewritten_msg)
         try:
             txt, lat, tok, tps = chat(
-                msgs, "Arnold" if persona_name == "Control" else persona_name
+                msgs, "Arnold" if persona_name == "Control" or not memmachine_enabled else persona_name
             )
             st.session_state.history.append(
                 {"role": "assistant", "persona": persona_name, "content": txt, "is_new": True}
@@ -605,6 +780,23 @@ if msg:
     st.rerun()
 
 
+# ──────────────────────────────────────────────────────────────
+# Memory Status Indicator
+# ──────────────────────────────────────────────────────────────
+memmachine_enabled = st.session_state.get("memmachine_enabled", True)
+status_emoji = "🧠" if memmachine_enabled else "⚪"
+status_text = "MemMachine Active" if memmachine_enabled else "No Memory Mode"
+
+# Add status indicator at the top of chat area
+status_html = f"""
+<div style="display: flex; justify-content: flex-end; margin-bottom: 1rem; padding: 0.5rem 1rem; background: #f0f2f6; border-radius: 0.5rem; border: 1px solid #e0e0e0;">
+    <span style="font-size: 0.9rem; color: #666;">
+        {status_emoji} <strong>{status_text}</strong>
+    </span>
+</div>
+"""
+st.markdown(status_html, unsafe_allow_html=True)
+
 # ──��───────────────────────────────────────────────────────────
 # Chat history display
 # ──────────────────────────────────────────────────────────────
@@ -624,11 +816,42 @@ for turn in st.session_state.history:
         content_items = list(turn["content"].items())
         is_new = turn.get("is_new", False)
         if len(content_items) >= 2:
+            # Enhanced comparison header
+            st.markdown("""
+            <div style="
+                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                padding: 0.75rem 1rem;
+                border-radius: 0.5rem 0.5rem 0 0;
+                margin-bottom: 0.5rem;
+                display: flex;
+                align-items: center;
+                gap: 0.5rem;
+                color: white;
+                font-weight: 600;
+            ">
+                <span style="font-size: 1.2rem;">⚖️</span>
+                <span>Side-by-Side Comparison</span>
+            </div>
+            """, unsafe_allow_html=True)
+            
             cols = st.columns([1, 0.03, 1])
             persona_label, persona_response = content_items[0]
             control_label, control_response = content_items[1]
             with cols[0]:
-                st.markdown(f"**{persona_label}**")
+                st.markdown(f"""
+                <div style="
+                    background: linear-gradient(135deg, rgba(102, 126, 234, 0.1) 0%, rgba(118, 75, 162, 0.1) 100%);
+                    padding: 1rem;
+                    border-radius: 0.5rem;
+                    border-left: 4px solid #667eea;
+                    margin-bottom: 1rem;
+                ">
+                    <div style="display: flex; align-items: center; gap: 0.5rem; margin-bottom: 0.5rem;">
+                        <span style="font-size: 1.2rem;">🧠</span>
+                        <strong style="color: #667eea;">{persona_label}</strong>
+                    </div>
+                </div>
+                """, unsafe_allow_html=True)
                 if is_new:
                     st.write_stream(typewriter_effect(persona_response))
                 else:
@@ -641,7 +864,20 @@ for turn in st.session_state.history:
                     '<div class="vertical-divider"></div>', unsafe_allow_html=True
                 )
             with cols[2]:
-                st.markdown(f"**{control_label}**")
+                st.markdown(f"""
+                <div style="
+                    background: linear-gradient(135deg, rgba(200, 200, 200, 0.1) 0%, rgba(150, 150, 150, 0.1) 100%);
+                    padding: 1rem;
+                    border-radius: 0.5rem;
+                    border-left: 4px solid #888;
+                    margin-bottom: 1rem;
+                ">
+                    <div style="display: flex; align-items: center; gap: 0.5rem; margin-bottom: 0.5rem;">
+                        <span style="font-size: 1.2rem;">⚪</span>
+                        <strong style="color: #666;">{control_label}</strong>
+                    </div>
+                </div>
+                """, unsafe_allow_html=True)
                 if is_new:
                     st.write_stream(typewriter_effect(control_response))
                 else:
@@ -660,4 +896,4 @@ for turn in st.session_state.history:
                     )
         # Mark as no longer new
         if is_new:
-            turn["is_new"] = False
+            turn["is_new"] = False

styles.css

@@ -23,4 +23,49 @@ div[data-testid="column"] {
     height: 100%;
     border-left: 1px solid #ccc;
     margin: 0 0.4rem;
-  }
+  }
+
+  /* Enhanced comparison section styling */
+  .comparison-card {
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    padding: 1rem;
+    border-radius: 0.5rem;
+    margin: 0.5rem 0;
+    border: 2px solid rgba(255, 255, 255, 0.2);
+    box-shadow: 0 4px 15px rgba(102, 126, 234, 0.3);
+    transition: transform 0.2s ease, box-shadow 0.2s ease;
+  }
+
+  .comparison-card:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 6px 20px rgba(102, 126, 234, 0.4);
+  }
+
+  /* Comparison columns styling */
+  .comparison-column {
+    background: rgba(102, 126, 234, 0.05);
+    padding: 1rem;
+    border-radius: 0.5rem;
+    border: 1px solid rgba(102, 126, 234, 0.2);
+  }
+
+  /* Make primary buttons purple */
+  button[kind="primary"] {
+    background-color: #667eea !important;
+    border-color: #667eea !important;
+  }
+  
+  button[kind="primary"]:hover {
+    background-color: #764ba2 !important;
+    border-color: #764ba2 !important;
+  }
+
+  /* Make success messages purple */
+  div[data-testid="stSuccess"] {
+    background-color: rgba(102, 126, 234, 0.1) !important;
+    border-left-color: #667eea !important;
+  }
+  
+  div[data-testid="stSuccess"] > div {
+    color: #667eea !important;
+  }