Use log to replace print

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 
-from urllib.parse import urlparse
+import logging
 
 import subprocess
 import threading
@@ -10,6 +10,10 @@ import os
 
 from giskard.settings import settings
 
+logger = logging.getLogger(__name__)
+logging.getLogger().setLevel(logging.INFO)
+logging.getLogger("giskard").setLevel(logging.INFO)
+
 
 LOG_FILE = "output.log"
 
@@ -45,17 +49,17 @@ def run_ml_worker(url, api_key, hf_token):
         stdout=open(LOG_FILE, "w"), stderr=subprocess.STDOUT
     )
     args = ml_worker.args
-    print(f"Process {args} exited with {ml_worker.wait()}")
+    logging.info(f"Process {args} exited with {ml_worker.wait()}")
     ml_worker = None
 
 
 def stop_ml_worker():
     global ml_worker, previous_url
     if ml_worker is not None:
-        print(f"Stopping ML worker for {previous_url}")
+        logging.info(f"Stopping ML worker for {previous_url}")
         ml_worker.terminate()
         ml_worker = None
-        print("ML worker stopped")
+        logging.info("ML worker stopped")
         return "ML worker stopped"
     return "ML worker not started"
 
@@ -67,7 +71,7 @@ def start_ml_worker(url, api_key, hf_token):
     # Always run an external ML worker
     stop_ml_worker()
 
-    print(f"Starting ML worker for {url}")
+    logging.info(f"Starting ML worker for {url}")
     thread = threading.Thread(target=run_ml_worker, args=(url, api_key, hf_token))
     thread.start()
     return f"ML worker running for {url}"