again!

requirements.txt

@@ -0,0 +1,2 @@
+websocket-client
+icecream

app.py

@@ -150,6 +150,7 @@ def main():
 
             st.write(f"Your text prompt: {text_prompt}")
             with st.spinner('Processing... Please wait.'):
+                gif_runner = st.image("https://huggingface.co/spaces/LeonJHK/AMDsnap/blob/main/wait-waiting.gif")
                 ws = websocket.WebSocket()
                 try:
                     ws.connect("ws://{}/ws?clientId={}&token={}".format(server_address, client_id, TOKEN))
@@ -160,6 +161,8 @@ def main():
 
                 images = get_images(ws, workflow, bg_color, text_prompt, password, images_base64)
                 ws.close()
+                gif_runner.empty()  # Remove the GIF after processing is done
+                
                 
             if images:
                 st.write("Click on an image to view in full size.")