Update app.py

app.py

@@ -17,16 +17,20 @@ def initialize_global_data():
     """
     if not os.path.exists(DATA_FILE):
         ds = load_dataset("gaeunseo/Taskmaster_sample_data", split="train")
-        data = list(ds)
-        with data_lock:
-            with open(DATA_FILE, "w", encoding="utf-8") as f:
-                json.dump(data, f, ensure_ascii=False, indent=2)
+        data = ds.to_pandas()
+        # 필요한 컬럼이 없으면 추가합니다.
+        if "used" not in data.columns:
+            data["used"] = False
+        if "overlapping" not in data.columns:
+            data["overlapping"] = ""
+        if "text" not in data.columns:
+            data["text"] = ""
+        data.to_csv(DATA_FILE, index=False)
         return data
     else:
         with data_lock:
-            with open(DATA_FILE, "r", encoding="utf-8") as f:
-                data = json.load(f)
-        return data
+            df = pd.read_csv(DATA_FILE)
+        return df
 
 def load_global_data():
     with data_lock: