Adding input button.

app.py

@@ -1,8 +1,7 @@
 import streamlit as st
 import streamlit_ace as st_ace
-from utils import execute_code, export_session
+from utils import execute_code
 from chatbot import render_chatbot
-import json
 
 # 1️⃣ Page configuration (must be first)
 st.set_page_config(
@@ -12,11 +11,7 @@ st.set_page_config(
 )
 
 # 2️⃣ Dark mode toggle
-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
+dm = st.sidebar.checkbox("🌙 Dark mode", value=False)
 
 # 3️⃣ Theme variables
 BG = "#0f1620" if dm else "#f5f5f5"
@@ -43,13 +38,7 @@ st.markdown(f"""
       border: 1px solid {BORDER} !important;
       border-radius: 4px !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; }}
+  button {{ background-color: {ACCENT} !important; color: #fff !important; border-radius: 6px !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; }}
@@ -60,64 +49,68 @@ st.markdown(f"""
 
 # 5️⃣ App header
 st.title("Pro Code Playground")
-st.markdown("Write, execute & export Python snippets, with built‑in AI assistance.")
+st.markdown("Write, execute Python snippets, with built‑in AI assistance.")
 
-# 6️⃣ Main layout
-gen, bot = st.columns((2,1), gap="large")
+# 6️⃣ Layout: Editor + Assistant
+col1, col2 = st.columns([2, 1], gap="large")
 
-with gen:
+with col1:
     st.subheader("Editor")
     code = st_ace.st_ace(
-        placeholder="Start typing your Python code…",
+        placeholder="Type your Python code…",
         language="python",
         theme=ACE_THEME,
         keybinding="vscode",
         font_size=14,
-        min_lines=20,
+        min_lines=15,
         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"):
-        out, err, exc = execute_code(code)
-        st.session_state.code_output = out
-        st.session_state.error_output = err or exc
+        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 input() to return provided values in order
+                stub = (
+                    "import builtins
+"
+                    f"vals = iter([{', '.join(repr(v) for v in inputs)}])
+"
+                    "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
     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 bot:
+with col2:
     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)
+    )