Fix code conversion error and update language codes

Author: haseeb-heaven

libs/geminiai.py

@@ -249,4 +249,5 @@ def convert_generated_code(self, code, code_language):
  else:
  logger.error("Error in code conversion: Please enter a valid code and language.")
  except Exception as exception:
+ st.toast(f"Error in code conversion: {exception}", icon="❌")
  logger.error(f"Error in code conversion: {traceback.format_exc()}")

libs/general_utils.py

@@ -6,7 +6,7 @@
 import subprocess
 import traceback
 import streamlit as st
-from libs.lang_codes import LangCodes
+from libs.lang_codes import get_language_codes
 import shutil
 import threading
 import time
@@ -113,19 +113,20 @@ def generate_dynamic_html(self,language, code_prompt):
  <script src="https://www.jdoodle.com/assets/jdoodle-pym.min.js" type="text/javascript"></script>
  </body>
  </html>
- """.format(language=LangCodes()[language], script_code=code_prompt)
+ """.format(language=get_language_codes()[language], script_code=code_prompt)
  return html_template
 
 
  def check_compilers(self, language):
- language = language.lower().strip()
- 
+ language_code=get_language_codes()[language]
+ logger.info(f"Checking compilers for language: {language} with lang_code: {language_code}")
+
  compilers = {
  "python": ["python", "--version"],
- "javascript": ["node", "--version"],
+ "nodejs": ["node", "--version"],
  "c": ["gcc", "--version"],
- "c++": ["g++", "--version"],
- "c#": ["csc", "--version"],
+ "cpp": ["g++", "--version"],
+ "csharp": ["csc", "--version"],
  "go": ["go", "--version"],
  "ruby": ["ruby", "--version"],
  "java": ["java", "--version"],
@@ -134,12 +135,12 @@ def check_compilers(self, language):
  "swift": ["swift", "--version"]
  }
 
- if language not in compilers:
- 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="❌")
+ if language_code not in compilers:
+ logger.error(f"Invalid language selected '{language_code}' not found in compilers list.")
+ st.toast(f"Invalid language selected '{language_code}' not found in compilers list.", icon="❌")
  return False
 
- compiler = subprocess.run(compilers[language], capture_output=True, text=True)
+ compiler = subprocess.run(compilers[language_code], capture_output=True, text=True)
  if compiler.returncode != 0:
  logger.error(f"{language.capitalize()} compiler not found.")
  st.toast(f"{language.capitalize()} compiler not found.", icon="❌")

libs/lang_codes.py

@@ -1,4 +1,4 @@
-def LangCodes():
+def get_language_codes():
 
  LANGUAGE_CODES = {
  'C': 'c',
@@ -11,7 +11,7 @@ def LangCodes():
  'Swift': 'swift',
  'JavaScript': 'nodejs',
  'Kotlin': 'kotlin',
- 'Python': 'python3',
+ 'Python': 'python',
  'GO Lang': 'go',
  }
  return LANGUAGE_CODES

libs/palmai.py

@@ -284,4 +284,5 @@ def convert_generated_code(self, code, code_language):
  else:
  logger.error("Error in code conversion: Please enter a valid code and language.")
  except Exception as exception:
+ st.toast(f"Error in code conversion: {exception}", icon="❌")
  logger.error(f"Error in code conversion: {traceback.format_exc()}")

script.py

@@ -18,7 +18,7 @@
 import streamlit as st
 from libs.vertexai_langchain import VertexAILangChain
 from libs.general_utils import GeneralUtils
-from libs.lang_codes import LangCodes
+from libs.lang_codes import get_language_codes
 from libs.openai_langchain import OpenAILangChain
 from libs.logger import logger
 from libs.utils import *
@@ -65,7 +65,7 @@ def main():
  st.selectbox("Select AI", ["Open AI", "Vertex AI", "Palm AI","Gemini AI"], key="ai_option")
 
  # Dropdown for selecting code language
- st.selectbox("Select language", list(LangCodes().keys()), key="code_language")
+ st.selectbox("Select language", list(get_language_codes().keys()), key="code_language")
 
  # Radio buttons for selecting compiler mode
  st.radio("Compiler Mode", ("Online", "Offline"), key="compiler_mode")
@@ -432,7 +432,7 @@ def main():
  wrap = st.checkbox("Wrap", value=True)
  auto_update = st.checkbox("Auto Update", value=False)
  readonly = st.checkbox("Readonly", value=False)
- language = st.selectbox("Language", options=list(LangCodes().keys()), index=list(LangCodes().keys()).index("Python"))
+ language = st.selectbox("Language", options=list(get_language_codes().keys()), index=list(get_language_codes().keys()).index("Python"))
 
  # Display the st_ace code editor with the selected settings
  display_code_editor(font_size, tab_size, theme, keybinding, show_gutter, show_print_margin, wrap, auto_update, readonly, language)