feat: txt file support

app.py

@@ -1,4 +1,4 @@
-import json
+import json, re
 import gradio as gr
 import numpy as np
 
@@ -8,11 +8,36 @@ from taiko import (
     COURSE
 )
 
-def handle(chart_path, music_path):
-    data = json.loads(open(chart_path, "r").read())
+def txt_loads(data_str:str):
+    data = []
+    current_course = None
+    chart = []
 
+    for line in data_str.split('\n'):
+        if line.startswith("course:"):
+            if current_course is not None:
+                data.append({"course": current_course, "chart": chart})
+                chart = []
+            current_course = int(re.search(r'\d+', line).group())
+        elif line:
+            line = line.strip("[]")
+            nums = line.split(",")
+            chart.append([int(nums[0]), float(nums[1])])
+
+    if current_course is not None:
+        data.append({"course": current_course, "chart": chart})
+
+    output_data = {"data": data}
+    return output_data
+
+def handle(chart_path:str, music_path:str):
+    if chart_path.endswith(".json"):
+        data = json.loads(open(chart_path, "r").read())
+    elif chart_path.endswith(".txt"):
+        data = txt_loads(open(chart_path, "r").read())
+    
     if len(data["data"]) < 0 and len(data["data"]) > 5:
-        raise("Issue occur: Json data")
+        raise("Issue occur: chart data")
 
     for d in data["data"]:
         chart = preprocess(d["chart"])
@@ -31,7 +56,7 @@ def handle(chart_path, music_path):
 
 if __name__ == "__main__":
     inputs = [
-        gr.File(label="太鼓達人譜面/Taiko Chart Data (json)", file_types=[".json"]),
+        gr.File(label="太鼓達人譜面/Taiko Chart Data (json or txt)", file_types=[".json", ".txt"]),
         gr.File(label="譜面音樂/Chart Music (ogg) (Optional)", file_types=[".ogg"])
     ]
     outputs = [gr.Audio(label=course["label"], format="mp3") for course in COURSE]

taiko.py

@@ -54,7 +54,7 @@ def preprocess(data:list, offset:float=0):
             if balloon_count >= m[3]:
                 chart.append((BALLOON_BANG_WAV, offset + m[1]))
         else:
-            raise ValueError("Your json file has some problems.")
+            raise ValueError("Your chart file has some problems.")
     
     return chart