Merge pull request haseeb-heaven#14 from haseeb-heaven:streamlit

Added Example Code Snippets

Author: haseeb-heaven

libs/code_runner.py

@@ -56,7 +56,7 @@ def get_lang_code(self, lang):
  self.logger.error(f"Language {lang} not found")
  return None
 
- def run_code(self, code, language):
+ def run_code(self, code, language,code_input=None,compile_only=False):
  try:
  if not code or not language:
  self.logger.error("Code or language is not provided.")
@@ -73,7 +73,7 @@ def run_code(self, code, language):
 
  # Replace escape sequences with empty characters
  code = code.replace("\\n", "").replace("\\t", "").replace("\\r", "").replace("\"", "\"")
- data = {"code": code, "language": language}
+ data = {"code": code, "language": language,"input" : code_input,"compileOnly" : compile_only}
 
  self.logger.info(f"Sending request to {self.server_url}/run_code and data is '{data}'")
  response = requests.post(f"{self.server_url}/run_code", json=data)

libs/geminiai.py

@@ -240,7 +240,9 @@ def convert_generated_code(self, code, code_language):
  if gemini_completion:
  # Extracted code from the palm completion
  code = gemini_completion.text
- extracted_code = self.utils.extract_code(code)
+ extracted_code = None
+ if code:
+ 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:

libs/general_utils.py

@@ -72,7 +72,7 @@ def execute_code(self, compiler_mode: str):
  # Execute code using JDoodle API
  elif compiler_mode.lower() == "api":
  logger.info("Executing code using JDoodle API")
- code_output = self.code_runer.run_code(generated_code, code_language)
+ code_output = self.code_runer.run_code(generated_code, code_language,code_input=st.session_state.code_input,compile_only=False)
  logger.info(f"Execution Output: {code_output}")
  return code_output
 

libs/tasks_parser.py

@@ -0,0 +1,48 @@
+import json
+from libs.logger import logger
+import random
+
+class CodingTasksParser:
+ """A class to parse coding tasks from a JSON file."""
+
+ def __init__(self):
+ """Initialize the parser with the file path."""
+ self.file_path = 'data/coding_tasks.json'
+ self._data = None
+ self._parse()
+
+ def _parse(self):
+ """Parse the JSON file and store the data."""
+ try:
+ with open(self.file_path, 'r') as f:
+ self._data = json.load(f)
+ logger.info(f'Successfully parsed file {self.file_path}')
+ except Exception as exception:
+ logger.error(f'Failed to parse file {self.file_path}: {exception}')
+ raise
+
+ def _get_tasks(self):
+ """Return a list of all tasks."""
+ if self._data is None:
+ raise ValueError('No data parsed yet')
+ return self._data['coding_tasks']
+
+ def _get_task(self, index):
+ """Return a specific task by its index."""
+ if self._data is None:
+ raise ValueError('No data parsed yet')
+ try:
+ return self._data['coding_tasks'][index]
+ except IndexError:
+ logger.error(f'No task at index {index}')
+ raise
+
+ def get_random_task(self):
+ """Return a random task."""
+ try:
+ tasks = self._get_tasks()
+ task = self._get_task(random.randint(0, len(tasks) - 1))
+ return task['task'], task['example']['input'], task['example']['output']
+ except Exception as exception:
+ logger.error(f'Failed to get random task: {exception}')
+ raise

libs/utils.py

@@ -62,6 +62,10 @@ def initialize_session_state():
  st.session_state.compiler_online_privacy_accepted = None
  if "compiler_api_privacy_accepted" not in st.session_state:
  st.session_state.compiler_api_privacy_accepted = None
+ if "general_utils" not in st.session_state:
+ st.session_state.general_utils = None
+ if "tasks_parser" not in st.session_state:
+ st.session_state.tasks_parser = None
 
  # Initialize session state for Vertex AI
  if "vertexai" not in st.session_state:
@@ -162,12 +166,12 @@ def load_css(file_name):
  st.markdown(f'<style>{f.read()}</style>', unsafe_allow_html=True)
 
 def display_code_editor(font_size, tab_size, theme, keybinding, show_gutter, show_print_margin, wrap, auto_update, readonly, language):
- if st.session_state.generated_code and st.session_state.compiler_mode == "Offline":
+ if st.session_state.generated_code and st.session_state.compiler_mode in ["Offline", "API"]:
  st.session_state.generated_code = st_ace(
  language=language.lower(),
  theme=theme,
  keybinding=keybinding,
- height=400,
+ height=600,
  value=st.session_state.generated_code,
  font_size=font_size,
  tab_size=tab_size,