added share func

.gitignore

@@ -0,0 +1,2 @@
+.streamlit
+__pycache__

.streamlit/config.toml

@@ -1,3 +0,0 @@
-[server]
-enableXsrfProtection = false
-enableCORS = false

.streamlit/secrets.toml

@@ -1 +0,0 @@
-OPENAI_KEY="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"

app.py

@@ -1,6 +1,9 @@
 from openai import OpenAI
 import streamlit as st
 from utils import im_2_b64
+import pickle
+from upload import upload_file, get_file
+import clipboard
 
 RANDOM_SEED = 42
 
@@ -14,11 +17,37 @@ if "messages" not in st.session_state:
 if "uploader_key" not in st.session_state:
     st.session_state["uploader_key"] = 0
 
+if "id" in st.query_params:
+    id = st.query_params["id"]
+    data = get_file(id, 'chatgpt-vision-007')
+    st.session_state.messages = pickle.loads(data)
+
 def clear_uploader():
     st.session_state["uploader_key"] += 1
     st.rerun()
 
+def undo():
+    if len(st.session_state.messages) > 0:
+        st.session_state.messages.pop()
+        st.session_state.messages.pop()
+        st.rerun()
+
+def share():
+    data = pickle.dumps(st.session_state.messages)
+    id = upload_file(data, 'chatgpt-vision-007')
+    return id
+
 with st.sidebar:
+    if st.button("Share"):
+        id = share()
+        url = f"https://umbc-nlp-chatgpt-vision.hf.space/?id={id}"
+        # st.code(f"https://umbc-nlp-chatgpt-vision.hf.space/?id={id}")
+        clipboard.copy(url)
+        st.write(f"URL copied to clipboard: {url}")
+
+    if st.button("Undo"):
+        undo()
+
     if st.button("Clear chat"):
         st.session_state.messages = []
         clear_uploader()

requirements.txt

@@ -1,2 +1,5 @@
 streamlit
-openai
+openai
+pydrive
+boto3
+clipboard

upload.py

@@ -0,0 +1,35 @@
+import streamlit as st
+import uuid
+import boto3
+
+s3 = boto3.client(
+    's3',
+    aws_access_key_id=st.secrets["AWS_ACCESS_KEY"],
+    aws_secret_access_key=st.secrets["AWS_SECRET_KEY"]
+)
+
+def upload_file(data, bucket=st.secrets["s3_bucket"]):
+    file_name = uuid.uuid4().hex
+    try:
+        key = f"{file_name}.pkl"
+        response = s3.put_object(Body=data, Bucket=bucket, Key=key)
+    except Exception as e:
+        return None
+    return file_name
+
+
+def get_file(file_name, bucket=st.secrets["s3_bucket"]):
+    try:
+        response = s3.get_object(Bucket=bucket, Key=f"{file_name}.pkl")
+        return response['Body'].read()
+    except Exception as e:
+        print(e)
+        return False
+    
+
+import pickle
+a = [1, 2, 3]
+id = upload_file(pickle.dumps(a), 'chatgpt-vision-007')
+print(id)
+data = get_file(id, 'chatgpt-vision-007')
+print(pickle.loads(data))