Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import bz2
 
-def analyze_cps(file):
+def analyze_and_unpack_cps(file):
     if file is None:
         return "", ""
     
@@ -9,16 +9,20 @@ def analyze_cps(file):
         with open(file.name, "rb") as f:
             raw_data = f.read()
 
-        # Показываем первые 256 байт в HEX
         hex_dump = ' '.join(f'{byte:02x}' for byte in raw_data[:256])
         hex_display = f"HEX дамп (первые 256 байт):\n{hex_dump}"
 
-        # Пробуем распаковать через bz2
+        # Проверяем заголовок
+        if raw_data.startswith(b'OPS1'):
+            compressed_data = raw_data[4:]  # Пропускаем OPS1
+        else:
+            compressed_data = raw_data
+
         try:
-            decompressed = bz2.decompress(raw_data)
+            decompressed = bz2.decompress(compressed_data)
             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
 
@@ -26,14 +30,14 @@ def analyze_cps(file):
         return f"Ошибка чтения файла: {e}", ""
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Анализ .cps файла")
+    gr.Markdown("## Распаковка .cps файла (OPS1 + BZIP2)")
 
     file_input = gr.File(label="Загрузите .cps файл", file_types=[".cps"])
     hex_output = gr.Textbox(label="HEX дамп файла", lines=5)
-    text_output = gr.Textbox(label="Распакованный текст / ошибка", lines=10)
+    text_output = gr.Textbox(label="Распакованный текст", lines=15)
 
     file_input.change(
-        fn=analyze_cps,
+        fn=analyze_and_unpack_cps,
         inputs=file_input,
         outputs=[hex_output, text_output]
     )