Automatically start ML worker if env provided

app.py

@@ -15,6 +15,11 @@ logging.getLogger().setLevel(logging.INFO)
 logging.getLogger("giskard").setLevel(logging.INFO)
 
 
+GSK_HUB_URL = 'GSK_HUB_URL'
+GSK_API_KEY = 'GSK_API_KEY'
+HF_SPACE_HOST = 'SPACE_HOST'
+HF_SPACE_TOKEN = 'GSK_HUB_HFS'
+
 LOG_FILE = "output.log"
 
 def read_logs():
@@ -83,15 +88,13 @@ theme = gr.themes.Soft(
 with gr.Blocks(theme=theme) as iface:
     with gr.Row():
         with gr.Column():
+            url = os.environ.get(GSK_HUB_URL) if os.environ.get(GSK_HUB_URL) else f"http://{settings.host}:{settings.ws_port}"
             url_input = gr.Textbox(
                 label="Giskard Hub URL",
-                value=f"htts://{os.environ.get('SPACE_HOST')}"
-                    if os.environ.get("SPACE_HOST")
-                    else f"http://{settings.host}:{settings.ws_port}",
+                value=url,
             )
             api_key_input = gr.Textbox(
                 label="Giskard Hub API Key",
-                value=os.environ.get("GSK_API_KEY"),
                 type="password",
                 placeholder="gsk-xxxxxxxxxxxxxxxxxxxxxxxxxxxx",
             )
@@ -113,5 +116,8 @@ with gr.Blocks(theme=theme) as iface:
     iface.load(read_logs, None, logs, every=0.5)
     iface.load(read_status, None, output, every=5)
 
+if os.environ.get(GSK_HUB_URL) and os.environ.get(GSK_API_KEY):
+    start_ml_worker(os.environ.get(GSK_HUB_URL), os.environ.get(GSK_API_KEY), os.environ.get(HF_SPACE_TOKEN))
+
 iface.queue()
 iface.launch()