Fixing narrate error

chatbot.py

@@ -100,7 +100,7 @@ def render_chatbot(code, output, error):
     st.session_state.setdefault('conversation', [])
     st.session_state.setdefault('chat_summary', "")
     st.session_state.setdefault('chat_display_count', 5)
-    st.session_state.setdefault('narrations', {})  # key: index, value: (text, file)
+    st.session_state.setdefault('narrated_audio', {})
 
     c1, c2 = st.columns([4, 1], gap='small')
     with c1:
@@ -125,9 +125,9 @@ def render_chatbot(code, output, error):
 
     total = len(st.session_state.conversation)
     start = max(0, total - st.session_state.chat_display_count)
-    visible = list(reversed(st.session_state.conversation[start:]))  # reversed
+    visible = list(reversed(st.session_state.conversation[start:]))
 
-    for i, (q, a) in enumerate(visible):
+    for idx, (q, a) in enumerate(visible):
         st.markdown(f'<div class="chat-message user-message">{escape(q)}</div>', unsafe_allow_html=True)
 
         def format_response(txt):
@@ -147,23 +147,22 @@ def render_chatbot(code, output, error):
         formatted = format_response(a)
         st.markdown(f'<div class="chat-message bot-message">{formatted}</div>', unsafe_allow_html=True)
 
-        # 🔊 Speak
-        speak_btn = st.button("🔊 Narrate", key=f"narrate_btn_{i}")
-        if speak_btn:
-            status_placeholder = st.empty()
-            status_placeholder.info("🧠 Generating narration...")
-            bot = CodeAssistantBot()
-            narration = bot.narrate_response(code, output, error, a, st.session_state.chat_summary)
-            status_placeholder.info("🎙️ Converting to audio...")
-            audio_filename = f"audio_{uuid.uuid4().hex}.mp3"
-            asyncio.run(text_to_speech(narration, audio_filename))
-            st.session_state.narrations[i] = (narration, audio_filename)
-            status_placeholder.success("🔊 Narration ready!")
-            st.audio(audio_filename, format="audio/mp3", autoplay=True)
-
-        # Playback of previously narrated
-        if i in st.session_state.narrations:
-            narration, audio_file = st.session_state.narrations[i]
+        # Check if already narrated
+        audio_file = st.session_state.narrated_audio.get((q, a))
+
+        if not audio_file:
+            if st.button("🔊 Narrate", key=f"narrate_{idx}"):
+                status_placeholder = st.empty()
+                status_placeholder.info("🧠 Generating narration...")
+                bot = CodeAssistantBot()
+                narration = bot.narrate_response(code, output, error, a, st.session_state.chat_summary)
+                status_placeholder.info("🎙️ Converting to audio...")
+                audio_file = f"audio_{uuid.uuid4().hex}.mp3"
+                asyncio.run(text_to_speech(narration, audio_file))
+                st.session_state.narrated_audio[(q, a)] = audio_file
+                status_placeholder.success("🔊 Narration ready!")
+                st.audio(audio_file, format="audio/mp3", autoplay=True)
+        else:
             st.audio(audio_file, format="audio/mp3", autoplay=False)
 
     if start > 0 and st.button("🔽 Show more"):
@@ -175,4 +174,4 @@ def render_chatbot(code, output, error):
     const c = window.parent.document.querySelector('.chat-container');
     if (c) c.scrollTop = c.scrollHeight;
     </script>
-    """, unsafe_allow_html=True)
+    """, unsafe_allow_html=True)