Update app.py via AI Editor

app.py

@@ -12,6 +12,7 @@ import threading
 import tempfile
 import shutil
 import logging
+import json
 from flask import request, make_response, g
 from dash.exceptions import PreventUpdate
 import PyPDF2
@@ -306,10 +307,9 @@ def handle_file_upload_and_delete(list_of_contents, delete_clicks, list_of_names
                     return get_file_cards(session_data['uploaded_files'])
             # Handle delete button click
             elif "delete-file-btn" in prop_id:
-                import ast
-                btn_id = prop_id.split('.')[0]
                 try:
-                    btn_id_dict = ast.literal_eval(btn_id)
+                    btn_id = prop_id.split('.')[0]
+                    btn_id_dict = json.loads(btn_id)
                     filename = btn_id_dict['index']
                 except Exception as e:
                     logger.warning(f"Could not extract filename from delete prop_id: {prop_id} error: {e}")
@@ -381,9 +381,8 @@ def handle_matrix_and_chat(*args):
         raise PreventUpdate
     triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
     if "matrix-btn" in triggered_id:
-        import ast
         try:
-            triggered = ast.literal_eval(triggered_id)
+            triggered = json.loads(triggered_id)
             matrix_type = triggered['index']
         except Exception:
             raise PreventUpdate