import json
from scripts.InflectionProcessor import InflectionProcessor


with open('samples/dabei_sein.json', 'r', encoding='utf-8') as f:
    entry = json.load(f)

print("Original entry forms length:", len(entry['forms']))

# Process it
processor = InflectionProcessor()
processed_entry = processor.process(entry)

print("Processed entry forms type:", type(processed_entry['forms']))
if isinstance(processed_entry['forms'], list):
    if processed_entry['forms'] and 'type' in processed_entry['forms'][0]:
        # Compressed array
        print("Number of compressed forms:", len(processed_entry['forms']))
        for i, form in enumerate(processed_entry['forms']):
            print(f"Form {i}: type={form['type']}, usage={form['data']['usage']}")
            print(f"  Infinitive: {form['data']['infinitive']}")
    else:
        # Uncompressed list
        print("Uncompressed forms list, length:", len(processed_entry['forms']))
elif isinstance(processed_entry['forms'], dict):
    print("Single compressed form")
    print(f"Type: {processed_entry['forms']['type']}")
    print(f"Usage: {processed_entry['forms']['data']['usage']}")
    print(f"Infinitive: {processed_entry['forms']['data']['infinitive']}")
else: