Update app.py

app.py

@@ -1,12 +1,19 @@
 import streamlit as st
 from streamlit_drawable_canvas import st_canvas
+
+from dropbox.files import WriteMode
+
 from PIL import Image
 import numpy as np
 import cv2
 import hashlib
-import pyimgur
+import dropbox
 import time
+import hmac
 import os
+st.set_page_config(layout="wide")
+dbx = dropbox.Dropbox(st.secrets["dbx"])
+password = st.secrets["dbx"]
 
 # Function to convert the canvas drawing to a binary mask
 def canvas_to_mask(canvas_result, img_shape):
@@ -79,31 +86,59 @@ def delete_files_in_folder(folder_path):
         print(f"Error deleting files: {e}")
 
 
-st.title("IB Geo IA Survey")
+def check_password():
+    """Returns `True` if the user had the correct password."""
 
-# Upload an image
+    def password_entered():
+        """Checks whether a password entered by the user is correct."""
+        if hmac.compare_digest(st.session_state["password"], password):
+            st.session_state["password_correct"] = True
+            del st.session_state["password"]  # Don't store the password.
+        else:
+            st.session_state["password_correct"] = False
+
+    # Return True if the password is validated.
+    if st.session_state.get("password_correct", False):
+        return True
+
+    # Show input for password.
+    st.text_input(
+        "Password", type="password", on_change=password_entered, key="password"
+    )
+    if "password_correct" in st.session_state:
+        st.error("😕 Password incorrect")
+    return False
 
+def backup(LOCALFILE, BACKUPPATH):
+    with open(LOCALFILE, 'rb') as f:
+        # We use WriteMode=overwrite to make sure that the settings in the file
+        # are changed on upload
+        print("Uploading " + LOCALFILE + " to Dropbox as " + BACKUPPATH + "...")
+        dbx.files_upload(f.read(), BACKUPPATH, mode=WriteMode('overwrite'))
 
-image = Image.open("img/map.png").convert("RGB")
+st.title("IB Geography Survey")
+
+# Upload an image
+
+image = Image.open("img/dlmap2.png").convert("RGB")
 img_array = np.array(image)
-width = st.slider('Stroke width', 0, 20, 5)
 
 # Create a canvas for drawing
-st.subheader("Highlight the central business district area:")
+st.subheader("Highlight the central business district:")
 canvas_result = st_canvas(
     fill_color="rgba(255, 165, 0, 0.7)",  # Use an orange, semi-transparent fill
-    stroke_width=width,
+    stroke_width=5,
     stroke_color="rgba(255, 165, 0, 0.7)",
-    background_image=Image.open("img/map.png"),
+    background_image=Image.open("img/dlmap2.png"),
     update_streamlit=True,
     height=img_array.shape[0],
     width=img_array.shape[1],
     drawing_mode="freedraw",
     key="canvas",
 )
-im = pyimgur.Imgur(
-    "253ebfef9de391c"
-)
+#im = pyimgur.Imgur(
+#    "253ebfef9de391c"
+#)
 if st.button("Save"):
 
     mask = canvas_to_mask(canvas_result, img_array.shape)
@@ -111,9 +146,10 @@ if st.button("Save"):
         mask = fill_enclosed_areas(mask)
         cur = generate_short_hash()
         cv2.imwrite(f"data/{cur}.png", mask)
-        uploaded_image = im.upload_image(f"data/{cur}.png", title="Data Backup")
-        st.markdown(uploaded_image.title)
-        st.markdown(uploaded_image.link)
+        # uploaded_image = im.upload_image(f"data/{cur}.png", title="Data Backup")
+        # st.markdown(uploaded_image.title)
+        # st.markdown(uploaded_image.link)
+        backup(f"data/{cur}.png", f"/geo/{cur}.png")
     else:
         st.warning("Please draw on the image.")
 
@@ -122,7 +158,7 @@ if st.button("Aggregate data"):
     mean_mask = calculate_mean_mask("data")
 
     if mean_mask is not None:
-        final_image = overlay_mask("img/map.png", mean_mask)
+        final_image = overlay_mask("img/dlmap2.png", mean_mask)
 
         st.image(final_image, caption='Where most people think the CBD is', use_column_width=True)
 
@@ -137,4 +173,5 @@ if st.button("Aggregate data"):
 
 
 if st.button("Clear data"):
-    delete_files_in_folder("data")
+    if check_password():
+        delete_files_in_folder("data")