client_script.py

"""
client_script.py

Client-side demonstration of real EEG inference.

Workflow:
1. Load a saved raw EEG epoch (22 × T).
2. Load the fitted CSP transformer.
3. Compute CSP features.
4. Send the features to the FastAPI server.
5. Print the predicted motor-imagery class.
"""

import numpy as np
import joblib
import requests
from pathlib import Path

API_URL = "http://127.0.0.1:8000/predict"


def main():

    # 1. Load sample EEG epoch
    epoch_path = Path("sample_data/sample_epoch.npy")
    if not epoch_path.exists():
        raise FileNotFoundError("sample_epoch.npy not found.")

    epoch = np.load(epoch_path)  # shape (22, T)

    # 2. Load CSP transformer
    csp_path = Path("models/csp_transformer.joblib")
    if not csp_path.exists():
        raise FileNotFoundError("csp_transformer.joblib missing.")

    csp = joblib.load(csp_path)

    # 3. Compute CSP features
    epoch_reshaped = epoch.reshape(1, epoch.shape[0], epoch.shape[1])
    features = csp.transform(epoch_reshaped)[0].tolist()

    print("Extracted features:", features)

    # 4. Send features to API
    payload = {"features": features}
    response = requests.post(API_URL, json=payload)

    if response.status_code != 200:
        print("API error:", response.text)
        return

    # 5. Print prediction
    result = response.json()
    print("Predicted class:", result["prediction"])


if __name__ == "__main__":
    main()