fixed copy

app.py

@@ -3,9 +3,6 @@ import streamlit as st
 from utils import im_2_b64
 import pickle
 from upload import upload_file, get_file
-import clipboard
-
-RANDOM_SEED = 42
 
 
 st.title("ChatGPT with Vision")
@@ -18,9 +15,11 @@ 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)
+    with st.spinner("Loading chat..."):
+        id = st.query_params["id"]
+        data = get_file(id, 'chatgpt-vision-007')
+        st.session_state.messages = pickle.loads(data)
+        st.query_params.clear()
 
 def clear_uploader():
     st.session_state["uploader_key"] += 1
@@ -35,28 +34,29 @@ def undo():
 def share():
     data = pickle.dumps(st.session_state.messages)
     id = upload_file(data, 'chatgpt-vision-007')
-    return id
+    url = f"https://umbc-nlp-chatgpt-vision.hf.space/?id={id}"
+    st.success(f"Share URL: {url}")
 
 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()
+        share()
 
-    if st.button("Clear chat"):
-        st.session_state.messages = []
-        clear_uploader()
+    cols = st.columns(2)
+    with cols[0]:
+        if st.button("Undo"):
+            undo()
+    with cols[1]:
+        if st.button("Clear chat"):
+            st.session_state.messages = []
+            clear_uploader()
 
     with st.expander("Advanced Configuration"):
         st.subheader("Temperature")
-        temperature = st.slider(label="x", min_value=0.1, max_value=1.0, value=0.5, step=0.1, label_visibility='hidden')
+        temperature = st.slider(label="x", min_value=0.1, max_value=1.0, value=0.2, step=0.1, label_visibility='collapsed')
         st.subheader("Max Tokens")
-        max_tokens = st.slider(label="x", min_value=32, max_value=1024, value=256, step=32, label_visibility='hidden')
+        max_tokens = st.slider(label="x", min_value=32, max_value=1024, value=256, step=32, label_visibility='collapsed')
+        st.subheader("Random Seed")
+        random_seed = st.number_input("Seed", min_value=0, max_value=1000000, value=42, step=1, label_visibility='collapsed')
 
     with st.expander("Image Input", expanded=True):
         images = st.file_uploader(
@@ -122,13 +122,12 @@ if prompt := st.chat_input("Type a message", key="chat_input", disabled=chat_inp
             {"role": m["role"], "content": m["content"]}
             for m in st.session_state.messages
         ]
-        # print("api call", messages)
         chat_input_disabled = True
         stream = client.chat.completions.create(
             model="gpt-4-vision-preview",
             messages=messages,
             stream=True,
-            seed=RANDOM_SEED,
+            seed=random_seed,
             temperature=temperature,
             max_tokens=max_tokens,
         )

requirements.txt

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

upload.py

@@ -25,11 +25,3 @@ def get_file(file_name, bucket=st.secrets["S3_BUCKET"]):
     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))