Updating

code_editor.py

@@ -4,6 +4,7 @@ import streamlit as st
 import streamlit_ace as st_ace
 from utils import execute_code
 
+# Default templates for each language
 DEFAULT_SNIPPETS = {
     "Python": '''# default code\na = int(input())\nb = int(input())\nprint("Sum:", a + b)''',
     "C": '''// default code\n#include <stdio.h>\nint main() {\n    int a, b;\n    scanf("%d %d", &a, &b);\n    printf("Sum: %d\\n", a + b);\n    return 0;\n}''',
@@ -17,18 +18,21 @@ DEFAULT_SNIPPETS = {
 def render_code_editor(selected_lang, ace_theme, editor_key):
     st.subheader("Editor")
 
-    # Initialize default code if needed
+    # Load the default code for the current language
     default_code = DEFAULT_SNIPPETS[selected_lang]
+    previous_lang = st.session_state.get("prev_language")
+
+    # Reset code when switching language or if empty
     if (
         st.session_state.get("code") is None
         or st.session_state.code.strip() == ""
-        or st.session_state.get("prev_language") != selected_lang
-        or st.session_state.code.strip() == DEFAULT_SNIPPETS.get(st.session_state.get("prev_language", ""), "")
+        or selected_lang != previous_lang
+        or st.session_state.code.strip() == DEFAULT_SNIPPETS.get(previous_lang, "")
     ):
         st.session_state.code = default_code
         st.session_state.prev_language = selected_lang
 
-    # Code Editor
+    # Code editor block
     code = st_ace.st_ace(
         value=st.session_state.code,
         placeholder=f"Start typing your {selected_lang} code…",
@@ -46,18 +50,19 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
     if code != st.session_state.code:
         st.session_state.code = code
 
-    # Input box
+    # Input area for stdin
     user_input = st.text_area(
-        label="📥 Input (stdin)",
+        "📥 Input (stdin)",
         value=st.session_state.stdin,
-        placeholder="Enter input() values here, one per line",
         height=100,
+        placeholder="Enter input() values here, one per line",
         key="stdin_input"
     )
+
     if user_input != st.session_state.stdin:
         st.session_state.stdin = user_input
 
-    # Run Button
+    # Run button
     if st.button("▶️ Run"):
         out, err, exc = execute_code(
             code=st.session_state.code,
@@ -67,7 +72,7 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
         st.session_state.code_output = out
         st.session_state.error_output = err or exc
 
-    # Output Display
+    # Output section
     if st.session_state.get("code_output"):
         st.text_area("Output", st.session_state.code_output, height=120)