Update app.py

app.py

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