Update app.py

app.py

@@ -8,29 +8,37 @@ import logging
 import json
 import threading
 import psutil
+from flask import Flask, request, jsonify
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Initialize Flask app
+app = Flask(__name__)
+
 connected_cpus = {}
 
-# Function to donate CPU (Workaround)
-def donate_cpu_handler(data):
+# Endpoint to donate CPU resources
+@app.route('/donate_cpu', methods=['POST'])
+def donate_cpu_handler():
+    data = request.get_json()
     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 json.dumps({"status": "success", "message": f"CPU donated by {host}"})
+    return jsonify({"status": "success", "message": f"CPU donated by {host}"})
 
-# Function to update CPU usage (Workaround)
-def update_cpu_usage_handler(data):
+# Endpoint to update CPU usage
+@app.route('/update_cpu_usage', methods=['POST'])
+def update_cpu_usage_handler():
+    data = request.get_json()
     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 json.dumps({"status": "success"})
+    return jsonify({"status": "success"})
 
 # Function to run the provided Python script using MPI
 def run_script(script_name, folder_path):
@@ -103,6 +111,8 @@ def gradio_interface():
     
     iface.launch()
 
-# Launch the Gradio interface
+# Launch the Gradio interface using Flask's run method
 if __name__ == "__main__":
     gradio_interface()
+    # Uncomment the line below if using Flask's run method for local testing
+    # app.run(host='0.0.0.0', port=7860)