correctly GPU ABORT

app/session_utils.py

@@ -138,29 +138,10 @@ def remove_session_hash_code_data(session_hash_code: str):
                 logging.warning(f"[{session_hash_code}] Failed to update {ACTIVE_SESSIONS_HASH_FILE}: {e}")
 
         # --- Define all possible session_hash_code file patterns ---
-        files_to_remove = [
-            get_active_task_flag_file(session_hash_code),
-            get_active_stream_flag_file(session_hash_code),
-        ]
-
         # --- Remove all temporary files ---
-        for fname in files_to_remove:
-            path = os.path.join(TMP_DIR, fname)
-            if os.path.exists(path):
-                try:
-                    os.remove(path)
-                    logging.debug(f"[{session_hash_code}] Removed file: {fname}")
-                except Exception as e:
-                    logging.warning(f"[{session_hash_code}] Failed to remove file {fname}: {e}")
-
-        # --- Remove chunk folder if exists ---
-        chunk_dir = os.path.join(TMP_DIR, f"chunks_{session_hash_code}")
-        if os.path.isdir(chunk_dir):
-            try:
-                shutil.rmtree(chunk_dir)
-                logging.debug(f"[{session_hash_code}] Removed chunk folder: chunks_{session_hash_code}")
-            except Exception as e:
-                logging.warning(f"[{session_hash_code}] Failed to remove chunk folder: {e}")
+        remove_active_task_flag_file(session_hash_code)
+        remove_active_stream_flag_file(session_hash_code)
+        remove_chunk_folder(session_hash_code)
 
         logging.info(f"[{session_hash_code}] session_hash_code fully reset.")
 
@@ -259,5 +240,16 @@ def remove_active_task_flag_file(session_hash_code: str):
         except Exception as e:
             logging.warning(f"[{session_hash_code}] Failed to remove file {fname}: {e}")
 
+
+def remove_chunk_folder(session_hash_code: str) :
+    # --- Remove chunk folder if exists ---
+    chunk_dir = os.path.join(TMP_DIR, f"chunks_{session_hash_code}")
+    if os.path.isdir(chunk_dir):
+        try:
+            shutil.rmtree(chunk_dir)
+            logging.debug(f"[{session_hash_code}] Removed chunk folder: chunks_{session_hash_code}")
+        except Exception as e:
+            logging.warning(f"[{session_hash_code}] Failed to remove chunk folder: {e}")
+
 def get_session_hashe_chunks_dir(session_hash_code: str):
      return os.path.join(TMP_DIR, f"{NAME_FOLDER_CHUNKS}{session_hash_code}")

app/ui_utils.py

@@ -5,6 +5,7 @@ import os
 
 from app.utils import (
     remove_active_task_flag_file,
+    remove_chunk_folder,
     task_fake,
     is_active_task,
     is_active_stream,
@@ -456,7 +457,8 @@ def start_task_asr_ast(
         # --- ERROR HANDLING (GPU ABORT / RUNTIME ERROR) ---
         error_msg = str(e)
         logging.error(f"Task Error for {session_hash_code}: {error_msg}", exc_info=True)
-
+        remove_active_task_flag_file(session_hash_code)
+        remove_chunk_folder(session_hash_code)
         # Detect specific Hugging Face / GPU errors
         if "GPU task aborted" in error_msg or "CUDA out of memory" in error_msg or "Device" in error_msg:
             display_msg = f"🛑 **System Error:** GPU Task Aborted. The model may have run out of memory. ({error_msg})"

app/utils.py

@@ -20,6 +20,7 @@ from app.session_utils import (
     get_active_stream_flag_file,
     remove_active_stream_flag_file,
     remove_active_task_flag_file,
+    remove_chunk_folder,
     get_session_hashe_chunks_dir
 )
 from app.supported_languages import (
@@ -189,7 +190,7 @@ asr_model = None
 # ASR_MODEL = load_model()
 
 
-@spaces.GPU
+@spaces.GPU(duration=10)
 def task_fake(session_hash_code: str,
         task_type, lang_source, lang_target,
         chunk_secs, left_context_secs, right_context_secs,
@@ -256,22 +257,13 @@ def task_fake(session_hash_code: str,
         return
         
     finally:
-        if os.path.exists(active_flag):
-            os.remove(active_flag)
-        logging.info(f"[{session_hash_code}] task stopped.")
-        
-        try:
-            if os.path.exists(chunk_dir) and not os.listdir(chunk_dir):
-                os.rmdir(chunk_dir)
-                logging.debug(f"[{session_hash_code}] Cleaned up empty chunk dir.")
-        except Exception as e:
-            logging.error(f"[{session_hash_code}] Cleanup error: {e}")
-            yield (f"Cleanup error: {e}", "error", None)
-            
+        remove_active_task_flag_file(session_hash_code)
+        remove_chunk_folder(session_hash_code)
         logging.info(f"[{session_hash_code}] Exiting task loop.")
         # yield ("Task finished and cleaned up.", "done", None)
 
 
+
 @spaces.GPU
 def task(session_hash_code: str,
         task_type, lang_source, lang_target,
@@ -356,9 +348,7 @@ def task(session_hash_code: str,
         yield (f"Unexpected error: {e}", "error", None)
         
     finally:
-        if os.path.exists(active_flag):
-            os.remove(active_flag)
-        logging.info(f"[{session_hash_code}] task stopped.")
+        remove_active_task_flag_file(session_hash_code)
         
         try:
             if os.path.exists(chunk_dir) and not os.listdir(chunk_dir):