Update pages/3_numeracy.py

pages/3_numeracy.py

@@ -366,8 +366,14 @@ def main():
 
                     if 0 <= st.session_state['msg_index1'] < len(st.session_state['lessonList1']):
                         response = st.session_state['lessonList1'][st.session_state['msg_index1']]
-                        if response.strip().startswith("image:"):
-                            st.session_state['img_prompt1'] = response[len("image:"):].strip()
+                        try:
+                            next_response= st.session_state['lessonList1'][st.session_state['msg_index1']+1]
+                            
+                            if next_response.strip().startswith("image:"):
+                                
+                                st.session_state['img_prompt1'] = next_response[len("image:"):].strip()
+                        except:
+                            pass
                         else:
                             audio_bytes= tts_predict(response)
                             st.session_state['messages1'].append(f"Mo3alimi: {response}")