Added code conversion functionality

Author: haseeb-heaven

libs/geminiai.py

@@ -197,3 +197,56 @@ def fix_generated_code(self, code, code_language, fix_instructions=""):
  except Exception as exception:
  st.toast(f"Error in code fixing: {exception}", icon="❌")
  logger.error(f"Error in code fixing: {traceback.format_exc()}")
+
+ def convert_generated_code(self, code, code_language):
+ """
+ Function to convert the generated code to a different language using the palm API.
+ """
+ try:
+ # Check for valid code
+ if not code or len(code) == 0:
+ logger.error("Error in code conversion: Please enter a valid code.")
+ return
+ 
+ logger.info(f"Converting code")
+ if code and len(code) > 0:
+ logger.info(f"Converting code {code[:100]}... to language {code_language}")
+ 
+ # Improved instructions template
+ template = f"""
+ Task: Convert the code snippet provided below to the {code_language} programming language, following the given instructions:
+
+ {code}
+
+ Instructions for Conversion:
+ 1. Identify the functionality of the original code.
+ 2. Translate the code into the {code_language} programming language, maintaining the same functionality.
+ 3. Verify that the converted code is displayed in the output.
+
+ Please make sure only the converted code should be included in the output.
+ """
+
+ # Prompt Templates
+ code_template = template
+ 
+ # LLM Chains definition
+ # Create a chain that generates the code
+ gemini_completion = self.model.generate_content(code_template)
+ logger.info("Text generation completed successfully.")
+
+ if gemini_completion:
+ # Extracted code from the palm completion
+ code = gemini_completion.text
+ extracted_code = self.utils.extract_code(code)
+ 
+ # Check if the code or extracted code is not empty or null
+ if not code or not extracted_code:
+ raise Exception("Error: Generated code or extracted code is empty or null.")
+ else:
+ return extracted_code
+ else:
+ raise Exception("Error in code conversion: Please enter a valid code.")
+ else:
+ logger.error("Error in code conversion: Please enter a valid code and language.")
+ except Exception as exception:
+ logger.error(f"Error in code conversion: {traceback.format_exc()}")

libs/general_utils.py

@@ -122,11 +122,11 @@ def check_compilers(self, language):
 
  compilers = {
  "python": ["python", "--version"],
- "nodejs": ["node", "--version"],
+ "javascript": ["node", "--version"],
  "c": ["gcc", "--version"],
  "c++": ["g++", "--version"],
- "csharp": ["csc", "--version"],
- "go": ["go", "version"],
+ "c#": ["csc", "--version"],
+ "go": ["go", "--version"],
  "ruby": ["ruby", "--version"],
  "java": ["java", "--version"],
  "kotlin": ["kotlinc", "--version"],
@@ -135,8 +135,8 @@ def check_compilers(self, language):
  }
 
  if language not in compilers:
- logger.error("Invalid language selected.")
- st.toast("Invalid language selected.", icon="❌")
+ logger.error(f"Invalid language selected. {language} not found in compilers list.")
+ st.toast(f"Invalid language selected. {language} not found in compilers list.", icon="❌")
  return False
 
  compiler = subprocess.run(compilers[language], capture_output=True, text=True)
@@ -171,9 +171,11 @@ def run_code(self,code, language):
  return output.stdout + output.stderr
 
  elif language == "C" or language == "C++":
+ 
  ext = ".c" if language == "C" else ".cpp"
  compiler = "gcc" if language == "C" else "g++"
  std = "-std=c11" if language == "C" else "-std=c++17"
+
  with tempfile.NamedTemporaryFile(mode="w", suffix=ext, delete=True) as src_file:
  src_file.write(code)
  src_file.flush()

libs/openai_langchain.py

@@ -229,4 +229,78 @@ def fix_generated_code(self, code_snippet, code_language, fix_instructions=""):
  logger.error("Error in code fixing: Please enter a valid code and language.")
  except Exception as exception:
  st.toast(f"Error in code fixing: {exception}", icon="❌")
