Update app.py

app.py

@@ -85,13 +85,13 @@ if prompt := st.chat_input("Ask Palm 2 anything..."):
             while current_attempt < MAX_ATTEMPTS:
                 try:
                     exec(current_code)
-                    break  # If successful, break out of the loop
+                    st.success('Code run succesfully!', icon="✅")
+                    break 
                 except Exception as e:
                     
                     st.error("Error running code...")
                     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)
-                    st.success('Code run succesfully!', icon="✅")
                     
                     with st.chat_message("assistant", avatar='🌴'):
                         st.markdown(new_response['content'])