Update app.py

app.py

@@ -1,54 +1,63 @@
 import gradio as gr
 import bz2
-from bson import BSON
+import bson
 
-def parse_cps_bson(file):
-    if not file:
-        return "", "", ""
+def analyze_cps(file):
+    if file is None:
+        return "", "", "", ""
 
     try:
         with open(file.name, "rb") as f:
-            raw = f.read()
+            raw_data = f.read()
 
-        if raw[:4] != b'OPS1':
-            return "Ошибка: неверная сигнатура файла", "", ""
+        # HEX дамп оригинала
+        hex_orig = ' '.join(f'{b:02x}' for b in raw_data[:256])
+        hex_original = f"Оригинал (первые 256 байт):\n{hex_orig}"
 
-        compressed = raw[12:]
-        decompressed = bz2.decompress(compressed)
+        # Проверка сигнатуры
+        if raw_data[:4] != b'OPS1':
+            return hex_original, "[Ошибка] Неверная сигнатура файла.", "", ""
 
-        # Парсим BSON
-        bson_data = BSON(decompressed)
-        decoded = bson_data.decode()
+        # Пропустить заголовок (12 байт)
+        compressed = raw_data[12:]
+        try:
+            decompressed = bz2.decompress(compressed)
+        except Exception as e:
+            return hex_original, f"[Ошибка bzip2]: {e}", "", ""
 
-        # Выводим красиво
-        import json
-        json_str = json.dumps(decoded, indent=2, ensure_ascii=False)
+        # HEX дамп распакованных данных
+        hex_decomp = ' '.join(f'{b:02x}' for b in decompressed[:512])
+        hex_decomp_view = f"Распакованные данные (первые 512 байт):\n{hex_decomp}"
 
-        # HEX дамп
-        hex_dump = ' '.join(f'{b:02x}' for b in decompressed[:512])
-        hex_view = f"Распакованные данные (первые 512 байт):\n{hex_dump}"
+        # Парсим BSON
+        try:
+            bson_doc = bson.decode(decompressed)
+            json_like = str(bson_doc)
+        except Exception as e:
+            json_like = f"[Ошибка BSON]: {e}"
 
-        return hex_view, json_str, ""
+        return hex_original, hex_decomp_view, json_like
 
     except Exception as e:
-        return "", f"[Ошибка]: {e}", ""
+        return f"[Ошибка]: {e}", "", "", ""
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Распаковка и парсинг .cps (BSON)")
-
+    gr.Markdown("## Распаковка и анализ .cps (The Powder Toy Save)")
     file_input = gr.File(label="Загрузите .cps файл", file_types=[".cps"])
-    hex_output = gr.Textbox(label="HEX дамп распакованных данных", lines=8)
-    json_output = gr.Textbox(label="Структура BSON (JSON)", lines=20)
+
+    hex_orig_out = gr.Textbox(label="HEX оригинала", lines=5)
+    hex_decomp_out = gr.Textbox(label="HEX распакованных данных", lines=10)
+    bson_parsed = gr.Textbox(label="Распарсенный BSON (Python dict)", lines=20)
 
     file_input.change(
-        fn=parse_cps_bson,
+        fn=analyze_cps,
         inputs=file_input,
-        outputs=[hex_output, json_output]
+        outputs=[hex_orig_out, hex_decomp_out, bson_parsed]
     )
 
     file_input.clear(
-        fn=lambda: ("", ""),
-        outputs=[hex_output, json_output]
+        fn=lambda: ("", "", ""),
+        outputs=[hex_orig_out, hex_decomp_out, bson_parsed]
     )
 
 demo.launch()