Update app.py

app.py

@@ -7,7 +7,6 @@ from zipfile import ZipFile
 import logging
 import json
 import psutil
-from flask import Flask, request, jsonify
 import threading
 
 # Configure logging
@@ -15,6 +14,8 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Initialize Flask app
+from flask import Flask, request, jsonify
+
 app = Flask(__name__)
 
 connected_cpus = {}
@@ -89,44 +90,34 @@ def get_cpu_info():
         info.append(f"{host}: {data['cpu_count']} CPUs, {data['usage']}% usage")
     return "\n".join(info)
 
-# Gradio interface
-def gradio_interface():
-    interface_inputs = [
-        gr.File(label="Upload Folder", file_count="multiple", file_types=['file']),
-        gr.Textbox(label="Python Script Name")
-    ]
-    interface_outputs = [
-        gr.Textbox(label="Log Output", interactive=False),
-        gr.File(label="Download Output Folder"),
-        gr.Textbox(label="Connected CPUs Info", interactive=False)
-    ]
-    
+# Function to run Flask app in background
+def run_flask_app():
+    app.run(host='0.0.0.0', port=7860, threaded=True, debug=False)
+
+# Function to launch Gradio interface
+def launch_gradio_interface():
     iface = gr.Interface(
         fn=handle_upload,
-        inputs=interface_inputs,
-        outputs=interface_outputs,
+        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,
         theme="light"  # Specify a theme that works, "light" is an example
     )
-    
-    return iface
+    iface.launch()
 
-# Function to run Flask app
-def run_flask_app():
-    app.run(host='0.0.0.0', port=7860)
-
-# Function to run Gradio interface
-def run_gradio_interface():
-    iface = gradio_interface()
-    iface.launch(port=7860)
-
-# Start Flask and Gradio interfaces in separate threads
+# Start Flask app in a separate thread
 if __name__ == "__main__":
     flask_thread = threading.Thread(target=run_flask_app)
-    gradio_thread = threading.Thread(target=run_gradio_interface)
-    
     flask_thread.start()
-    gradio_thread.start()
-    
+
+    # Launch Gradio interface in the main thread
+    launch_gradio_interface()
+
     flask_thread.join()
-    gradio_thread.join()