Update app.py via AI Editor

app.py

@@ -20,10 +20,10 @@ SESSION_DATA = {}
 SESSION_LOCKS = {}
 
 def get_session_id():
-    session_id = flask.request.cookies.get('session-id')
-    if not session_id:
-        session_id = str(uuid.uuid4())
-    return session_id
+    sid = flask.request.cookies.get('session-id')
+    if not sid:
+        sid = str(uuid.uuid4())
+    return sid
 
 def get_session_dir(session_id):
     base_tmp = tempfile.gettempdir()
@@ -38,6 +38,7 @@ def clean_session(session_id):
             shutil.rmtree(session_dir)
         SESSION_DATA.pop(session_id, None)
         SESSION_LOCKS.pop(session_id, None)
+        logging.info(f"Cleaned session for {session_id}")
     except Exception as e:
         logging.error(f"Error cleaning session {session_id}: {e}")
 
@@ -212,7 +213,10 @@ app.layout = dbc.Container(
 )
 def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, session_data):
     trigger = ctx.triggered_id
-    session_id = get_session_id()
+    # Always get current session id from cookie
+    session_id = flask.request.cookies.get('session-id')
+    if not session_id:
+        session_id = str(uuid.uuid4())
     flask.g.session_id = session_id
     session_dir = get_session_dir(session_id)
     lock = get_session_lock(session_id)
@@ -224,6 +228,7 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, se
     if trigger == 'clear-session':
         clean_session(session_id)
         resp_data = {}
+        logging.info(f"Session cleared for {session_id}")
         return "", True, get_split_results_placeholder(), resp_data
 
     # Handle Delete Upload (detect ANY delete button press)
@@ -239,6 +244,7 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, se
         pdf_path = os.path.join(session_dir, orig_filename)
         if os.path.exists(pdf_path):
             os.remove(pdf_path)
+            logging.info(f"Deleted uploaded file {pdf_path} for session {session_id}")
         session_data = {}
         if os.path.exists(session_dir):
             for file in os.listdir(session_dir):
@@ -293,12 +299,14 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, se
     if trigger == 'split-btn':
         orig_filename = session_data.get('orig_filename')
         if not orig_filename:
+            logging.error(f"Split button clicked but no file to split for session {session_id}")
             return html.Div("No file to split.", style={'color': 'red'}), True, get_split_results_placeholder(), session_data
         pdf_path = os.path.join(session_dir, orig_filename)
         if not os.path.exists(pdf_path):
+            logging.error(f"Split button clicked but uploaded file not found for session {session_id}")
             return html.Div("Uploaded file not found. Please upload again.", style={'color': 'red'}), True, get_split_results_placeholder(), {}
         try:
-            logging.info(f"Splitting PDF for session {session_id} on user request.")
+            logging.info(f"Splitting PDF for session {session_id} on user request. File: {pdf_path}")
             with lock:
                 split_files = intelligent_pdf_split(pdf_path, session_dir)
                 zip_path = make_zip_of_splits(split_files, session_dir)