Added AutoFix for PALM

Author: haseeb-heaven

libs/general_utils.py

@@ -47,6 +47,7 @@ def execute_code(self, compiler_mode: str):
  if "error" in code_output.lower() or "exception" in code_output.lower() or "SyntaxError" in code_output.lower() or "NameError" in code_output.lower():
 
  logger.error(f"Error in code execution: {code_output}")
+ st.session_state.stderr = code_output
  response = st.session_state.sequential_chain({'code_topic': generated_code})
  fixed_code = response['code_fix']
  st.code(fixed_code, language=code_language.lower())

libs/palmai.py

@@ -209,7 +209,7 @@ def fix_generated_code(self, code, code_language, fix_instructions=""):
 
  logger.info(f"Fixing code")
  if code and len(code) > 0:
- logger.info(f"Fixing code {code[:100]}... in language {code_language}")
+ logger.info(f"Fixing code {code[:100]}... in language {code_language} and error is {st.session_state.stderr}")
 
  # Improved instructions template
  template = f"""
@@ -259,7 +259,10 @@ def fix_generated_code(self, code, code_language, fix_instructions=""):
  else:
  logger.warning("No module-related error found in the output.")
 
- 
+ else:
+ st.toast("No error in previous execution.", icon="✅")
+ return code
+
  # Prompt Templates
  code_template = template
 

script.py

@@ -68,8 +68,10 @@ def initialize_session_state():
  st.session_state.gemini_langchain = None
  if "code_fix_instructions" not in st.session_state:
  st.session_state.code_fix_instructions = None
- if "session_state.sequential_chain" not in st.session_state:
+ if "sequential_chain" not in st.session_state:
  st.session_state.sequential_chain = None
+ if "stderr" not in st.session_state:
+ st.session_state.stderr = None
 
  # Initialize session state for Vertex AI
  if "vertexai" not in st.session_state:
@@ -431,7 +433,6 @@ def main():
  logger.info(f"Fixing code with instructions: {st.session_state.code_fix_instructions}")
  st.session_state.generated_code = st.session_state.palm_langchain.fix_generated_code(st.session_state.generated_code, st.session_state.code_language,st.session_state.code_fix_instructions)
 
-
  # Run Code button in the fourth column
  with run_code_col:
  execute_submitted = st.form_submit_button("Execute")