working on refresh issue

app.py

@@ -12,9 +12,6 @@ init_session_state()
 st.title("Pro Code Playground")
 st.markdown("Write, execute & export multi-language snippets, with built‑in AI assistance.")
 
-# Row for language selector + theme toggle
-lang_col, spacer, theme_col = st.columns([3, 6, 1])
-
 # Define default code snippets
 DEFAULT_SNIPPETS = {
     "Python": '''a = int(input())\nb = int(input())\nprint("Sum:", a + b)''',
@@ -25,15 +22,19 @@ DEFAULT_SNIPPETS = {
     "C#": '''using System;\npublic class Program {\n    public static void Main(string[] args) {\n        int a = Convert.ToInt32(Console.ReadLine());\n        int b = Convert.ToInt32(Console.ReadLine());\n        Console.WriteLine("Sum: " + (a + b));\n    }\n}'''
 }
 
-# Language dropdown
+# Language + Theme selection
+lang_col, spacer, theme_col = st.columns([3, 6, 1])
 with lang_col:
-    selected_lang = st.selectbox(
-        "Language",
-        options=list(DEFAULT_SNIPPETS.keys()),
-        index=0
-    )
+    selected_lang = st.selectbox("Language", list(DEFAULT_SNIPPETS.keys()), index=0)
+
+with theme_col:
+    theme_choice = st.radio("Theme", options=["☀️", "🌙"], horizontal=True, label_visibility="collapsed")
+    st.session_state.dark_mode = (theme_choice == "🌙")
+
+# Apply theme
+colors, ace_theme = apply_theme()
 
-# If language changed OR code was never set before, reset with default
+# Set default code when language changes or editor is empty
 prev_lang = st.session_state.get("prev_language", "")
 if (
     selected_lang != prev_lang
@@ -43,28 +44,20 @@ if (
 ):
     st.session_state.code = DEFAULT_SNIPPETS[selected_lang]
 
-# Update language and previous language
-lang = selected_lang
-st.session_state.prev_language = 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
 
-with theme_col:
-    theme_choice = st.radio("Theme", options=["☀️", "🌙"], horizontal=True, label_visibility="collapsed")
-    st.session_state.dark_mode = (theme_choice == "🌙")
-
-# Apply selected theme
-colors, ace_theme = apply_theme()
-
-# Editor + Chatbot Columns
+# Editor + Chatbot layout
 gen, bot = st.columns((2, 1), gap="large")
-
 with gen:
     st.subheader("Editor")
 
     code = st_ace.st_ace(
-        value=st.session_state.code,
+        value=st.session_state["editor_code"],
         placeholder=f"Start typing your {lang} code…",
-        language=lang.lower() if lang != "C++" else "c_cpp",  # Ace language codes
+        language=lang.lower() if lang != "C++" else "c_cpp",
         theme=ace_theme,
         keybinding="vscode",
         font_size=14,