Adding to get input

app.py

@@ -36,6 +36,7 @@ with assistant_col:
     st.subheader("Code Assistant")
     render_chatbot(
         st.session_state.code,
+        st.session_state.get("stdin", ""),
         st.session_state.get("code_output", ""),
         st.session_state.get("error_output", "")
     )