update

README.md

@@ -1,6 +1,6 @@
 ---
-title: Bigcodebench Evaluator
-emoji: 📈
+title: BigCodeBench Evaluator
+emoji: 💻
 colorFrom: indigo
 colorTo: indigo
 sdk: docker

app.py

@@ -8,23 +8,7 @@ import uuid
 import glob
 import shutil
 from pathlib import Path
-
-class Logger:
-    def __init__(self, filename):
-        self.terminal = sys.stdout
-        self.log = open(filename, "w")
-
-    def write(self, message):
-        self.terminal.write(message)
-        self.log.write(message)
-        self.log.flush()
-
-    def flush(self):
-        self.terminal.flush()
-        self.log.flush()
-
-    def isatty(self):
-        return False
+from apscheduler.schedulers.background import BackgroundScheduler
 
 default_command = "bigcodebench.evaluate"
 is_running = False
@@ -91,7 +75,7 @@ def run_bigcodebench(command):
     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"
@@ -107,12 +91,14 @@ def run_bigcodebench(command):
 
 def stream_logs(command, jsonl_file=None):
     global is_running
+    local_filename = None
     if jsonl_file is not None:
         local_filename = os.path.basename(jsonl_file.name)
+        
     if is_running:
         yield "A command is already running. Please wait for it to finish.\n"
         return
-
+    
     cleanup_previous_files(local_filename)
     yield "Cleaned up previous files.\n"
 
@@ -148,8 +134,6 @@ with gr.Blocks() as demo:
         download_btn = gr.DownloadButton(label="Download Result", visible=False)
     log_output = gr.Textbox(label="Execution Logs", lines=20)
     
-    # Hidden component to store the result file path
-    # result_file_path = gr.State("")
 
     def update_command(*args):
         return generate_command(*args)
@@ -172,7 +156,7 @@ with gr.Blocks() as demo:
             print(f"Result file: {result_file}")
             return (gr.update(label="Evaluation completed. Result file found."),
                     gr.Button(visible=False),
-                    gr.DownloadButton("Download Result", value=result_file, visible=True))
+                    gr.DownloadButton(label="Download Result", value=result_file, visible=True))
         else:
             return (gr.update(label="Evaluation completed. No result file found."),
                     gr.Button("Run Evaluation", visible=True),
@@ -183,4 +167,7 @@ with gr.Blocks() as demo:
                  outputs=[log_output, submit_btn, download_btn])
 
 if __name__ == "__main__":
-    demo.queue(max_size=300).launch(server_name="0.0.0.0", server_port=7860)
+    demo.queue(max_size=300).launch(server_name="0.0.0.0", server_port=7860)
+    scheduler = BackgroundScheduler()
+    scheduler.add_job(restart_space, "interval", hours=3) # restarted every 3h as backup in case automatic updates are not working
+    scheduler.start()