Update app.py

app.py

@@ -4,25 +4,41 @@ import zlib
 
 def load_and_display_info(gameFilePath: str):
     """Загружает файл и возвращает game.title + game объект для дальнейшей обработки"""
-    with open(gameFilePath, "rb") as f:
-        compressed_data = f.read()
-    data = zlib.decompress(compressed_data)
-    game = fanl_yp.decode.decode(data)
-    title = getattr(game, 'title', 'Название не найдено')
-    return title, game  # Возвращаем game для последующего использования
+    if gameFilePath is None:
+        return "", None, gr.update(interactive=False)
+    
+    try:
+        with open(gameFilePath, "rb") as f:
+            compressed_data = f.read()
+        data = zlib.decompress(compressed_data)
+        game = fanl_yp.decode.decode(data)
+        title = getattr(game, 'title', 'Название не найдено')
+        return title, game, gr.update(interactive=True)
+    except Exception as e:
+        return f"Ошибка загрузки файла: {str(e)}", None, gr.update(interactive=False)
 
 def process_save_file(game_obj, original_path: str):
     """Обрабатывает сохранение и возвращает путь к новому файлу"""
-    output_path = original_path + "_fixed"
-    output = fanl_yp.encode.encode(game_obj, output_path)
-    output.flush()
-    return output_path
+    if game_obj is None or original_path is None:
+        return None
+    
+    try:
+        output_path = original_path + "_fixed"
+        output = fanl_yp.encode.encode(game_obj, output_path)
+        output.flush()
+        return output_path
+    except Exception as e:
+        raise gr.Error(f"Ошибка обработки файла: {str(e)}")
+
+def clear_handler():
+    """Обработчик очистки файла"""
+    return "", None, gr.update(interactive=False)
 
 with gr.Blocks() as demo:
     gr.Markdown("## Обработка сохранений игры")
 
     with gr.Row():
-        file_input = gr.File(label="Загрузите файл сохранения")
+        file_input = gr.File(label="Загрузите файл сохранения", file_types=None)
 
     title_output = gr.Textbox(label="Информация о сохранении", interactive=False)
 
@@ -37,12 +53,15 @@ with gr.Blocks() as demo:
     file_input.upload(
         fn=load_and_display_info,
         inputs=file_input,
-        outputs=[title_output, game_state],
+        outputs=[title_output, game_state, process_button],
         queue=False
-    ).then(
-        fn=lambda: gr.update(interactive=True),
+    )
+
+    # Когда файл удален — очищаем все поля и блокируем кнопку
+    file_input.clear(
+        fn=clear_handler,
         inputs=None,
-        outputs=process_button,
+        outputs=[title_output, game_state, process_button],
         queue=False
     )