Adding file formate with downloading

code_editor.py

@@ -103,4 +103,21 @@ def render_code_editor(selected_lang, ace_theme, editor_key):
         st.markdown(f"💾 **Memory Used:** `{mem_used:.2f} KB`")
 
     if st.session_state.code:
-        st.download_button("📥 Download Code", st.session_state.code, file_name=f"code_{selected_lang.lower()}.txt")
+        # Map Streamlit language to proper file extensions
+        lang_extension = {
+            "Python": "py",
+            "C": "c",
+            "C++": "cpp",
+            "Java": "java",
+            "JavaScript": "js",
+            "C#": "cs"
+        }
+        ext = lang_extension.get(selected_lang, "txt")
+
+        st.download_button(
+            label="💾 Download Code",
+            data=st.session_state.code,
+            file_name=f"code.{ext}",
+            mime="text/plain"
+        )
+