Update interface to chat-like format with clear submit button and conversation history

src/streamlit_app.py

@@ -424,16 +424,25 @@ st.markdown("""
 </div>
 """, unsafe_allow_html=True)
 
-# Journal input
+# Journal input and chat interface
 st.markdown("### Share what's on your heart...")
-journal_entry = st.text_area(
-    "",
-    height=200,
-    key="journal_input",
-    help="This is your space to reflect freely. Your words are not stored or saved unless you choose to save them."
-)
-
-if journal_entry:
+col1, col2 = st.columns([4, 1])
+with col1:
+    journal_entry = st.text_area(
+        "",
+        height=100,
+        key="journal_input",
+        help="Share your thoughts here. Press the 'Send' button or Ctrl+Enter to submit.",
+        placeholder="Type your reflection here..."
+    )
+with col2:
+    submit = st.button("Send 💌", use_container_width=True)
+
+# Initialize chat history in session state if it doesn't exist
+if 'chat_history' not in st.session_state:
+    st.session_state.chat_history = []
+
+if submit and journal_entry:  # Only process if there's text and button is clicked
     try:
         message = c.messages.create(
             model="claude-3-opus-20240229",
@@ -442,35 +451,62 @@ if journal_entry:
             messages=[{"role": "user", "content": journal_entry}]
         )
         
-        st.markdown("### Reflection")
-        reflection_text = f"{TONE_STYLES[selected_tone]['icon']} {TONE_STYLES[selected_tone]['prefix']}\n\n{message.content[0].text}"
-        st.markdown(reflection_text)
+        # Add the new exchange to chat history
+        st.session_state.chat_history.append({
+            "user_message": journal_entry,
+            "ai_response": f"{TONE_STYLES[selected_tone]['icon']} {TONE_STYLES[selected_tone]['prefix']}\n\n{message.content[0].text}",
+            "tone": selected_tone,
+            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M")
+        })
         
-        # Simple save option
-        if st.button("💌 Save this reflection"):
-            st.session_state.journal_entries.append({
-                "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M"),
-                "tone": selected_tone,
-                "content": journal_entry,
-                "reflection": reflection_text
-            })
-            st.success("Reflection saved.")
+        # Clear the input area after sending
+        st.rerun()
             
     except Exception as e:
         st.error(f"Error getting reflection: {str(e)}")
 
+# Display chat history in reverse chronological order
+st.markdown("### Conversation")
+for exchange in reversed(st.session_state.chat_history):
+    # User message in a light container
+    st.markdown("""
+    <div style="background-color: #F8F9FA; padding: 1rem; border-radius: 8px; margin: 0.5rem 0;">
+        <p style="color: #666666; margin-bottom: 0.5rem;"><em>You</em></p>
+        <p style="margin: 0;">{}</p>
+    </div>
+    """.format(exchange["user_message"]), unsafe_allow_html=True)
+    
+    # AI response in a warm container
+    st.markdown("""
+    <div style="background-color: #FFF8F3; padding: 1rem; border-radius: 8px; margin: 0.5rem 0;">
+        <p style="color: #666666; margin-bottom: 0.5rem;"><em>NurtureNest • {}</em></p>
+        <p style="margin: 0;">{}</p>
+    </div>
+    """.format(exchange["tone"], exchange["ai_response"]), unsafe_allow_html=True)
+
+# Save conversation button at the bottom
+if st.session_state.chat_history:
+    if st.button("📥 Save this conversation"):
+        st.session_state.journal_entries.extend(st.session_state.chat_history)
+        st.success("Conversation saved to your journal.")
+
 # Display saved reflections if any
 if st.session_state.journal_entries:
     with st.expander("📔 Your Saved Reflections", expanded=False):
         for entry in reversed(st.session_state.journal_entries):
+            timestamp = entry.get("timestamp", "")
+            tone = entry.get("tone", "")
+            content = entry.get("user_message", entry.get("content", ""))
+            reflection = entry.get("ai_response", entry.get("reflection", ""))
+            
             st.markdown(f"""
-            **{entry['timestamp']}** - *{entry['tone']}*
+            **{timestamp}** - *{tone}*
             
             *Your reflection:*  
-            {entry['content']}
+            {content}
             
             *Response:*  
-            {entry['reflection']}
+            {reflection}
             
             ---
             """)