Fix: Integrated mood-aware journaling correctly

Author: aila-reddemma-ydv

TalkHeal.py

@@ -1,78 +1,79 @@
 # --- 0. IMPORTS ---
+import uuid  # standard library import
 import streamlit as st
-import google.generativeai as genai
 
-# --- 1. FIRST COMMAND: PAGE CONFIG ---
+# --- 1. PAGE CONFIG FIRST ---
 from core.config import PAGE_CONFIG
-st.set_page_config(**PAGE_CONFIG)
-
-GEMINI_API_KEY = st.secrets["GEMINI_API_KEY"]
+st.set_page_config(**PAGE_CONFIG) 
+from css.styles import apply_custom_css
 
-# --- 2. CONTINUED IMPORTS ---
-from core.utils import save_conversations, load_conversations, get_current_time, create_new_conversation
 from core.config import configure_gemini, get_tone_prompt, get_selected_mood
-from css.styles import apply_custom_css
+from core.utils import save_conversations, load_conversations, get_current_time, create_new_conversation
+
 from components.header import render_header
 from components.sidebar import render_sidebar
 from components.chat_interface import render_chat_interface, handle_chat_input
 from components.emergency_page import render_emergency_page
 
-# --- 3. SESSION STATE INITIALIZATION ---
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
-if "conversations" not in st.session_state:
-    st.session_state.conversations = load_conversations()
-if "active_conversation" not in st.session_state:
-    st.session_state.active_conversation = -1
-if "show_emergency_page" not in st.session_state:
-    st.session_state.show_emergency_page = False
-if "sidebar_state" not in st.session_state:
-    st.session_state.sidebar_state = "expanded"
-if "mental_disorders" not in st.session_state:
-    st.session_state.mental_disorders = [
+# --- 1. PAGE CONFIG & CSS ---
+apply_custom_css()
+
+# --- 2. SESSION STATE INITIALIZATION ---
+default_values = {
+    "conversations": [],
+    "active_conversation": -1,
+    "send_chat_message": False,
+    "pre_filled_chat_input": "",
+    "selected_mood_context": "",
+    "current_mood_val": "",
+    "chat_history": [],
+    "show_emergency_page": False,
+    "sidebar_state": "expanded",
+    "mental_disorders": [
         "Depression & Mood Disorders", "Anxiety & Panic Disorders", "Bipolar Disorder",
         "PTSD & Trauma", "OCD & Related Disorders", "Eating Disorders",
         "Substance Use Disorders", "ADHD & Neurodevelopmental", "Personality Disorders",
         "Sleep Disorders"
-    ]
-if "selected_tone" not in st.session_state:
-    st.session_state.selected_tone = "Compassionate Listener"
-if "selected_mood" not in st.session_state:
-    st.session_state.selected_mood = "🙂"  # Default emoji mood
+    ],
+    "selected_tone": "Compassionate Listener",
+    "selected_mood": "🙂"
+}
 
-# --- 4. STYLES & GEMINI SETUP ---
-apply_custom_css()
+for key, val in default_values.items():
+    if key not in st.session_state:
+        st.session_state[key] = val
+
+# --- 3. GEMINI MODEL SETUP ---
 model = configure_gemini()
 
-# --- 5. SIDEBAR ---
+# --- 4. SIDEBAR ---
 render_sidebar()
 
-# --- 6. MAIN PAGE ROUTING LOGIC ---
+# --- 5. MAIN LOGIC CONTAINER ---
 main_area = st.container()
 
-# Load conversations or start a new one
+# --- 6. Load or Start Conversation ---
 if not st.session_state.conversations:
-    saved_convos = load_conversations()
-    if saved_convos:
-        st.session_state.conversations = saved_convos
+    saved = load_conversations()
+    if saved:
+        st.session_state.conversations = saved
         if st.session_state.active_conversation == -1:
             st.session_state.active_conversation = 0
     else:
         create_new_conversation()
         st.session_state.active_conversation = 0
     st.rerun()
 
-# --- 7. MAIN VIEW DISPLAY ---
-if st.session_state.get("show_emergency_page"):
-    with main_area:
+# --- 7. ROUTING ---
+with main_area:
+    if st.session_state.get("show_emergency_page"):
         render_emergency_page()
-else:
-    with main_area:
+    else:
         render_header()
         st.subheader(f"🗣️ Current Chatbot Tone: **{st.session_state['selected_tone']}**")
         st.markdown(f"**🧠 Mood Selected:** {get_selected_mood()}")
         render_chat_interface()
