Working on refresh issue

app.py

@@ -34,8 +34,8 @@ with theme_col:
 # Apply theme
 colors, ace_theme = apply_theme()
 
-# Set default code when language changes or editor is empty
-prev_lang = st.session_state.get("prev_language", "")
+# Initialize code for language
+prev_lang = st.session_state.get("prev_language", None)
 if (
     selected_lang != prev_lang
     or st.session_state.get("code") is None
@@ -43,21 +43,20 @@ if (
     or st.session_state.code.strip() == DEFAULT_SNIPPETS.get(prev_lang, "")
 ):
     st.session_state.code = DEFAULT_SNIPPETS[selected_lang]
-
-# Force update the Ace editor value
-st.session_state["editor_code"] = st.session_state.code
 st.session_state.prev_language = selected_lang
-lang = selected_lang  # Use `lang` everywhere below
 
-# Editor + Chatbot layout
+# Generate dynamic editor key so it refreshes
+editor_key = f"editor_{selected_lang}"
+
+# Layout: Editor + Chatbot
 gen, bot = st.columns((2, 1), gap="large")
 with gen:
     st.subheader("Editor")
 
     code = st_ace.st_ace(
-        value=st.session_state["editor_code"],
-        placeholder=f"Start typing your {lang} code…",
-        language=lang.lower() if lang != "C++" else "c_cpp",
+        value=st.session_state.code,
+        placeholder=f"Start typing your {selected_lang} code…",
+        language=selected_lang.lower() if selected_lang != "C++" else "c_cpp",
         theme=ace_theme,
         keybinding="vscode",
         font_size=14,
@@ -65,8 +64,9 @@ with gen:
         show_gutter=True,
         wrap=True,
         auto_update=True,
-        key="editor_code"
+        key=editor_key  # key now varies with language
     )
+
     if code != st.session_state.code:
         st.session_state.code = code
 
@@ -84,7 +84,7 @@ with gen:
         out, err, exc = execute_code(
             code=st.session_state.code,
             stdin=st.session_state.stdin,
-            language=lang
+            language=selected_lang
         )
         st.session_state.code_output = out
         st.session_state.error_output = err or exc