Update app.py

app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-from cryptography.fernet import Fernet
 import time
 import pandas as pd
 import io
@@ -10,27 +9,8 @@ import plotly.express as px
 
 st.subheader("Table Question Answering (QA)", divider="blue")
 
-# generate Fernet key
-if 'fernet_key' not in st.session_state:
-    st.session_state.fernet_key = Fernet.generate_key()
 
-key = st.session_state.fernet_key
-
-
-# function for generating and validating fernet key
-def generate_fernet_token(key, data):
-    fernet = Fernet(key)
-    token = fernet.encrypt(data.encode())
-    return token
-
-def validate_fernet_token(key, token, ttl_seconds):
-    
-    fernet = Fernet(key)
-    try:
-        decrypted_data = fernet.decrypt(token, ttl=ttl_seconds).decode()
-        return decrypted_data, None
-    except Exception as e:
-        return None, f"Expired token: {e}"
+ 
 
 # sidebar
 with st.sidebar:
@@ -154,40 +134,10 @@ if upload_file is not None:
         st.warning("Unsupported file type.")
         st.stop()
 
-# generate and validate Fernet token for the current file
-if 'fernet_token' not in st.session_state:
-    if 'df' in st.session_state: 
-        df = st.session_state.df
-    
-        st.session_state.fernet_token = generate_fernet_token(key, df.to_json())
-    else:
-        st.stop()
 
-decrypted_data_streamlit, error_streamlit = validate_fernet_token(key, st.session_state.fernet_token, ttl_seconds=3600)
-
-if error_streamlit:
-    st.warning("Please press Request Authorization. Please note that a file should be uploaded before you press Request Authorization.")
-    if st.button("Request Authorization"):
-        if 'df' in st.session_state: 
-            df = st.session_state.df
-            st.session_state.fernet_token = generate_fernet_token(key, df.to_json())
-            st.success("Authorization granted")
-            decrypted_data_streamlit, error_streamlit = validate_fernet_token(key, st.session_state.fernet_token, ttl_seconds=3600)
-        if error_streamlit:
-            st.error(f"Your authorization has expired: {error_streamlit}")
-            st.stop()
-        if error_streamlit:
-            st.error("Please upload a file.")
-            st.stop()
-        else:
-            try:
-                df = pd.read_json(decrypted_data_streamlit)
-            except Exception as e:
-                st.error(f"Error decoding data: {e}")
-                st.stop()
-    else:
-        st.error(f"Your authorization has expired: {error_streamlit}")
-        st.stop()
+
+
+ 
 
 st.divider()