Update app.py

app.py

@@ -15,22 +15,22 @@ logger = logging.getLogger(__name__)
 
 connected_cpus = {}
 
-# Function to donate CPU
-def donate_cpu(data):
+# Function to donate CPU (Workaround)
+def donate_cpu_handler(data):
     host = data['host']
     cpu_count = data['cpu_count']
     connected_cpus[host] = {"cpu_count": cpu_count, "usage": 0.0}
     logger.info(f"CPU donated by {host} with {cpu_count} CPUs.")
-    return {"status": "success", "message": f"CPU donated by {host}"}
+    return json.dumps({"status": "success", "message": f"CPU donated by {host}"})
 
-# Function to update CPU usage
-def update_cpu_usage(data):
+# Function to update CPU usage (Workaround)
+def update_cpu_usage_handler(data):
     host = data['host']
     usage = data['usage']
     if host in connected_cpus:
         connected_cpus[host]['usage'] = usage
         logger.info(f"Updated CPU usage for {host}: {usage}%")
-    return {"status": "success"}
+    return json.dumps({"status": "success"})
 
 # Function to run the provided Python script using MPI
 def run_script(script_name, folder_path):
@@ -83,27 +83,23 @@ def get_cpu_info():
 
 # Gradio interface
 def gradio_interface():
-    with gr.Blocks() as demo:
-        gr.Markdown("## Python Script Executor with Distributed Computing")
-        
-        with gr.Row():
-            folder = gr.File(label="Upload Folder", file_count="multiple", file_types=['file'])
-            script_name = gr.Textbox(label="Python Script Name")
-        
-        log_output = gr.Textbox(label="Log Output", interactive=False)
-        output_folder = gr.File(label="Download Output Folder")
-        cpu_info = gr.Textbox(label="Connected CPUs Info", interactive=False)
-        
-        run_button = gr.Button("Run Script")
-        refresh_button = gr.Button("Refresh CPU Info")
-        
-        run_button.click(fn=handle_upload, inputs=[folder, script_name], outputs=[log_output, output_folder])
-        refresh_button.click(fn=get_cpu_info, inputs=[], outputs=[cpu_info])
-        
-        # Define the donate CPU endpoint
-        demo.api(donate_cpu, inputs=gr.JSON(), outputs=gr.JSON(), name="donate_cpu")
-    
-    demo.launch()
+    with gr.Interface(
+        fn=handle_upload,
+        inputs=[
+            gr.File(label="Upload Folder", file_count="multiple", file_types=['file']),
+            gr.Textbox(label="Python Script Name")
+        ],
+        outputs=[
+            gr.Textbox(label="Log Output", interactive=False),
+            gr.File(label="Download Output Folder"),
+            gr.Textbox(label="Connected CPUs Info", interactive=False)
+        ],
+        live=True,
+        capture_session=True,
+        theme="compact"
+    ) as iface:
+        iface.launch()
 
+# Launch the Gradio interface
 if __name__ == "__main__":
     gradio_interface()