Simplify chat UX: fixed scrollable container, input at bottom

app.py

@@ -91,103 +91,86 @@ def main():
     if "question_only" not in st.session_state:
         st.session_state.question_only = False
 
-    # Sidebar
-    with st.sidebar:
-        st.markdown("### 🪞 GSPT")
-        st.caption("Generating Safer Passages of Text")
-
-        st.markdown("---")
-
-        # Mode toggle
-        st.markdown("### Mode")
-        question_only = st.toggle(
-            "Question Only",
-            value=st.session_state.question_only,
-            help="When on, GSPT responds only with questions"
-        )
-        st.session_state.question_only = question_only
-
-        if question_only:
-            st.caption("*Only asking questions*")
-        else:
-            st.caption("*Brief reflections + questions*")
-
-        st.markdown("---")
-
-        st.markdown("### About")
-        st.markdown("""
-        Not therapy. Not a confidant.
-
-        aI generates text — warm, boundaried reflections that bridge back to human connection.
-        """)
-
-        st.markdown("---")
-
-        st.markdown("### Crisis")
-        st.markdown("**988** · **741741** · **911**")
-
-        st.markdown("---")
-
-        if st.button("Clear All", use_container_width=True):
-            st.session_state.messages = []
-            st.session_state.journal = ""
-            st.rerun()
-
     # Check for API client
     client = get_api_client()
     if not client:
         st.error("Please set ANTHROPIC_API_KEY in your environment or Hugging Face secrets.")
         st.stop()
 
-    # Main layout: Chat and Journal
+    # Two column layout
     chat_col, journal_col = st.columns([3, 2])
 
+    # ==================== CHAT COLUMN ====================
     with chat_col:
-        st.markdown("## 🪞 GSPT")
-
-        # Display chat history
-        for message in st.session_state.messages:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
-
-        # Welcome message if empty
-        if not st.session_state.messages:
-            with st.chat_message("assistant"):
-                welcome = "What's alive for you right now?"
-                st.markdown(welcome)
+        # Header with controls
+        header_col, toggle_col, clear_col = st.columns([3, 1, 1])
+        with header_col:
+            st.markdown("### 🪞 GSPT")
+        with toggle_col:
+            question_only = st.toggle("Q only", value=st.session_state.question_only)
+            st.session_state.question_only = question_only
+        with clear_col:
+            if st.button("Clear"):
+                st.session_state.messages = []
+                st.rerun()
+
+        st.divider()
+
+        # Chat container
+        chat_container = st.container(height=450)
+
+        with chat_container:
+            # Welcome if empty
+            if not st.session_state.messages:
+                st.markdown("*What's alive for you right now?*")
+
+            # Display conversation
+            for msg in st.session_state.messages:
+                if msg["role"] == "user":
+                    st.markdown(f"**You:** {msg['content']}")
+                else:
+                    st.markdown(f"**GSPT:** {msg['content']}")
+                st.markdown("")  # spacing
+
+        # Input at bottom
+        st.divider()
+        prompt = st.text_input(
+            "Type here",
+            key="user_input",
+            placeholder="Share what's on your mind...",
+            label_visibility="collapsed"
+        )
 
-        # Chat input
-        if prompt := st.chat_input("Share what's on your mind..."):
+        if prompt:
             st.session_state.messages.append({"role": "user", "content": prompt})
-            with st.chat_message("user"):
-                st.markdown(prompt)
-
-            with st.chat_message("assistant"):
-                with st.spinner(""):
-                    response = generate_response(
-                        client,
-                        st.session_state.messages,
-                        question_only=st.session_state.question_only
-                    )
-                    st.markdown(response)
 
+            response = generate_response(
+                client,
+                st.session_state.messages,
+                question_only=st.session_state.question_only
+            )
             st.session_state.messages.append({"role": "assistant", "content": response})
+            st.rerun()
 
+    # ==================== JOURNAL COLUMN ====================
     with journal_col:
-        st.markdown("## 📓 Journal")
-        st.caption("A private space to capture what's emerging")
+        st.markdown("### 📓 Journal")
+        st.caption("Private notes — not sent anywhere")
+
+        st.divider()
 
         journal_text = st.text_area(
-            "Your notes",
+            "Journal",
             value=st.session_state.journal,
-            height=400,
-            placeholder="Write freely here...\n\nThis is just for you — not sent anywhere.",
+            height=450,
+            placeholder="Write freely here...",
             label_visibility="collapsed"
         )
         st.session_state.journal = journal_text
 
-        st.markdown("---")
-        st.caption("*Your journal stays in your browser. Nothing is saved or sent.*")
+    # Footer
+    st.divider()
+    st.caption("Not therapy. Not a confidant. aI generates text that bridges back to human connection. | Crisis: **988** · **741741** · **911**")
 
 
 if __name__ == "__main__":