-        handle_chat_input(model, system_prompt=get_tone_prompt())
+        handle_chat_input(model=model, system_prompt=get_tone_prompt())
 
 # --- 8. AUTO SCROLL SCRIPT ---
 st.markdown("""

api.py

@@ -0,0 +1,5 @@
+import google.generativeai as genai
+genai.configure(api_key="AIzaSyAgtTtPjJzTaNSrXaWFkMEJKIpMiFI3QNw")
+model = genai.GenerativeModel("gemini-1.5-flash")
+response = model.generate_content("Hello!")
+print(response.text)

components/chat_interface.py

@@ -1,82 +1,53 @@
 import streamlit as st
-import streamlit.components.v1 as components
-from datetime import datetime
-from core.utils import get_current_time, get_ai_response, save_conversations
-
-# Inject JS to get user's local time zone
-def set_user_time_in_session():
-    if "user_time_offset" not in st.session_state:
-        components.html("""
-            <script>
-            const offset = new Date().getTimezoneOffset(); 
-            const time = new Date().toLocaleString();      
-            const data = {offset: offset, time: time};
-            window.parent.postMessage({type: 'USER_TIME', data: data}, '*');
-            </script>
-        """, height=0)
-
-        st.markdown("""
-        <script>
-        window.addEventListener("message", (event) => {
-            if (event.data.type === "USER_TIME") {
-                const payload = JSON.stringify(event.data.data);
-                fetch("/", {
-                    method: "POST",
-                    headers: {"Content-Type": "application/json"},
-                    body: payload
-                }).then(() => location.reload());
-            }
-        });
-        </script>
-        """, unsafe_allow_html=True)
-
-set_user_time_in_session()
-
-# Display chat messages
-def render_chat_interface():    
-    if st.session_state.active_conversation >= 0:
-        active_convo = st.session_state.conversations[st.session_state.active_conversation]
-
-        if not active_convo["messages"]:
-            st.markdown(f"""
-            <div class="welcome-message">
-                <strong>Hello! I'm TalkHeal, your mental health companion 🤗</strong><br>
-                How are you feeling today? You can write below or start a new topic.
-                <div class="message-time">{get_current_time()}</div>
-            </div>
-            """, unsafe_allow_html=True)
-
-        for msg in active_convo["messages"]:
-            css_class = "user-message" if msg["sender"] == "user" else "bot-message"
-            st.markdown(f"""
-            <div class="{css_class}">
-                {msg["message"]}
-                <div class="message-time">{msg["time"]}</div>
-            </div>
-            """, unsafe_allow_html=True)
-
-# Handle chat input and generate AI response
+from core.utils import save_conversations, get_current_time
+from core.gemini import get_ai_response
+
+def render_chat_interface():
+    st.markdown("### 🗨️ TalkHeal Conversation")
+
+    active_index = st.session_state.get("active_conversation", -1)
+    if active_index == -1 or active_index >= len(st.session_state.conversations):
+        st.info("Start a conversation to see messages here.")
+        return
+
+    conversation = st.session_state.conversations[active_index]
+
+    for msg in conversation["messages"]:
+        sender = msg["sender"]
+        message = msg["message"]
+        time = msg["time"]
+
+        if sender == "user":
+            with st.chat_message("user", avatar="🧍‍♀️"):
+                st.markdown(f"**You:** {message}\n\n<sub>{time}</sub>", unsafe_allow_html=True)
+        else:
+            with st.chat_message("assistant", avatar="💬"):
+                st.markdown(f"{message}\n\n<sub>{time}</sub>", unsafe_allow_html=True)
+
+
 def handle_chat_input(model, system_prompt):
-    if "pre_filled_chat_input" not in st.session_state:
-        st.session_state.pre_filled_chat_input = ""
-    initial_value = st.session_state.pre_filled_chat_input
+    pre_filled = st.session_state.get("pre_filled_chat_input", "")
     st.session_state.pre_filled_chat_input = ""
 
-    with st.form(key="chat_form", clear_on_submit=True):
+    form_key = f"chat_form_{st.session_state.get('active_conversation', 0)}"
+
+    with st.form(key=form_key, clear_on_submit=True):
         col1, col2 = st.columns([5, 1])
         with col1:
             user_input = st.text_input(
                 "Share your thoughts...",
                 key="message_input",
                 label_visibility="collapsed",
                 placeholder="Type your message here...",
-                value=initial_value
+                value=pre_filled
             )
         with col2:
             send_pressed = st.form_submit_button("Send", use_container_width=True)
 
-    if (send_pressed or st.session_state.get("send_chat_message", False)) and user_input.strip():
-        if 'send_chat_message' in st.session_state:
+    auto_send = st.session_state.get("send_chat_message", False)
+
+    if (send_pressed or auto_send) and user_input.strip():
+        if auto_send:
             st.session_state.send_chat_message = False
 
         if st.session_state.active_conversation >= 0:
@@ -90,32 +61,31 @@ def handle_chat_input(model, system_prompt):
                 "time": current_time
             })
 
-            # Set title if it's the first message
             if len(active_convo["messages"]) == 1:
                 title = user_input[:30] + "..." if len(user_input) > 30 else user_input
                 active_convo["title"] = title
 
             save_conversations(st.session_state.conversations)
 
-            # Format memory
             def format_memory(convo_history, max_turns=10):
                 context = ""
-                for msg in convo_history[-max_turns*2:]:  # user + bot per turn
+                for msg in convo_history[-max_turns*2:]:
                     sender = "User" if msg["sender"] == "user" else "Bot"
                     context += f"{sender}: {msg['message']}\n"
                 return context
 
             try:
                 with st.spinner("TalkHeal is thinking..."):
                     memory = format_memory(active_convo["messages"])
-                    mood = st.session_state.get("selected_mood", "😊 Happy")
+                    mood = st.session_state.get("selected_mood_context") or st.session_state.get("current_mood_val", "okay")
+
                     prompt = (
                         f"{system_prompt}\n\n"
-                        f"The user is currently feeling: {mood}. Consider this mood while responding empathetically.\n\n"
+                        f"User is feeling {mood}. Respond empathetically.\n\n"
                         f"{memory}\nUser: {user_input.strip()}\nBot:"
                     )
 
-                    ai_response = get_ai_response(prompt, model)
+                    ai_response = get_ai_response(prompt, "gemini-1.5-flash")
 
                     active_convo["messages"].append({
                         "sender": "bot",
@@ -133,3 +103,5 @@ def format_memory(convo_history, max_turns=10):
 
             save_conversations(st.session_state.conversations)
             st.rerun()
+
+__all__ = ["render_chat_interface", "handle_chat_input"]