bug fix

app.py

@@ -3,6 +3,7 @@ from layout import init_session_state, apply_theme
 from code_editor import render_code_editor
 from chatbot import render_chatbot
 
+# ── Page Config ──────────────────────────────
 st.set_page_config(
     page_title="Pro Code Playground",
     page_icon="💻",
@@ -10,23 +11,28 @@ st.set_page_config(
 )
 init_session_state()
 
+# ── Header ───────────────────────────────────
 st.title("Pro Code Playground")
 st.markdown("Write, execute & export multi-language snippets, with built‑in AI assistance.")
 
-# Theme toggle only
+# ── Theme Toggle ─────────────────────────────
 _, _, theme_col = st.columns([3, 6, 1])
 with theme_col:
     if st.button("🌙 Dark Mode" if not st.session_state.dark_mode else "☀️ Light Mode"):
         st.session_state.dark_mode = not st.session_state.dark_mode
         st.rerun()
 
+# ── Apply Theme ──────────────────────────────
 colors, ace_theme = apply_theme()
 
+# ── Layout ───────────────────────────────────
 editor_col, assistant_col = st.columns((2, 1), gap="large")
 
 with editor_col:
     st.subheader("Editor")
-    render_code_editor(ace_theme)
+    selected_lang = st.session_state.get("language", "Python")
+    editor_key = f"editor_{selected_lang}"
+    render_code_editor(selected_lang, ace_theme, editor_key)
 
 with assistant_col:
     st.subheader("Code Assistant")
@@ -36,6 +42,7 @@ with assistant_col:
         st.session_state.get("error_output", "")
     )
 
+# ── Footer ───────────────────────────────────
 st.markdown("""
 <div style='text-align:center; margin-top:1rem; opacity:0.6;'>
   Built with ❤️ & Streamlit by Vaibhav