Use temp file for the result of api request

app.py

@@ -26,9 +26,8 @@ def infer(prompt):
     #     if re.search(rf"\b{filter}\b", prompt):
     #         raise gr.Error("Unsafe content found. Please try again with different prompts.")
     #
-    images = []
-    image = text_to_image(prompt)
-    image = np.array(Image.open(image).convert('RGB'))
+    image_file = text_to_image(prompt)
+    image = np.array(Image.open(image_file).convert('RGB'))
     images = [image]
     return images
     

fair.py

@@ -1,11 +1,12 @@
 import json
 import os
 import time
-from typing import List
+from typing import List, BinaryIO
 import logging
 logger = logging.getLogger()
 
 import requests
+import tempfile
 
 SERVER_ADRESS="https://faircompute.com:8000/api/v1"
 #SERVER_ADRESS="http://localhost:8000/api/v1"
@@ -99,7 +100,7 @@ def put_job_stream_eof(token, job_id, name):
     return response.text
 
 
-def wait_for_file(token, job_id, path, local_path, attempts=10):
+def wait_for_file(token, job_id, path, attempts=10) -> BinaryIO:
     headers = {
         'Authorization': f'Bearer {token}'
     }
@@ -109,17 +110,18 @@ def wait_for_file(token, job_id, path, local_path, attempts=10):
         try:
             with requests.get(url=url, headers=headers, stream=True) as r:
                 r.raise_for_status()
-                with open(local_path, 'wb') as f:
-                    for chunk in r.iter_content(chunk_size=8192):
-                        f.write(chunk)
+                f = tempfile.TemporaryFile()
+                for chunk in r.iter_content(chunk_size=8192):
+                    f.write(chunk)
 
-                print(f"File {local_path} ready")
-                return local_path
+                print(f"File {path} ready")
+                f.seek(0, 0)
+                return f
         except Exception as e:
             print(e)
             time.sleep(0.5)
 
-    print(f"Failed to receive {local_path}")
+    print(f"Failed to receive {path}")
 
 
 def text_to_image(text):
@@ -180,75 +182,16 @@ def text_to_image(text):
         logger.info("Done!")
     else:
         logger.info("Job Failed")
-    resp = wait_for_file(token=token,
+    file = wait_for_file(token=token,
                          job_id=job_id,
-                         path="%2Fworkspace%2Fresult.png",
-                         local_path="result.png")
-    logger.info(resp)
-    return resp
+                         path="%2Fworkspace%2Fresult.png")
+    return file
 
 
 if __name__=="__main__":
-    email = os.getenv('FAIRCOMPUTE_EMAIL')
-    password = os.environ.get('FAIRCOMPUTE_PASSWORD')
-    token = authenticate(email=email, password=password)
-
-    print(token)
-
-    summary = get_cluster_summary(token=token)
-    print("Summary:")
-    print(summary)
-    program_id = put_program(token=token,
-                             launcher="Docker",
-                             image=DOCKER_IMAGE,
-                             runtime="nvidia",
-                             command=[])
-    print(program_id)
-
-    job_id = put_job(token=token,
-                     program_id=program_id,
-                     input_files=[],
-                     output_files=["/workspace/result.png"])
-
-    print(job_id)
+    from PIL import Image
+    file = text_to_image(text="Robot dinozaur\n")
+    image = Image.open(file)
+    image.save("result.png")
 
-    status = get_job_status(token=token,
-                            job_id=job_id)
-    print(status)
-
-    while status != "Processing" and status != "Completed":
-         status = get_job_status(token=token,
-                                 job_id=job_id)
-         print(status)
-         time.sleep(0.5)
-
-    res = put_job_stream_data(token=token,
-                        job_id=job_id,
-                        name="stdin",
-                        data="Robot dinozaur\n")
-    print(res)
-
-    res = put_job_stream_eof(token=token,
-                       job_id=job_id,
-                       name="stdin")
-    print(res)
-
-    status = get_job_status(token=token,
-                            job_id=job_id)
-    print(status)
-
-    while status == "Processing":
-         status = get_job_status(token=token,
-                                 job_id=job_id)
-         print(status)
-         time.sleep(0.5)
-    if status == "Completed":
-        print("Done!")
-    else:
-        print("Job Failed")
-    resp = wait_for_file(token=token,
-                         job_id=job_id,
-                         path="%2Fworkspace%2Fresult.png",
-                         local_path="result.png")
-    print(resp)