Adding lots of things

code_editor.py

@@ -1,11 +1,11 @@
-# code_editor.py
-
 import streamlit as st
 import streamlit_ace as st_ace
 import os
+from pathlib import Path
 from utils import execute_code
+import time, psutil
 
-# Default templates for each language
+# 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}''',
@@ -25,18 +25,31 @@ EXT_LANG_MAP = {
 }
 
 def render_code_editor(selected_lang, ace_theme, editor_key):
-    uploaded_file = st.file_uploader("📤 Upload code file", type=list([e[1:] for e in EXT_LANG_MAP.keys()]))
+    # ── Handle File Upload ──────────────────────────────
+    uploaded_file = st.file_uploader("📤 Upload file", type=["py", "cpp", "c", "java", "js", "cs", "txt"])
 
     if uploaded_file:
-        code_str = uploaded_file.read().decode("utf-8")
-        _, ext = os.path.splitext(uploaded_file.name)
-        detected_lang = EXT_LANG_MAP.get(ext.lower())
-        if detected_lang:
-            st.session_state.language = detected_lang
-            selected_lang = detected_lang
-            st.toast(f"✅ Language auto-detected: {detected_lang}", icon="🔄")
-        st.session_state.code = code_str
-
+        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="🔄")
+            elif ext == ".txt":
+                st.toast("📄 Loaded text file", icon="📄")
+            else:
+                st.error("❌ Unsupported file format.")
+                return
+
+            st.session_state.code = content
+
+    # ── Load/Reset Code ──────────────────────────────
     default_code = DEFAULT_SNIPPETS[selected_lang]
     previous_lang = st.session_state.get("prev_language")
 
@@ -49,6 +62,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 = st_ace.st_ace(
         value=st.session_state.code,
         placeholder=f"Start typing your {selected_lang} code…",
@@ -66,19 +80,19 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
     if code != st.session_state.code:
         st.session_state.code = code
 
+    # ── Input for stdin ──────────────────────────────
     user_input = st.text_area(
         "📥 Input (stdin)",
         value=st.session_state.stdin,
         height=100,
-        placeholder="Enter input() values here, one per line",
+        placeholder="Enter input values, one per line",
         key="stdin_input"
     )
-
     if user_input != st.session_state.stdin:
         st.session_state.stdin = user_input
 
+    # ── Execute Button ──────────────────────────────
     if st.button("▶️ Run"):
-        import time, psutil
         start_time = time.perf_counter()
         process = psutil.Process()
         mem_before = process.memory_info().rss
@@ -91,19 +105,20 @@ 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
+        mem_used = (mem_after - mem_before) / 1024  # KB
 
         st.session_state.code_output = out
         st.session_state.error_output = err or exc
 
-        st.text_area("Output", out or "(no output)", height=120)
+        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 ──────────────────────────────
     if st.session_state.code:
-        # Map Streamlit language to proper file extensions
         lang_extension = {
             "Python": "py",
             "C": "c",
@@ -120,4 +135,3 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
             file_name=f"code.{ext}",
             mime="text/plain"
         )
-