- logger.error(f"Error in code fixing: {traceback.format_exc()}")
+ logger.error(f"Error in code fixing: {traceback.format_exc()}")
+
+ def convert_generated_code(self, code_snippet, code_language):
+ """
+ Function to convert the generated code to a different language using the palm API.
+ """
+ try:
+ # Check for valid code
+ if not code_snippet or len(code_snippet) == 0:
+ logger.error("Error in code conversion: Please enter a valid code.")
+ return
+ 
+ logger.info(f"Converting code")
+ if code_snippet and len(code_snippet) > 0:
+ logger.info(f"Converting code {code_snippet[:100]}... to language {code_language}")
+ 
+ # Improved instructions template
+ template = f"""
+ Task: Convert the code snippet provided below to the {code_language} programming language, following the given instructions:
+
+ {code_snippet}
+
+ Instructions for Conversion:
+ 1. Identify the functionality of the original code.
+ 2. Translate the code into the {code_language} programming language, maintaining the same functionality.
+ 3. Verify that the converted code is displayed in the output.
+
+ Please make sure only the converted code should be included in the output.
+ """
+
+ # Prompt Templates
+ code_template = template
+ 
+ # LLM Chains definition
+ # Create a chain that fixed the code
+ fix_generated_template = PromptTemplate(
+ input_variables=['code_prompt', 'code_language'],
+ template=code_template
+ )
+
+ fix_generated_chain = LLMChain(
+ llm=self.lite_llm,
+ prompt=fix_generated_template,
+ output_key='fixed_code',
+ memory=self.memory,
+ verbose=True
+ )
+
+ # Prepare the input for the chain
+ input_data = {
+ 'code_prompt': code_snippet,
+ 'code_language': code_language
+ }
+
+ # Run the chain
+ output = fix_generated_chain.run(input_data)
+
+ logger.info("Text generation completed successfully.")
+
+ if output:
+ # Extracted code from the palm completion
+ fixed_code = output['code_fix']
+ extracted_code = self.utils.extract_code(fixed_code)
+ 
+ # Check if the code or extracted code is not empty or null
+ if not code_snippet or not extracted_code:
+ raise Exception("Error: Generated code or extracted code is empty or null.")
+ else:
+ return extracted_code
+ else:
+ raise Exception("Error in code conversion: Please enter a valid code.")
+ else:
+ logger.error("Error in code conversion: Please enter a valid code and language.")
+ except Exception as exception:
+ logger.error(f"Error in code conversion: {traceback.format_exc()}")

libs/palmai.py

@@ -204,8 +204,7 @@ def fix_generated_code(self, code, code_language, fix_instructions=""):
  max_output_tokens=self.max_output_tokens,
  top_k=self.top_k,
  top_p=self.top_p,
- stop_sequences=[],
- safety_settings=[{"category":"HARM_CATEGORY_DEROGATORY","threshold":1},{"category":"HARM_CATEGORY_TOXICITY","threshold":1},{"category":"HARM_CATEGORY_VIOLENCE","threshold":2},{"category":"HARM_CATEGORY_SEXUAL","threshold":2},{"category":"HARM_CATEGORY_MEDICAL","threshold":2},{"category":"HARM_CATEGORY_DANGEROUS","threshold":2}],
+ stop_sequences=[]
  )
 
  if palm_completion:
@@ -225,4 +224,64 @@ def fix_generated_code(self, code, code_language, fix_instructions=""):
  logger.error("Error in code fixing: Please enter a valid code and language.")
  except Exception as exception:
  st.toast(f"Error in code fixing: {exception}", icon="❌")
