modules/case_loader.py

import os, json
from webtest_prompt import build_webtest_prompt
from inference import run_inference

BASE_DIR = os.path.dirname(os.path.dirname(__file__))  # modules/ 상위 폴더
TEST_CASES_PATH = os.path.join(BASE_DIR, "test_cases.json")

with open(TEST_CASES_PATH, "r", encoding="utf-8") as f:
    TEST_CASES = json.load(f)

def get_case_names():
    return [f"{i+1}. {c['description']}" for i, c in enumerate(TEST_CASES)]

def load_case(idx):
    case = TEST_CASES[idx]
    return json.dumps(case, ensure_ascii=False, indent=2), case["player_utterance"]

def run_case(idx, player_utt):
    case = TEST_CASES[idx].copy()
    case["player_utterance"] = player_utt
    prompt = build_webtest_prompt(case["npc_id"], case["npc_location"], player_utt)
    result = run_inference(prompt)
    return result["npc_output_text"], result["deltas"], result["flags_prob"]