Create utils.py

utils.py

@@ -0,0 +1,22 @@
+import streamlit as st
+from PIL import Image
+import io
+import pyperclip
+
+def load_image(uploaded_file):
+    image_bytes = uploaded_file.read()
+    image = Image.open(io.BytesIO(image_bytes))
+    return image
+
+def save_caption(caption):
+    # Function can be added here to save the caption to some database
+    print(f"Saved caption: {caption}")
+
+def copy_to_clipboard(text):
+    pyperclip.copy(text)
+    st.success("Caption copied to clipboard!")
+
+def get_image_file_buffer(image):
+    buf = io.BytesIO()
+    image.save(buf, format='PNG')
+    return buf.getvalue()