added setuptools==69.2.0

app.py

@@ -3,8 +3,7 @@ import json
 import os
 from Chunker import CodeChunker
 from utils import count_tokens
-
-# Load JSON data for code file paths
+# Function to load JSON data
 def load_json_file(file_path):
     with open(file_path, 'r') as file:
         return json.load(file)
@@ -15,7 +14,9 @@ st.set_page_config(page_title="Cintra Code Chunker", layout="wide")
 # Assuming app.py and mock_codefiles.json are in the same directory
 json_file_path = os.path.join(os.path.dirname(__file__), 'mock_codefiles.json')
 code_files_data = load_json_file(json_file_path)
-code_files = code_files_data['files']
+
+# Extract filenames and contents
+code_files = list(code_files_data.keys())
 
 # UI Elements
 st.title('Cintra Code Chunker')
@@ -23,23 +24,34 @@ st.title('Cintra Code Chunker')
 # File selection
 selected_file_name = st.selectbox("Select a code file", code_files)
 
-# Assuming you have the path or the content in the JSON, adjust accordingly
-# This example assumes paths are stored in the JSON
-file_path = os.path.join(os.path.dirname(__file__), 'example_code_files', selected_file_name)
-with open(file_path, "r") as file:
-    code_content = file.read()
+# Assuming you have the content as a string in the JSON, extract directly
+code_content = code_files_data[selected_file_name]
+
+file_extension = selected_file_name.split('.')[-1]
+
+# Determine the language for syntax highlighting
+def get_language_by_extension(file_extension):
+    if file_extension in ['py', 'python']:
+        return 'python'
+    elif file_extension in ['js', 'jsx', 'javascript']:
+        return 'javascript'
+    elif file_extension == 'css':
+        return 'css'
+    else:
+        return None  # Default to no syntax highlighting if extension is not recognized
+
+language = get_language_by_extension(file_extension)
 
 col1, col2 = st.columns(2)
 
 with col1:
     st.subheader('Original File')
-    st.code(code_content, language='python')  # Adjust language dynamically based on file extension if necessary
+    st.code(code_content, language=language)
 
 with col2:
     token_chunk_size = st.sidebar.slider('Token Chunk Size Target', min_value=5, max_value=50, value=25)
     if st.sidebar.button("Chunk Code"):
-        # Initialize the code chunker, assuming it takes file extension and encoding name
-        file_extension = selected_file_name.split('.')[-1]
+        # Initialize the code chunker
         code_chunker = CodeChunker(file_extension=file_extension)
 
         # Chunk the code content
@@ -50,4 +62,4 @@ with col2:
         selected_chunk_key = st.selectbox("Select Chunk", options=chunk_keys)
 
         st.subheader('Chunked Code')
-        st.code(chunked_code_dict[selected_chunk_key], language='python')  # Adjust language dynamically based on file extension if necessary
+        st.code(chunked_code_dict[selected_chunk_key], language=language)

requirements.txt

@@ -43,3 +43,4 @@ typing_extensions==4.11.0
 tzdata==2024.1
 urllib3==2.2.1
 watchdog==4.0.0
+setuptools==69.2.0