Update app.py

app.py

@@ -32,6 +32,7 @@ class State(TypedDict):
     input : str
     code : str
     explanation:str
+    execution_result=str
  
 # LLM
 # llm = ChatGroq(model="deepseek-r1-distill-llama-70b")
@@ -143,7 +144,7 @@ def explain_code(state):
     full_prompt = system_prompt + prompt
     explanation = code_generator.invoke(full_prompt)
     
-    return {"input": user_prompt, "code": code, "explanation": explanation}
+    return {**state, "explanation": explanation}
 
 
 
@@ -154,6 +155,7 @@ def router(state):
         return "Generate_Code"
     else:
         return "AI_Assistance"
+        
 def execute_code(state: State) -> State:
     code = state.get("code", "")
     buffer = io.StringIO()
@@ -248,6 +250,11 @@ if "latest_explanation" in st.session_state:
     with st.expander("🔍 Show Code Explanation"):
         st.markdown(f"<div class='stChatMessage bot'>**🔍 Code Explanation:**<br><br>{st.session_state.latest_explanation}</div>", unsafe_allow_html=True)
 
+if "latest_code" in st.session_state:
+    with st.expander("💻 Show Generated Code"):
+        st.code(st.session_state.latest_code, language="python")
+
+
 # Custom Chat Input Box UI
 with st.container():
     with st.form("chat_form", clear_on_submit=True):