Update code_editor.py

code_editor.py

@@ -5,7 +5,6 @@ from pathlib import Path
 from utils import execute_code
 import time, psutil
 
-# Default code snippets
 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}''',
@@ -24,25 +23,29 @@ EXT_LANG_MAP = {
     ".cs": "C#"
 }
 
-def render_code_editor(selected_lang, ace_theme, editor_key):
-    # ── Handle File Upload ──────────────────────────────
+def render_code_editor(ace_theme):
+    # ── Language Selector ───────────────────────
+    languages = list(DEFAULT_SNIPPETS.keys())
+    selected_lang = st.selectbox("Language", languages, index=languages.index(st.session_state.get("language", "Python")))
+    st.session_state.language = selected_lang
+
+    # ── File Upload ─────────────────────────────
     uploaded_file = st.file_uploader("📤 Upload file", type=["py", "cpp", "c", "java", "js", "cs", "txt"])
+    
+    if "file_uploaded" not in st.session_state:
+        st.session_state.file_uploaded = False
 
-    if uploaded_file:
+    if uploaded_file and not st.session_state.file_uploaded:
         if uploaded_file.size > 10 * 1024 * 1024:
             st.error("🚫 File too large. Max allowed is 10MB.")
         else:
             content = uploaded_file.read().decode("utf-8", errors="ignore")
             ext = Path(uploaded_file.name).suffix.lower()
-
             detected_lang = EXT_LANG_MAP.get(ext)
+
             if detected_lang:
                 st.session_state.language = detected_lang
-                st.session_state.prev_language = detected_lang
-                selected_lang = detected_lang
                 st.toast(f"✅ Auto-switched to {detected_lang}", icon="🔄")
-                st.rerun()
-
             elif ext == ".txt":
                 st.toast("📄 Loaded text file", icon="📄")
             else:
@@ -50,8 +53,11 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
                 return
 
             st.session_state.code = content
+            st.session_state.file_uploaded = True
+            st.rerun()
 
-    # ── Load/Reset Code ──────────────────────────────
+    # ── Reset Language if Needed ─────────────────
+    selected_lang = st.session_state.language
     default_code = DEFAULT_SNIPPETS[selected_lang]
     previous_lang = st.session_state.get("prev_language")
 
@@ -64,7 +70,7 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
         st.session_state.code = default_code
         st.session_state.prev_language = selected_lang
 
-    # ── ACE Editor ──────────────────────────────
+    # ── Code Editor ─────────────────────────────
     code = st_ace.st_ace(
         value=st.session_state.code,
         placeholder=f"Start typing your {selected_lang} code…",
@@ -76,13 +82,13 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
         show_gutter=True,
         wrap=True,
         auto_update=True,
-        key=editor_key
+        key=f"editor_{selected_lang}"
     )
 
     if code != st.session_state.code:
         st.session_state.code = code
 
-    # ── Input for stdin ──────────────────────────────
+    # ── Stdin ─────────────────────────────
     user_input = st.text_area(
         "📥 Input (stdin)",
         value=st.session_state.stdin,
@@ -93,7 +99,7 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
     if user_input != st.session_state.stdin:
         st.session_state.stdin = user_input
 
-    # ── Execute Button ──────────────────────────────
+    # ── Run Button ─────────────────────────────
     if st.button("▶️ Run"):
         start_time = time.perf_counter()
         process = psutil.Process()
@@ -107,7 +113,7 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
 
         exec_time = time.perf_counter() - start_time
         mem_after = process.memory_info().rss
-        mem_used = (mem_after - mem_before) / 1024  # KB
+        mem_used = (mem_after - mem_before) / 1024
 
         st.session_state.code_output = out
         st.session_state.error_output = err or exc
@@ -119,7 +125,7 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
         st.markdown(f"⏱️ **Execution Time:** `{exec_time:.4f}s`")
         st.markdown(f"💾 **Memory Used:** `{mem_used:.2f} KB`")
 
-    # ── Download Button ──────────────────────────────
+    # ── Download Button ─────────────────────────────
     if st.session_state.code:
         lang_extension = {
             "Python": "py",