Update app.py

app.py

@@ -278,23 +278,47 @@ with st.container():
             #     st.session_state.chat_history.append(AIMessage(content=result["code"]))
             # elif result.get("explanation"):
             #     st.session_state.chat_history.append(AIMessage(content=result["explanation"]))
+            # elif result.get("messages"):
+            #     st.session_state.chat_history.append(result["messages"][-1])
+            # if result.get("code"):
+            #     st.session_state.latest_code = result["code"]
+            #     st.session_state.chat_history.append(AIMessage(content=result["code"]))
+            
+            # if result.get("explanation"):
+            #     st.session_state.latest_explanation = result["explanation"]
+
+            # if result.get("execution_result"):
+            #     st.session_state.chat_history.append(
+            #         AIMessage(content="**🧪 Code Execution Result:**\n\n" + result["execution_result"])
+            #     )
+            
             # elif result.get("messages"):
             #     st.session_state.chat_history.append(result["messages"][-1])
             if result.get("code"):
                 st.session_state.latest_code = result["code"]
-                st.session_state.chat_history.append(AIMessage(content=result["code"]))
+                st.session_state.chat_history.append(
+                    AIMessage(content="**💻 Generated Code:**\n\n```python\n" + result["code"] + "\n```")
+                )
             
-            if result.get("explanation"):
-                st.session_state.latest_explanation = result["explanation"]
-
             if result.get("execution_result"):
                 st.session_state.chat_history.append(
                     AIMessage(content="**🧪 Code Execution Result:**\n\n" + result["execution_result"])
                 )
             
+            if result.get("explanation"):
+                st.session_state.latest_explanation = result["explanation"]
+                st.session_state.chat_history.append(
+                    # AIMessage(content="**🔍 Code Explanation:**\n\n" + result["explanation"])
+                    AIMessage(content="**🔍 Code Explanation:**\n\n```\n" + result["explanation"] + "\n```")
+
+                )
+            
             elif result.get("messages"):
                 st.session_state.chat_history.append(result["messages"][-1])
+                
 
 
 
             st.rerun()
+            st.markdown("<script>window.scrollTo(0, document.body.scrollHeight);</script>", unsafe_allow_html=True)
+