import json
import os
import re
from deep_translator import GoogleTranslator

def process_json_files(start, end, step, length=0, tlang="en", spaces_include=False, strip_in_braces=True, strip_diacritics=True):
    base_path = "resources/texts"
    translator = GoogleTranslator(source='auto', target=tlang)
    results = []

    for i in range(start, end + 1):
        file_name = f"{base_path}/{i:02}.json"
        try:
            with open(file_name, 'r', encoding='utf-8') as file:
                data = json.load(file)
                text_blocks = data["text"]

                full_text = ""
                for block in text_blocks:
                    full_text += ' '.join(block)

                clean_text = full_text
                if strip_in_braces:
                    clean_text = re.sub(r"\[.*?\]", "", clean_text, flags=re.DOTALL)
                if strip_diacritics:
                    clean_text = re.sub(r"[^\u05D0-\u05EA ]+", "", clean_text)
                if not spaces_include:
                    clean_text = clean_text.replace(" ", "")

                if length != 0:
                    selected_characters = clean_text[step - 1::step][:length]
                else:
                    selected_characters = clean_text[step - 1::step]  # If length is 0, select all characters from step

                translated_text = translator.translate(''.join(selected_characters))
                if selected_characters != "":
                    results.append({
                        "book": i,
                        "title": data["title"],
                        "original_text": selected_characters,
                        "translated_text": translated_text
                    })

        except FileNotFoundError:
            results.append({"error": f"File {file_name} not found."})
        except json.JSONDecodeError:
            results.append({"error": f"File {file_name} could not be read as JSON."})
        except KeyError:
            results.append({"error": f"Expected key 'text' is missing in {file_name}."})

    return results