Change

app.py

@@ -2,7 +2,6 @@ import os
 from fastapi import FastAPI
 from datasets import load_dataset
 import random
-import json
 
 # Set Hugging Face cache directory programmatically as a fallback
 os.environ["HF_HOME"] = "/app/cache"
@@ -19,15 +18,9 @@ async def get_idioms():
     # Extract required fields
     response = []
     for item in idioms:
-        # Parse the output field as JSON
-        try:
-            output_data = json.loads(item.get("output", "{}"))
-            compound_type = output_data.get("Compound Type", "").strip('"')
-        except json.JSONDecodeError:
-            compound_type = ""
         response.append({
             "idiom": item.get("compound", ""),
             "example": item.get("sentence", ""),
-            "definition": compound_type
+            "definition": item.get("output", ""),
         })
     return response