upl

app.py

@@ -40,9 +40,21 @@ def compute(dim):
     samples = np.array(api.model.sample(torch.tensor(random_normal_samples(1000,api.scaled.shape[-1])).float()).detach())
     
     return api.scaler.inverse_transform(samples)
+
+with st.form('login_form'):
+    st.write('Token for generation:')
+    token = st.text_input()
+    submit = st.form_submit_button('Submit')
+
+if token in st.secrets['tokens'] and submit:
+    
+    if uploaded_file is not None:
+        dims = len(uploaded_file.getvalue().decode("utf-8").split('\n')[0].split(','))-1
+        samples=compute(dims)
+        st.download_button('Download generated CSV', pd.DataFrame(samples).to_csv(), 'text/csv')
     
+    elif not uploaded_file:
+        st.write('Upload your file')
     
-if uploaded_file is not None:
-    dims = len(uploaded_file.getvalue().decode("utf-8").split('\n')[0].split(','))-1
-    samples=compute(dims)
-    st.download_button('Download generated CSV', pd.DataFrame(samples).to_csv(), 'text/csv')
+    else:
+        st.markdown('## :red[You dont have access]')