Update app.py

app.py

@@ -44,7 +44,7 @@ with st.sidebar:
         }
         """,
     ):
-        st.button("ONE-DAY SUBSCRIPTION")
+        st.button("DEMO APP")
    
 
     expander = st.expander("**Important notes on the Table Question Answering (QA) App**")
@@ -53,32 +53,24 @@ with st.sidebar:
     **Supported File Formats**
     This app accepts files in .csv and .xlsx formats.
     
-    
     **How to Use**
     Upload your file first. Then, type your question into the text area provided and click the 'Retrieve your answer' button.
     
-    
     **Usage Limits**
-    You can ask up to 30 questions per day. Once you reach this limit, you will need to wait until the daily automatic renewal to continue using the app. The app's daily renewal occurs automatically.
-    
+    You can ask up to 5 questions. 
     
     **Subscription Management**
-    This app uses a one-day subscription plan. To cancel your subscription, please visit your Account settings.
-    
+    This demo app offers a one-day subscription, expiring after 24 hours. If you are interested in building your own Table Question Answering (QA) Web App, we invite you to explore our NLP Web App Store on our website. You can select your desired features, place your order, and we will deliver your custom app in five business days. If you wish to delete your Account with us, please contact us at info@nlpblogs.com
     
     **Authorization**
-    For security purposes, your authorization access expires hourly. To restore access, click the "Request Authorization" button. A file must be uploaded before you can request authorization.
-    
+    For security purposes, your authorization access expires hourly. To restore access, click the 'Request Authorization' button.
     
     **Customization**
     To change the app's background color to white or black, click the three-dot menu on the right-hand side of your app, go to Settings and then Choose app theme, colors and fonts.
     
-    
     **File Handling and Errors**
-    The app may display an error message if your file is corrupt, contains missing values, or has other errors.
-    To get your file cleaned and pre-processed before using this QA app, please use our Text Preprocessing Service which can be found in the navigation menu of our website: https://nlpblogs.com/
-    
-    For any other errors or inquiries, please contact us at [email protected]
+    The app may display an error message if your file has errors or date values.
+    For any errors or inquiries, please contact us at info@nlpblogs.com
     
 ''')
     
@@ -87,27 +79,28 @@ with st.sidebar:
 if 'question_attempts' not in st.session_state:
     st.session_state['question_attempts'] = 0
 
-max_attempts = 3
+max_attempts = 5
 
 # upload file
 upload_file = st.file_uploader("Upload your file. Accepted file formats include: .csv, .xlsx", type=['csv', 'xlsx'])
-df = None
+
 
 if upload_file is not None:
     file_extension = upload_file.name.split('.')[-1].lower()
     if file_extension == 'csv':
         try:
-            df = pd.read_csv(io.StringIO(upload_file.getvalue().decode("utf-8")))
+            df = pd.read_csv(upload_file, na_filter=False)
             if df.isnull().values.any():
                 st.error("Error: The CSV file contains missing values.")
                 st.stop()
             else:
                 new_columns = [f'column_{i+1}' for i in range(len(df.columns))]
                 df.columns = new_columns
-                
                 st.dataframe(df, key="csv_dataframe")
+                
                 st.write("_number of rows_", df.shape[0])
                 st.write("_number of columns_", df.shape[1])
+                st.session_state.df = df  
         except pd.errors.ParserError:
             st.error("Error: The CSV file is not readable or is incorrectly formatted.")
             st.stop()
@@ -119,7 +112,8 @@ if upload_file is not None:
             st.stop()
     elif file_extension == 'xlsx':
         try:
-            df = pd.read_excel(io.BytesIO(upload_file.getvalue()))
+            df = pd.read_excel(upload_file, na_filter=False)
+            
             if df.isnull().values.any():
                 st.error("Error: The Excel file contains missing values.")
                 st.stop()
@@ -129,6 +123,7 @@ if upload_file is not None:
                 st.dataframe(df, key="excel_dataframe")
                 st.write("_number of rows_", df.shape[0])
                 st.write("_number of columns_", df.shape[1])
+                st.session_state.df = df  
         except ValueError:
             st.error("Error: The Excel file is not readable or is incorrectly formatted.")
             st.stop()
@@ -141,19 +136,23 @@ if upload_file is not None:
 
 # generate and validate Fernet token for the current file
 if 'fernet_token' not in st.session_state:
-    if df is not None:
+    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=10)
+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"):
-        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=10)
+        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()
@@ -182,7 +181,7 @@ st.button("Clear question", on_click=clear_question)
 #retrive answer
 if st.button("Retrieve your answer"):
     if st.session_state['question_attempts'] >= max_attempts:
-        st.error(f"You have asked {max_attempts} questions. You have reached your daily request limit. Your subscription will renew automatically. To cancel, please go to your Account.")
+        st.error(f"You have asked {max_attempts} questions. Maximum question attempts reached.")
         st.stop()
     st.session_state['question_attempts'] += 1
     if error_streamlit:
@@ -191,8 +190,8 @@ if st.button("Retrieve your answer"):
         with st.spinner('Wait for it...'):
             time.sleep(2)
             if df is not None:
-                tqa = pipeline(task="table-question-answering", model="google/tapas-base-finetuned-wikisql-supervised")
+                tqa = pipeline(task="table-question-answering", model="microsoft/tapex-large-finetuned-wtq")
                 st.write(tqa(table=df, query=question)['answer'])
 
 st.divider()
-st.write(f"Number of questions asked: {st.session_state['question_attempts']}/{max_attempts}")
+st.write(f"Number of questions asked: {st.session_state['question_attempts']}/{max_attempts}")