load_model.py

import pickle
import json

def load_sms_extractor():
    """Load the SMS Transaction Extractor"""
    return pickle.load(open("extractor.pkl", "rb"))

def extract_from_sms(sms_text):
    """Extract transaction data from SMS"""
    extractor = load_sms_extractor()
    return extractor.extract(sms_text)

# Example usage
if __name__ == "__main__":
    sms = "A/c 2XXXXX9311 credited by Rs. 2500 on 15-08-2025"
    result = extract_from_sms(sms)
    print(json.dumps(result, indent=2))