Updating

layout.py

@@ -1,17 +1,15 @@
 import streamlit as st
 
 def init_session_state():
-    if "dark_mode" not in st.session_state:
-        st.session_state.dark_mode = False
-    if "code" not in st.session_state:
-        st.session_state.code = ""
-    if "stdin" not in st.session_state:
-        st.session_state.stdin = ""
-    if "language" not in st.session_state:
-        st.session_state.language = "Python"  # Default
+    """Set up initial values in session_state if not already defined."""
+    st.session_state.setdefault("dark_mode", False)
+    st.session_state.setdefault("code", "")
+    st.session_state.setdefault("stdin", "")
+    st.session_state.setdefault("language", "Python")
 
 
 def apply_theme():
+    """Apply the selected theme and return color palette + ACE theme."""
     dark = st.session_state.dark_mode
 
     colors = {
@@ -23,24 +21,21 @@ def apply_theme():
         "shadow": "rgba(0,0,0,0.3)" if dark else "rgba(0,0,0,0.1)",
     }
 
-    # Map selected language to Ace supported values
-    language = st.session_state.get("language", "python").lower()
-    if language == "c++":
-        language = "c_cpp"
-    elif language == "c#":
-        language = "csharp"
-    elif language == "javascript":
-        language = "javascript"
-    elif language == "python":
-        language = "python"
-    elif language == "java":
-        language = "java"
-    elif language == "c":
-        language = "c"
-
+    # Normalize ACE editor theme and language
     ace_theme = "monokai" if dark else "chrome"
+    lang = st.session_state.get("language", "python").lower()
+
+    ace_lang_map = {
+        "c++": "c_cpp",
+        "c#": "csharp",
+        "javascript": "javascript",
+        "python": "python",
+        "java": "java",
+        "c": "c"
+    }
+    ace_lang = ace_lang_map.get(lang, "python")  # fallback to python
 
-    # Inject CSS styles
+    # Inject theme-specific CSS
     st.markdown(
         f"""
         <style>