fixing bug

app.py

@@ -4,19 +4,20 @@ from utils import execute_code
 from chatbot import render_chatbot
 from layout import init_session_state, apply_theme
 
-# Set initial page config
+# Page config
 st.set_page_config(page_title="Pro Code Playground", page_icon="💻", layout="wide")
 
-# Init session
+# Session init
 init_session_state()
 
-# Theme Toggle
+# Theme toggle (emojis)
 theme_choice = st.radio("🎨 Theme", options=["☀️", "🌙"], horizontal=True, label_visibility="collapsed")
 st.session_state.dark_mode = (theme_choice == "🌙")
-apply_theme(st.session_state.theme)
 
+# Apply theme + get styles
+colors, ace_theme = apply_theme()
 
-# App Header
+# Header
 st.title("Pro Code Playground")
 st.markdown("Write, execute & export Python snippets, with built‑in AI assistance.")
 
@@ -30,7 +31,7 @@ with gen:
         value=st.session_state.code,
         placeholder="Start typing your Python code…",
         language="python",
-        theme="chrome" if st.session_state.theme == "light" else "monokai",
+        theme=ace_theme,
         keybinding="vscode",
         font_size=14,
         min_lines=20,
@@ -43,7 +44,7 @@ with gen:
         st.session_state.code = code
 
     user_input = st.text_area(
-        "📥 Input (stdin)",
+        label="📥 Input (stdin)",
         value=st.session_state.stdin,
         placeholder="Enter input() values here, one per line",
         height=100,
@@ -72,7 +73,7 @@ with bot:
     )
 
 # Footer
-st.markdown("""
+st.markdown(f"""
 <div style='text-align:center; margin-top:1rem; opacity:0.6;'>
   Built with ❤️ & Streamlit by Vaibhav
 </div>