- logger.error(f"Error in code fixing: {traceback.format_exc()}")
+ logger.error(f"Error in code fixing: {traceback.format_exc()}")
+
+ def convert_generated_code(self, code, code_language):
+ """
+ Function to convert the generated code to a different language using the palm API.
+ """
+ try:
+ # Check for valid code
+ if not code or len(code) == 0:
+ logger.error("Error in code conversion: Please enter a valid code.")
+ return
+ 
+ logger.info(f"Converting code")
+ if code and len(code) > 0:
+ logger.info(f"Converting code {code[:100]}... to language {code_language}")
+ 
+ # Improved instructions template
+ template = f"""
+ Task: Convert the code snippet provided below to the {code_language} programming language, following the given instructions:
+
+ {code}
+
+ Instructions for Conversion:
+ 1. Identify the functionality of the original code.
+ 2. Translate the code into the {code_language} programming language, maintaining the same functionality.
+ 3. Verify that the converted code is displayed in the output.
+
+ Please make sure only the converted code should be included in the output.
+ """
+
+ # Prompt Templates
+ code_template = template
+ 
+ # LLM Chains definition
+ # Create a chain that generates the code
+ palm_completion = palm.generate_text(
+ model=self.model,
+ prompt=code_template,
+ temperature=self.temperature,
+ max_output_tokens=self.max_output_tokens,
+ top_k=self.top_k,
+ top_p=self.top_p,
+ stop_sequences=[]
+ )
+
+ if palm_completion:
+ # Extracted code from the palm completion
+ code = palm_completion.result
+ extracted_code = self.utils.extract_code(code)
+ 
+ # Check if the code or extracted code is not empty or null
+ if not code or not extracted_code:
+ raise Exception("Error: Generated code or extracted code is empty or null.")
+ else:
+ return extracted_code
+ else:
+ raise Exception("Error in code conversion: Please enter a valid code.")
+ else:
+ logger.error("Error in code conversion: Please enter a valid code and language.")
+ except Exception as exception:
+ logger.error(f"Error in code conversion: {traceback.format_exc()}")

script.py

@@ -281,7 +281,7 @@ def main():
 
  with st.form('code_controls_form'):
  # Create columns for alignment
- file_name_col, save_code_col,generate_code_col,run_code_col,fix_code_col = st.columns(5)
+ file_name_col, save_code_col,generate_code_col,run_code_col,debug_code_col,convert_code_col = st.columns(6)
 
  # Input Box (for entering the file name) in the first column
  with file_name_col:
@@ -352,11 +352,11 @@ def main():
  st.session_state.generated_code = ""
  logger.error(f"Please select a valid AI option selected '{st.session_state.ai_option}' option")
 
- # Fix Code button in the fourth column
- with fix_code_col:
- fix_submitted = st.form_submit_button("Debug")
+ # Debug Code button in the fourth column
+ with debug_code_col:
+ debug_submitted = st.form_submit_button("Debug")
  ai_llm_selected = None
- if fix_submitted:
+ if debug_submitted:
  # checking for the selected AI option
  if st.session_state.ai_option == "Palm AI":
  ai_llm_selected = st.session_state.palm_langchain
@@ -372,6 +372,23 @@ def main():
  logger.info(f"Fixing code with instructions: {st.session_state.code_fix_instructions}")
  st.session_state.generated_code = ai_llm_selected.fix_generated_code(st.session_state.generated_code, st.session_state.code_language,st.session_state.code_fix_instructions)
 
+ # Debug Code button in the fourth column
+ with convert_code_col:
+ convert_submitted = st.form_submit_button("Convert")
+ ai_llm_selected = None
+ if convert_submitted:
+ # checking for the selected AI option
+ if st.session_state.ai_option == "Palm AI":
+ ai_llm_selected = st.session_state.palm_langchain
+ elif st.session_state.ai_option == "Gemini AI":
+ ai_llm_selected = st.session_state.gemini_langchain
+ elif st.session_state.ai_option == "Open AI":
+ ai_llm_selected = st.session_state.openai_langchain
+ 
+ logger.info(f"Converting code with instructions: {st.session_state.code_fix_instructions}")
+ st.session_state.generated_code = ai_llm_selected.convert_generated_code(st.session_state.generated_code, st.session_state.code_language)
+
+
  # Run Code button in the fourth column
  with run_code_col:
  execute_submitted = st.form_submit_button("Execute")