Adding default code and file limit

code_editor.py

@@ -1,9 +1,9 @@
 import streamlit as st
 import streamlit_ace as st_ace
 import os
+import time, psutil
 from pathlib import Path
 from utils import execute_code
-import time, psutil
 
 DEFAULT_SNIPPETS = {
     "Python": '''# default code\na = int(input())\nb = int(input())\nprint("Sum:", a + b)''',
@@ -23,60 +23,52 @@ EXT_LANG_MAP = {
     ".cs": "C#"
 }
 
-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 ─────────────────────────────
+def render_code_editor(selected_lang, ace_theme, editor_key):
     uploaded_file = st.file_uploader("📤 Upload file", type=["py", "cpp", "c", "java", "js", "cs", "txt"])
 
     if uploaded_file:
         if uploaded_file.size > 10 * 1024 * 1024:
             st.error("🚫 File too large. Max allowed is 10MB.")
         else:
-            file_content = uploaded_file.read().decode("utf-8", errors="ignore")
+            content = uploaded_file.read().decode("utf-8", errors="ignore")
             filename = uploaded_file.name
-    
-            # Detect change: file name or content changed
+            ext = Path(filename).suffix.lower()
+
             prev_name = st.session_state.get("uploaded_file_name")
             prev_content = st.session_state.get("uploaded_file_content")
 
-            if file_content != prev_content or filename != prev_name:
-                ext = Path(filename).suffix.lower()
-                detected_lang = EXT_LANG_MAP.get(ext)
-
+            if content != prev_content or filename != prev_name:
                 st.session_state.uploaded_file_name = filename
-                st.session_state.uploaded_file_content = file_content
+                st.session_state.uploaded_file_content = content
 
+                detected_lang = EXT_LANG_MAP.get(ext)
                 if detected_lang:
                     st.session_state.language = detected_lang
                     st.session_state.prev_language = detected_lang
-                    st.session_state.code = file_content
+                    st.session_state.code = content
                     st.toast(f"✅ Auto-switched to {detected_lang}", icon="🔄")
                     st.rerun()
                 elif ext == ".txt":
-                    st.session_state.code = file_content
+                    st.session_state.code = content
                     st.toast("📄 Loaded text file", icon="📄")
                 else:
                     st.error("❌ Unsupported file format.")
-
-    # ── Reset Language if Needed ─────────────────
-        selected_lang = st.session_state.language
-        default_code = DEFAULT_SNIPPETS[selected_lang]
-        previous_lang = st.session_state.get("prev_language")
-
-        if (
-            st.session_state.get("code") is None
-            or st.session_state.code.strip() == ""
-            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 ─────────────────────────────
+                    return
+
+    # Handle default fallback (only if no upload or no valid code set)
+    previous_lang = st.session_state.get("prev_language")
+    default_code = DEFAULT_SNIPPETS[selected_lang]
+
+    if (
+        st.session_state.get("code") is None
+        or st.session_state.code.strip() == ""
+        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
+
+    # ── ACE Editor ──────────────────────────────
     code = st_ace.st_ace(
         value=st.session_state.code,
         placeholder=f"Start typing your {selected_lang} code…",
@@ -88,13 +80,13 @@ def render_code_editor(ace_theme):
         show_gutter=True,
         wrap=True,
         auto_update=True,
-        key=f"editor_{selected_lang}"
+        key=editor_key
     )
 
     if code != st.session_state.code:
         st.session_state.code = code
 
-    # ── Stdin ─────────────────────────────
+    # ── Input ──────────────────────────────
     user_input = st.text_area(
         "📥 Input (stdin)",
         value=st.session_state.stdin,
@@ -105,7 +97,7 @@ def render_code_editor(ace_theme):
     if user_input != st.session_state.stdin:
         st.session_state.stdin = user_input
 
-    # ── Run Button ─────────────────────────────
+    # ── Run Button ──────────────────────────────
     if st.button("▶️ Run"):
         start_time = time.perf_counter()
         process = psutil.Process()
@@ -127,21 +119,16 @@ def render_code_editor(ace_theme):
         st.text_area("📤 Output", out or "(no output)", height=120)
         if err or exc:
             st.error(err or exc)
-
         st.markdown(f"⏱️ **Execution Time:** `{exec_time:.4f}s`")
         st.markdown(f"💾 **Memory Used:** `{mem_used:.2f} KB`")
 
-    # ── Download Button ─────────────────────────────
+    # ── Download ──────────────────────────────
     if st.session_state.code:
-        lang_extension = {
-            "Python": "py",
-            "C": "c",
-            "C++": "cpp",
-            "Java": "java",
-            "JavaScript": "js",
-            "C#": "cs"
+        lang_ext = {
+            "Python": "py", "C": "c", "C++": "cpp", "Java": "java",
+            "JavaScript": "js", "C#": "cs"
         }
-        ext = lang_extension.get(selected_lang, "txt")
+        ext = lang_ext.get(selected_lang, "txt")
 
         st.download_button(
             label="💾 Download Code",