update

app.py

@@ -1,4 +1,3 @@
-from fastapi import FastAPI
 import gradio as gr
 import subprocess
 import sys
@@ -10,16 +9,10 @@ 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,
@@ -74,7 +67,7 @@ def find_result_file():
     return None
 
 def run_bigcodebench(command):
-    global is_running, process
+    global is_running
     with lock:
         if is_running:
             yield "A command is already running. Please wait for it to finish.\n"
@@ -84,12 +77,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, preexec_fn=os.setsid)
+        process = subprocess.Popen(command.split(), stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True)
         
         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"
@@ -104,13 +97,6 @@ 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
@@ -146,24 +132,6 @@ 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")
@@ -206,17 +174,5 @@ with gr.Blocks() as demo:
                  inputs=[command_output, jsonl_file, subset, split],
                  outputs=[log_output, download_btn])
 
-@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)
+demo.queue(max_size=300).launch(share=True, server_name="0.0.0.0", server_port=7860)
 scheduler = BackgroundScheduler()