update

app.py

@@ -1,3 +1,4 @@
+from fastapi import FastAPI
 import gradio as gr
 import subprocess
 import sys
@@ -9,10 +10,16 @@ import glob
 import shutil
 from pathlib import Path
 from apscheduler.schedulers.background import BackgroundScheduler
+import signal
+import uvicorn
+
 
 default_command = "bigcodebench.evaluate"
 is_running = False
 lock = threading.Lock()
+process = None
+
+app = FastAPI()
 
 def generate_command(
     jsonl_file, split, subset, parallel,
@@ -67,7 +74,7 @@ def find_result_file():
     return None
 
 def run_bigcodebench(command):
-    global is_running
+    global is_running, process
     with lock:
         if is_running:
             yield "A command is already running. Please wait for it to finish.\n"
@@ -77,12 +84,12 @@ def run_bigcodebench(command):
     try:
         yield f"Executing command: {command}\n"
         
-        process = subprocess.Popen(command.split(), stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True)
+        process = subprocess.Popen(command.split(), stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True, preexec_fn=os.setsid)
         
         for line in process.stdout:
             yield line
         
-        # process.wait()
+        process.wait()
         
         if process.returncode != 0:
             yield f"Error: Command exited with status {process.returncode}\n"
@@ -97,6 +104,13 @@ def run_bigcodebench(command):
     finally:
         with lock:
             is_running = False
+            process = None
+
+def kill_process():
+    global process
+    if process:
+        os.killpg(os.getpgid(process.pid), signal.SIGTERM)
+        process = None
 
 def stream_logs(command, jsonl_file=None):
     global is_running
@@ -132,6 +146,24 @@ with gr.Blocks() as demo:
         check_gt_only = gr.Checkbox(label="Check GT Only")
         no_gt = gr.Checkbox(label="No GT")
     
+    kill_process_btn = gr.Button("Kill Process", visible=False)
+    kill_process_btn.click(kill_process)
+
+    # Add this JavaScript to handle window closing
+    gr.HTML("""
+    <script>
+    window.addEventListener('beforeunload', function (e) {
+        fetch('/kill_process', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+            },
+            body: JSON.stringify({}),
+        });
+    });
+    </script>
+    """)
+    
     command_output = gr.Textbox(label="Command", value=default_command, interactive=False)
     with gr.Row():
         submit_btn = gr.Button("Run Evaluation")
@@ -174,5 +206,21 @@ with gr.Blocks() as demo:
                  inputs=[command_output, jsonl_file, subset, split],
                  outputs=[log_output, download_btn])
 
-demo.queue(max_size=300).launch(share=True, server_name="0.0.0.0", server_port=7860)
+@app.get("/")
+def read_main():
+    return {"message": "This is your main app"}
+
+@app.post("/kill_process")
+async def api_kill_process():
+    kill_process()
+    return {"status": "success"}
+
+# demo.queue(max_size=300).launch(
+#     share=True, 
+#     server_name="0.0.0.0", 
+#     server_port=7860,
+#     additional_routes={"/kill_process": kill_process_api}
+# )
+app = gr.mount_gradio_app(app, demo, path="/gradio")
+uvicorn.run(app, host="0.0.0.0", port=7860)
 scheduler = BackgroundScheduler()