Update app.py

app.py

@@ -87,23 +87,19 @@ if prompt := st.chat_input("Ask Palm 2 anything..."):
                     exec(current_code)
                     break  # If successful, break out of the loop
                 except Exception as e:
-                    st.write(f"ERROR {e}...")
                     
-                    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}. The last code you provided was: ```{current_code}```. Can you provide a corrected version?"
-                        new_response = predict(new_prompt)
+                    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'])
                     
-                    st.session_state.messages.append({"role": "assistant", "content": new_response['content']})
-                    
-                    # Update the current code with the new response and increment the attempt counter
                     current_code = new_response['code']
                     current_attempt += 1
     
             if current_attempt == MAX_ATTEMPTS:
-                st.write("Reached maximum attempts. Unable to execute the code successfully.")
+                st.warning("Reached maximum attempts. Unable to execute the code successfully.")
 
     st.session_state.messages.append({"role": "assistant", "content": response['content']})