clean up code

text_to_image.py

@@ -128,11 +128,6 @@ def start_inference_server(fc: FairClient):
            detach=True)
 
 
-def start_services(fc):
-    start_tunnel(fc)
-    start_inference_server(fc)
-
-
 def text_to_image(text):
     global endpoint_client
     global fair_client
@@ -140,18 +135,19 @@ def text_to_image(text):
         fair_client = create_fair_client()
 
     try:
-        if endpoint_client is None:  # try connecting to the server
-            endpoint_client = create_endpoint_client(fair_client, 1)
-        else:  # try inference
+        # client is configured, try to do inference right away
+        if endpoint_client is not None:
             return endpoint_client.infer(text)
+        # client is not configured, try connecting to the inference server, maybe it is running
+        else:
+            endpoint_client = create_endpoint_client(fair_client, 1)
     except (requests.exceptions.ConnectionError, requests.exceptions.HTTPError, requests.exceptions.Timeout):
-        endpoint_client = None
-
-    # start all services
-    if endpoint_client is None:
-        start_services(fair_client)
+        # inference server is not ready, start inference server and open the tunnel
+        start_inference_server(fair_client)
+        start_tunnel(fair_client)
         endpoint_client = create_endpoint_client(fair_client, retries=10)
 
+    # run inference
     return endpoint_client.infer(text)