Restoring

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
 import streamlit_ace as st_ace
-from utils import execute_code
+from utils import execute_code, export_session
 from chatbot import render_chatbot
+import json
 
 # 1️⃣ Page configuration (must be first)
 st.set_page_config(
@@ -11,7 +12,11 @@ st.set_page_config(
 )
 
 # 2️⃣ Dark mode toggle
-dm = st.sidebar.checkbox("🌙 Dark mode", value=False)
+if 'dark_mode' not in st.session_state:
+    st.session_state.dark_mode = True
+
+dm = st.checkbox("🌙 Dark mode", value=st.session_state.dark_mode)
+st.session_state.dark_mode = dm
 
 # 3️⃣ Theme variables
 BG = "#0f1620" if dm else "#f5f5f5"
@@ -38,7 +43,13 @@ st.markdown(f"""
       border: 1px solid {BORDER} !important;
       border-radius: 4px !important;
   }}
-  button {{ background-color: {ACCENT} !important; color: #fff !important; border-radius: 6px !important; }}
+  button, .stDownloadButton > button {{
+      background-color: {ACCENT} !important;
+      color: #fff !important;
+      border-radius: 6px !important;
+      transition: transform 0.1s;
+  }}
+  button:hover {{ transform: scale(1.02) !important; }}
   .chat-container {{ background: {PANEL_BG} !important; border: 1px solid {BORDER} !important; border-radius: 8px !important; padding: 1rem; max-height: 480px; overflow-y: auto; }}
   .chat-message {{ margin-bottom: 1rem; padding: 0.75rem 1rem; border-radius: 12px; }}
   .user-message {{ background: rgba(100,149,237,0.2); align-self: flex-end; }}
@@ -49,69 +60,64 @@ st.markdown(f"""
 
 # 5️⃣ App header
 st.title("Pro Code Playground")
-st.markdown("Write, execute Python snippets, with built‑in AI assistance.")
+st.markdown("Write, execute & export Python snippets, with built‑in AI assistance.")
 
-# 6️⃣ Layout: Editor + Assistant
-col1, col2 = st.columns([2, 1], gap="large")
+# 6️⃣ Main layout
+gen, bot = st.columns((2,1), gap="large")
 
-with col1:
+with gen:
     st.subheader("Editor")
     code = st_ace.st_ace(
-        placeholder="Type your Python code…",
+        placeholder="Start typing your Python code…",
         language="python",
         theme=ACE_THEME,
         keybinding="vscode",
         font_size=14,
-        min_lines=15,
+        min_lines=20,
         show_gutter=True,
         wrap=True,
         auto_update=True
     )
-
-    # 7️⃣ Detect input() calls and collect values
-    input_count = code.count("input(")
-    inputs = []
-    if input_count > 0:
-        st.info(f"Your code expects {input_count} input() call(s). Please provide values:")
-        for i in range(input_count):
-            val = st.text_input(f"Input value #{i+1}", key=f"in_{i}", help="Enter value for input() call")
-            inputs.append(val)
-        if not all(val.strip() for val in inputs):
-            st.warning("Please fill all input fields before running the code.")
-
-    # 8️⃣ Run code
     if st.button("▶️ Run"):
-        if input_count > 0 and not all(val.strip() for val in inputs):
-            st.error("Cannot run: missing input values.")
-        else:
-            exec_code = code
-            if input_count > 0:
-                # Override builtins.input to return provided values in order
-                vals_list = [repr(v) for v in inputs]
-                stub = (
-                    "import builtins
-"
-                    f"vals = iter([{', '.join(vals_list)}])
-"
-                    "builtins.input = lambda prompt='': next(vals)
-"
-                )
-                exec_code = stub + exec_code
-
-            out, err, exc = execute_code(exec_code)
-            st.session_state.code_output = out
-            st.session_state.error_output = err or exc
-
-    # 9️⃣ Display output and errors and errors
+        out, err, exc = execute_code(code)
+        st.session_state.code_output = out
+        st.session_state.error_output = err or exc
     if st.session_state.get("code_output"):
         st.text_area("Output", st.session_state.code_output, height=120)
     if st.session_state.get("error_output"):
         st.error(st.session_state.error_output)
 
-with col2:
+with bot:
     st.subheader("Code Assistant")
     render_chatbot(
         code,
         st.session_state.get("code_output", ""),
         st.session_state.get("error_output", "")
-    )
+    )
+
+# 7️⃣ Export section
+st.markdown("---")
+st.subheader("Export")
+export_data = export_session(
+    code,
+    st.session_state.get("code_output", ""),
+    st.session_state.get("error_output", "")
+)
+col1, col2 = st.columns(2)
+with col1:
+    st.download_button(
+        "📦 JSON",
+        data=json.dumps(export_data, indent=2),
+        file_name="session.json",
+        mime="application/json"
+    )
+with col2:
+    txt = f"# Code\n{code}\n\n# Output\n{st.session_state.get('code_output','')}\n\n# Errors\n{st.session_state.get('error_output','')}"
+    st.download_button("📝 Text", txt, "session.txt", "text/plain")
+
+# 8️⃣ Footer
+st.markdown(f"""
+<div style='text-align:center; margin-top:1rem; color:{TEXT}66;'>
+  Built with ❤️ & Streamlit
+</div>
+""", unsafe_allow_html=True)