Update app.py

app.py

@@ -1,6 +1,26 @@
 import gradio as gr
 import bz2
 import bson
+import struct
+
+PARTICLE_FORMAT = "<iii ffff f ii iiiii I"  # 14 полей = 52 байта
+PARTICLE_SIZE = 128  # размер записи в parts
+
+FIELDS = [
+    "type", "life", "ctype",
+    "x", "y", "vx", "vy", "temp",
+    "tmp3", "tmp4", "flags", "tmp", "tmp2", "dcolour"
+]
+
+def parse_parts(binary_data):
+    count = len(binary_data) // PARTICLE_SIZE
+    particles = []
+    for i in range(count):
+        chunk = binary_data[i * PARTICLE_SIZE : i * PARTICLE_SIZE + PARTICLE_SIZE]
+        values = struct.unpack(PARTICLE_FORMAT, chunk[:52])
+        particle = dict(zip(FIELDS, values))
+        particles.append(particle)
+    return particles
 
 def analyze_cps(file):
     if file is None:
@@ -10,44 +30,47 @@ def analyze_cps(file):
         with open(file.name, "rb") as f:
             raw_data = f.read()
 
-        # HEX дамп оригинала
         hex_orig = ' '.join(f'{b:02x}' for b in raw_data[:256])
         hex_original = f"Оригинал (первые 256 байт):\n{hex_orig}"
 
-        # Проверка сигнатуры
         if raw_data[:4] != b'OPS1':
             return hex_original, "[Ошибка] Неверная сигнатура файла.", "", ""
 
-        # Пропустить заголовок (12 байт)
         compressed = raw_data[12:]
         try:
             decompressed = bz2.decompress(compressed)
         except Exception as e:
             return hex_original, f"[Ошибка bzip2]: {e}", "", ""
 
-        # HEX дамп распакованных данных
         hex_decomp = ' '.join(f'{b:02x}' for b in decompressed[:512])
         hex_decomp_view = f"Распакованные данные (первые 512 байт):\n{hex_decomp}"
 
-        # Парсим BSON
         try:
-            bson_doc = bson.decode(decompressed)
-            json_like = str(bson_doc)
+            doc = bson.decode(decompressed)
         except Exception as e:
-            json_like = f"[Ошибка BSON]: {e}"
+            return hex_original, hex_decomp_view, f"[Ошибка BSON]: {e}", ""
+
+        # Парсим parts
+        if 'parts' in doc:
+            try:
+                binary_data = doc['parts']
+                parsed_parts = parse_parts(bytes(binary_data))
+                doc['parts'] = parsed_parts
+            except Exception as e:
+                doc['parts'] = f"[Ошибка парсинга parts]: {e}"
 
-        return hex_original, hex_decomp_view, json_like
+        return hex_original, hex_decomp_view, str(doc)
 
     except Exception as e:
         return f"[Ошибка]: {e}", "", "", ""
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Распаковка и анализ .cps (The Powder Toy Save)")
+    gr.Markdown("## Распаковка .cps с красивым parts")
     file_input = gr.File(label="Загрузите .cps файл", file_types=[".cps"])
 
     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)
+    bson_parsed = gr.Textbox(label="Распарсенный BSON с parts", lines=30)
 
     file_input.change(
         fn=analyze_cps,