Fixing bug

app.py

@@ -86,15 +86,12 @@ with col1:
         else:
             exec_code = code
             if input_count > 0:
-                # Override input() to return provided values in order
+                # Override builtins.input to return provided values in order
+                vals_list = [repr(v) for v in inputs]
                 stub = (
-    "import builtins
+                    "import builtins
 "
-    f"vals = iter([{', '.join(repr(v) for v in inputs)}])
-"
-    "builtins.input = lambda prompt='': next(vals)
-"
-) for v in inputs)}])
+                    f"vals = iter([{', '.join(vals_list)}])
 "
                     "builtins.input = lambda prompt='': next(vals)
 "
@@ -105,7 +102,7 @@ with col1:
             st.session_state.code_output = out
             st.session_state.error_output = err or exc
 
-    # 9️⃣ Display output and errors
+    # 9️⃣ Display output and errors 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"):