Update app.py

app.py

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