Adding session to fix minor bugs

app.py

@@ -68,17 +68,25 @@ st.markdown(f"""
 </style>
 """, unsafe_allow_html=True)
 
-# 5️⃣ App header
+# 5️⃣ Session state initialization
+if 'code' not in st.session_state:
+    st.session_state.code = ""
+
+if 'stdin' not in st.session_state:
+    st.session_state.stdin = ""
+
+# 6️⃣ App header
 st.title("Pro Code Playground")
 st.markdown("Write, execute & export Python snippets, with built‑in AI assistance.")
 
-# 6️⃣ Layout
+# 7️⃣ Layout
 gen, bot = st.columns((2,1), gap="large")
 
 with gen:
     st.subheader("Editor")
 
     code = st_ace.st_ace(
+        value=st.session_state.code,
         placeholder="Start typing your Python code…",
         language="python",
         theme=ACE_THEME,
@@ -87,17 +95,24 @@ with gen:
         min_lines=20,
         show_gutter=True,
         wrap=True,
-        auto_update=True
+        auto_update=True,
+        key="editor_code"
     )
+    if code != st.session_state.code:
+        st.session_state.code = code
 
     user_input = st.text_area(
         "📥 Input (stdin)", 
+        value=st.session_state.stdin,
         placeholder="Enter input() values here, one per line",
-        height=100
+        height=100,
+        key="stdin_input"
     )
+    if user_input != st.session_state.stdin:
+        st.session_state.stdin = user_input
 
     if st.button("▶️ Run"):
-        out, err, exc = execute_code(code, user_input)
+        out, err, exc = execute_code(st.session_state.code, st.session_state.stdin)
         st.session_state.code_output = out
         st.session_state.error_output = err or exc
 
@@ -110,7 +125,7 @@ with gen:
 with bot:
     st.subheader("Code Assistant")
     render_chatbot(
-        code,
+        st.session_state.code,
         st.session_state.get("code_output", ""),
         st.session_state.get("error_output", "")
     )