Adding file upload

code_editor.py

@@ -2,6 +2,7 @@
 
 import streamlit as st
 import streamlit_ace as st_ace
+import os
 from utils import execute_code
 
 # Default templates for each language
@@ -14,14 +15,31 @@ DEFAULT_SNIPPETS = {
     "C#": '''// default code\nusing System;\npublic class Program {\n    public static void Main(string[] args) {\n        int a = Convert.ToInt32(Console.ReadLine());\n        int b = Convert.ToInt32(Console.ReadLine());\n        Console.WriteLine("Sum: " + (a + b));\n    }\n}'''
 }
 
+EXT_LANG_MAP = {
+    ".py": "Python",
+    ".cpp": "C++",
+    ".c": "C",
+    ".java": "Java",
+    ".js": "JavaScript",
+    ".cs": "C#"
+}
 
 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()]))
+
+    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
 
-    # 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() == ""
@@ -31,7 +49,6 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
         st.session_state.code = default_code
         st.session_state.prev_language = selected_lang
 
-    # Code editor block
     code = st_ace.st_ace(
         value=st.session_state.code,
         placeholder=f"Start typing your {selected_lang} code…",
@@ -49,7 +66,6 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
     if code != st.session_state.code:
         st.session_state.code = code
 
-    # Input area for stdin
     user_input = st.text_area(
         "📥 Input (stdin)",
         value=st.session_state.stdin,
@@ -61,19 +77,30 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
     if user_input != st.session_state.stdin:
         st.session_state.stdin = user_input
 
-    # Run button
     if st.button("▶️ Run"):
+        import time, psutil
+        start_time = time.perf_counter()
+        process = psutil.Process()
+        mem_before = process.memory_info().rss
+
         out, err, exc = execute_code(
             code=st.session_state.code,
             stdin=st.session_state.stdin,
             language=selected_lang
         )
+
+        exec_time = time.perf_counter() - start_time
+        mem_after = process.memory_info().rss
+        mem_used = (mem_after - mem_before) / 1024
+
         st.session_state.code_output = out
         st.session_state.error_output = err or exc
 
-    # Output section
-    if st.session_state.get("code_output"):
-        st.text_area("Output", st.session_state.code_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`")
 
-    if st.session_state.get("error_output"):
-        st.error(st.session_state.error_output)
+    if st.session_state.code:
+        st.download_button("📥 Download Code", st.session_state.code, file_name=f"code_{selected_lang.lower()}.txt")