adjust to work with a new server

fair.py

@@ -79,10 +79,10 @@ class FairApiClient:
 
         return response['id'], response['pid']
 
-    def get_job_status(self, job_id):
-        url = f"{self.server_address}/jobs/{job_id}/status"
-        response = self.get(url=url)
-        return response.text
+    def get_job_info(self, job_id):
+        url = f"{self.server_address}/jobs/{job_id}/stat"
+        response = self.get(url=url).json()
+        return response
 
     def get_cluster_summary(self):
         url = f"{self.server_address}/nodes/summary"
@@ -143,15 +143,15 @@ class ServerNotReadyException(Exception):
     pass
 
 
-def wait_for_server(retries, timeout):
+def wait_for_server(retries, timeout, delay=1.0):
     for i in range(retries):
         try:
-            r = requests.get(ENDPOINT_ADDRESS)
+            r = requests.get(ENDPOINT_ADDRESS, timeout=timeout)
             r.raise_for_status()
             return
         except (requests.exceptions.ConnectionError, requests.exceptions.HTTPError, requests.exceptions.Timeout):
             logger.info("Server is not ready yet")
-            time.sleep(timeout)
+            time.sleep(delay)
     else:
         raise ServerNotReadyException("Failed to start the server")
 
@@ -172,12 +172,12 @@ def start_server():
 
     logger.info(f"Job id: {job_id}, pid: {job_pid}")
 
-    status = client.get_job_status(job_id=job_id)
-    logger.info(status)
+    info = client.get_job_info(job_id=job_id)
+    logger.info(info)
 
-    while status != "Processing":
-        status = client.get_job_status(job_id=job_id)
-        logger.info(status)
+    while info["status"] != "Processing":
+        info = client.get_job_info(job_id=job_id)
+        logger.info(info)
         time.sleep(0.5)
 
     # wait until the server is ready
@@ -186,7 +186,7 @@ def start_server():
 
 def text_to_image(text):
     try:
-        wait_for_server(retries=1, timeout=0.0)
+        wait_for_server(retries=1, timeout=0.1, delay=0.0)
     except ServerNotReadyException:
         start_server()