Update app.py

app.py

@@ -240,6 +240,16 @@ with st.container():
             }
         
             result = graph.invoke(state_input, config=config)
+
+            if result.get("code"):
+                st.session_state.chat_history.append(AIMessage(content=result["code"]))
+            
+            if result.get("explanation"):
+                st.session_state.chat_history.append(AIMessage(content=result["explanation"]))
+            
+            if result.get("messages"):
+                st.session_state.chat_history.append(result["messages"][-1])
+
         
             # If it’s a code response, append it as-is
             # if result.get("code"):
@@ -253,14 +263,14 @@ with st.container():
             # elif result.get("messages"):
             #     st.session_state.chat_history.append(result["messages"][-1])
 
-            if result.get("code"):
-                st.session_state.chat_history.append(AIMessage(content=result["code"]))
+            # if result.get("code"):
+            #     st.session_state.chat_history.append(AIMessage(content=result["code"]))
             
-            if result.get("explanation"):
-                st.session_state.chat_history.append(AIMessage(content="**🔍 Code Explanation:**\n\n" + result["explanation"]))
+            # if result.get("explanation"):
+            #     st.session_state.chat_history.append(AIMessage(content="**🔍 Code Explanation:**\n\n" + result["explanation"]))
             
-            if result.get("messages"):
-                st.session_state.chat_history.append(result["messages"][-1])
+            # if result.get("messages"):
+            #     st.session_state.chat_history.append(result["messages"][-1])
 
 
             st.rerun()