From ed5d44ef9394b2fb53b7e10a09e4968985a9c214 Mon Sep 17 00:00:00 2001
From: Yuvraj Singh <satishyuvraj.singh3178@gmail.com>
Date: Tue, 2 Jun 2026 08:31:19 +0530
Subject: [PATCH] Fix: ensure temp file created during script instrumentation
 is always cleaned up

---
 app.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/app.py b/app.py
index 5f46d39..87cee31 100644
--- a/app.py
+++ b/app.py
@@ -3351,6 +3351,7 @@ def generate():
         execution = None
         stop_event = threading.Event()
         t_reader = None
+        temp_path_created = None
         try:
             # 1. Initialize execution record with arguments
             execution = _start_execution_record(
@@ -3374,6 +3375,8 @@ def generate():
                     temp_fd, temp_path = tempfile.mkstemp(
                         suffix=".sh", prefix=".tmp_run_", dir=temp_dir
                     )
+                    # Track created temp path so we can always clean it up
+                    temp_path_created = temp_path
                     with os.fdopen(
                         temp_fd, "w", encoding="utf-8", newline="\n"
                     ) as temp_f:
@@ -3623,7 +3626,7 @@ def stream_reader(stream, q):
                 proc,
                 execution,
                 run_id=run_id,
-                temp_path=run_path if run_path != full_path else None,
+                temp_path=(temp_path_created if temp_path_created is not None else (run_path if run_path != full_path else None)),
                 was_aborted=True,
                 error_message="Client disconnected",
                 stop_event=stop_event,
@@ -3636,7 +3639,7 @@ def stream_reader(stream, q):
                 proc,
                 execution,
                 run_id=run_id,
-                temp_path=run_path if run_path != full_path else None,
+                temp_path=(temp_path_created if temp_path_created is not None else (run_path if run_path != full_path else None)),
                 was_aborted=False,
                 error_message="Execution timed out",
                 stop_event=stop_event,
@@ -3654,7 +3657,7 @@ def stream_reader(stream, q):
                 proc,
                 execution,
                 run_id=run_id,
-                temp_path=run_path if run_path != full_path else None,
+                temp_path=(temp_path_created if temp_path_created is not None else (run_path if run_path != full_path else None)),
                 was_aborted=False,
                 error_message=str(e),
                 stop_event=stop_event,
@@ -3668,7 +3671,7 @@ def stream_reader(stream, q):
                 proc,
                 execution,
                 run_id=run_id,
-                temp_path=run_path if run_path != full_path else None,
+                temp_path=(temp_path_created if temp_path_created is not None else (run_path if run_path != full_path else None)),
                 stop_event=stop_event,
                 reader_thread=t_reader,
             )