Update app.py

app.py

@@ -64,22 +64,46 @@ if prompt := st.chat_input("Ask Palm 2 anything..."):
     with st.chat_message("assistant", avatar='🌴'):
         st.markdown(response['content'])
     
+    MAX_ATTEMPTS = 3  # Define a maximum number of attempts to execute the code
+
     if response['code']:
         if exportToReplIt:
             with st.status("Exporting replit..."):
-                fale = False
+                fail = False
                 try:
-                    url = bard.export_replit(code=response['code'],program_lang=response['program_lang'])['url']
-                except error:
-                    fale=True
-                    st.write('ERROR')
-            if not fale:
+                    url = bard.export_replit(code=response['code'], program_lang=response['program_lang'])['url']
+                except Exception as error:
+                    fail = True
+                    st.write('ERROR exporting to repl.it')
+            if not fail:
                 st.title('Export to repl.it')
                 st.markdown(f'[link]({url})')
+        
         if code_interpreter:
-            try:
-                exec(response['code'])
-            except Exception as e:
-                st.write(f"ERROR {e}...")
+            current_attempt = 0
+            current_code = response['code']
+            while current_attempt < MAX_ATTEMPTS:
+                try:
+                    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}. Can you provide a corrected version?"
+                        new_response = predict(new_prompt)
+                    
+                    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.session_state.messages.append({"role": "assistant", "content": response['content']})