Update app.py

app.py

@@ -91,7 +91,7 @@ if prompt := st.chat_input("Ask Palm 2 anything..."):
                     
                     with st.status("Rebuilding code..."):  # Status indicating code is being rebuilt
                         # Ask the AI model to regenerate the code and mention the problem
-                        new_prompt = f"{prompt} . There was an error executing the code: {e}. Can you provide a corrected version?"
+                        new_prompt = f"{prompt} . There was an error executing the code: {e}. The last code you provided was: ```{current_code}```. Can you provide a corrected version?"
                         new_response = predict(new_prompt)
                     
                     with st.chat_message("assistant", avatar='🌴'):