Update neo_sages4.py

neo_sages4.py

@@ -268,13 +268,13 @@ def fetch_response(user_input, milvus_client, emb, vector_index_properties, vect
         )
         prompt_data = prompt_data.replace("__grounding__", grounding)
 
-        # Add debug information to column 1 if enabled
-        if genparam.INPUT_DEBUG_VIEW == 1:
-            with col1:  # Access first column
-                bot_name = genparam.BOT_2_NAME if chat_history == st.session_state.chat_history_2 else genparam.BOT_3_NAME
-                bot_avatar = genparam.BOT_2_AVATAR if chat_history == st.session_state.chat_history_2 else genparam.BOT_3_AVATAR
-                st.markdown(f"**{bot_avatar} {bot_name} Prompt Data:**")
-                st.code(prompt_data, language="text")
+        # # Add debug information to column 1 if enabled
+        # if genparam.INPUT_DEBUG_VIEW == 1:
+        #     with col1:  # Access first column
+        #         bot_name = genparam.BOT_2_NAME if chat_history == st.session_state.chat_history_2 else genparam.BOT_3_NAME
+        #         bot_avatar = genparam.BOT_2_AVATAR if chat_history == st.session_state.chat_history_2 else genparam.BOT_3_AVATAR
+        #         st.markdown(f"**{bot_avatar} {bot_name} Prompt Data:**")
+        #         st.code(prompt_data, language="text")
 
     # Continue with normal processing for columns 2 and 3
     watsonx_llm = ModelInference(
@@ -503,6 +503,14 @@ def main():
                 system_prompt,
                 st.session_state.chat_history_2
             )
+
+            if genparam.INPUT_DEBUG_VIEW == 1:
+                with col1:  # Access first column
+                    bot_name = genparam.BOT_2_NAME if chat_history == st.session_state.chat_history_2 else genparam.BOT_3_NAME
+                    bot_avatar = genparam.BOT_2_AVATAR if chat_history == st.session_state.chat_history_2 else genparam.BOT_3_AVATAR
+                    st.markdown(f"**{bot_avatar} {bot_name} Prompt Data:**")
+                    st.code(prompt_data, language="text")
+            
             st.session_state.chat_history_2.append({"role": genparam.BOT_2_NAME, "content": response, "avatar": genparam.BOT_2_AVATAR})
             st.markdown("</div></div>", unsafe_allow_html=True)
             
@@ -534,6 +542,14 @@ def main():
                 system_prompt,
                 st.session_state.chat_history_3
             )
+            
+            if genparam.INPUT_DEBUG_VIEW == 1:
+                with col1:  # Access first column
+                    bot_name = genparam.BOT_2_NAME if chat_history == st.session_state.chat_history_2 else genparam.BOT_3_NAME
+                    bot_avatar = genparam.BOT_2_AVATAR if chat_history == st.session_state.chat_history_2 else genparam.BOT_3_AVATAR
+                    st.markdown(f"**{bot_avatar} {bot_name} Prompt Data:**")
+                    st.code(prompt_data, language="text")
+            
             st.session_state.chat_history_3.append({"role": genparam.BOT_3_NAME, "content": response, "avatar": genparam.BOT_3_AVATAR})
             st.markdown("</div></div>", unsafe_allow_html=True)