Move client api to backend.py

app/streamlit/app.py

@@ -1,28 +1,9 @@
 #!/usr/bin/env python
 # coding: utf-8
 
-import base64
-from io import BytesIO
-
-import requests
 import streamlit as st
-from PIL import Image
-
-
-class ServiceError(Exception):
-    def __init__(self, status_code):
-        self.status_code = status_code
-
-
-def get_images_from_backend(prompt, backend_url):
-    r = requests.post(backend_url, json={"prompt": prompt})
-    if r.status_code == 200:
-        images = r.json()["images"]
-        images = [Image.open(BytesIO(base64.b64decode(img))) for img in images]
-        return images
-    else:
-        raise ServiceError(r.status_code)
 
+from backend import ServiceError, get_images_from_backend, get_model_version
 
 st.sidebar.markdown(
     """

app/streamlit/backend.py

@@ -0,0 +1,30 @@
+# Client requests to Dalle-Mini Backend server
+
+import requests
+from io import BytesIO
+import base64
+from PIL import Image
+
+class ServiceError(Exception):
+    def __init__(self, status_code):
+        self.status_code = status_code
+
+def get_images_from_backend(prompt, backend_url):
+    r = requests.post(
+        backend_url, 
+        json={"prompt": prompt}
+    )
+    if r.status_code == 200:
+        images = r.json()["images"]
+        images = [Image.open(BytesIO(base64.b64decode(img))) for img in images]
+        return images
+    else:
+        raise ServiceError(r.status_code)
+
+def get_model_version(url):
+    r = requests.get(url)
+    if r.status_code == 200:
+        version = r.json()["version"]
+        return version
+    else:
+        raise ServiceError(r.status_code)