First version working

.gitignore

@@ -1,3 +1,4 @@
+.idea
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

requirements.txt

@@ -0,0 +1 @@
+streamlit~=0.76.0

session_state.py

@@ -0,0 +1,86 @@
+# From https://gist.github.com/okld/0aba4869ba6fdc8d49132e6974e2e662
+from streamlit.hashing import _CodeHasher
+
+try:
+    # Before Streamlit 0.65
+    from streamlit.ReportThread import get_report_ctx
+    from streamlit.server.Server import Server
+except ModuleNotFoundError:
+    # After Streamlit 0.65
+    from streamlit.report_thread import get_report_ctx
+    from streamlit.server.server import Server
+
+
+class SessionState:
+    def __init__(self, session, hash_funcs):
+        """Initialize SessionState instance."""
+        self.__dict__["_state"] = {
+            "data": {},
+            "hash": None,
+            "hasher": _CodeHasher(hash_funcs),
+            "is_rerun": False,
+            "session": session,
+        }
+
+    def __call__(self, **kwargs):
+        """Initialize state data once."""
+        for item, value in kwargs.items():
+            if item not in self._state["data"]:
+                self._state["data"][item] = value
+
+    def __getitem__(self, item):
+        """Return a saved state value, None if item is undefined."""
+        return self._state["data"].get(item, None)
+
+    def __getattr__(self, item):
+        """Return a saved state value, None if item is undefined."""
+        return self._state["data"].get(item, None)
+
+    def __setitem__(self, item, value):
+        """Set state value."""
+        self._state["data"][item] = value
+
+    def __setattr__(self, item, value):
+        """Set state value."""
+        self._state["data"][item] = value
+
+    def clear(self):
+        """Clear session state and request a rerun."""
+        self._state["data"].clear()
+        self._state["session"].request_rerun()
+
+    def sync(self):
+        """Rerun the app with all state values up to date from the beginning to fix rollbacks."""
+
+        # Ensure to rerun only once to avoid infinite loops
+        # caused by a constantly changing state value at each run.
+        #
+        # Example: state.value += 1
+        if self._state["is_rerun"]:
+            self._state["is_rerun"] = False
+
+        elif self._state["hash"] is not None:
+            if self._state["hash"] != self._state["hasher"].to_bytes(self._state["data"], None):
+                self._state["is_rerun"] = True
+                self._state["session"].request_rerun()
+
+        self._state["hash"] = self._state["hasher"].to_bytes(self._state["data"], None)
+
+
+def get_session():
+    session_id = get_report_ctx().session_id
+    session_info = Server.get_current()._get_session_info(session_id)
+
+    if session_info is None:
+        raise RuntimeError("Couldn't get your Streamlit Session object.")
+
+    return session_info.session
+
+
+def get_state(hash_funcs=None):
+    session = get_session()
+
+    if not hasattr(session, "_custom_session_state"):
+        session._custom_session_state = SessionState(session, hash_funcs)
+
+    return session._custom_session_state

streamlit_app.py

@@ -0,0 +1,76 @@
+from PIL import Image
+import streamlit as st
+import booste
+
+from session_state import SessionState, get_state
+
+# Unfortunately Streamlit sharing does not allow to hide enviroment variables yet.
+# Do not copy this API key, go to https://www.booste.io/ and get your own, it is free!
+BOOSTE_API_KEY = "3818ba84-3526-4029-9dc8-ef3038697ea2"
+
+
+task_name: str = st.sidebar.radio("Task", options=["Image classification", "Image ranking", "Prompt ranking"])
+
+st.markdown("# CLIP playground")
+st.markdown("### Try OpenAI's CLIP model in your browser")
+st.markdown(" "); st.markdown(" ")
+with st.beta_expander("What is CLIP?"):
+    st.markdown("Nice CLIP explaination")
+st.markdown(" "); st.markdown(" ")
+if task_name == "Image classification":
+    session_state = get_state()
+    uploaded_image = st.file_uploader("Upload image", type=[".png", ".jpg", ".jpeg"],
+                                      accept_multiple_files=False)
+    st.markdown("or choose one from")
+    col1, col2, col3 = st.beta_columns(3)
+    with col1:
+        default_image_1 = "https://cdn.pixabay.com/photo/2014/10/13/21/34/clipper-487503_960_720.jpg"
+        st.image(default_image_1, use_column_width=True)
+        if st.button("Select image 1"):
+            session_state.image = default_image_1
+    with col2:
+        default_image_2 = "https://cdn.pixabay.com/photo/2019/12/17/18/20/peacock-4702197_960_720.jpg"
+        st.image(default_image_2, use_column_width=True)
+        if st.button("Select image 2"):
+            session_state.image = default_image_2
+    with col3:
+        default_image_3 = "https://cdn.pixabay.com/photo/2016/11/15/16/24/banana-1826760_960_720.jpg"
+        st.image(default_image_3, use_column_width=True)
+        if st.button("Select image 3"):
+            session_state.image = default_image_3
+    raw_classes = st.text_input("Enter the classes to chose from separated by a comma."
+                                " (f.x. `banana, sailing boat, honesty, apple`)")
+    if raw_classes:
+        session_state.processed_classes = raw_classes.split(",")
+        input_prompts = ["A picture of a " + class_name for class_name in session_state.processed_classes]
+
+col1, col2 = st.beta_columns([2, 1])
+with col1:
+    st.markdown("Image to classify")
+    if session_state.image is not None:
+        st.image(session_state.image, use_column_width=True)
+    else:
+        st.warning("Select an image")
+
+with col2:
+    st.markdown("Classes to choose from")
+    if session_state.processed_classes is not None:
+        for class_name in session_state.processed_classes:
+            st.write(class_name)
+    else:
+        st.warning("Enter the classes to classify from")
+
+# Possible way of customize this https://discuss.streamlit.io/t/st-button-in-a-custom-layout/2187/2
+if st.button("Predict"):
+    with st.spinner("Predicting..."):
+        clip_response = booste.clip(BOOSTE_API_KEY,
+                        prompts=input_prompts,
+                        images=[session_state.image],
+                        pretty_print=True)
+        st.write(clip_response)
+
+
+session_state.sync()
+
+
+