Update app.py

app.py

@@ -5,20 +5,20 @@ import textflowsms as tf
 from datetime import datetime
 import pytz
 
-# 1. Central Time Zone Adjustment
+# Central Time Zone Adjustment
 central = pytz.timezone('US/Central')
 
-# 2. Function to format phone number
+# Function to format phone number
 def format_phone_number(phone_number):
     if len(phone_number) == 10 and not phone_number.startswith('+'):
         return '+1' + phone_number
     return phone_number
 
-# 3. Function to hash a password
+# Function to hash a password
 def hash_password(password):
     return hashlib.sha256(password.encode()).hexdigest()
 
-# 4. Function to save user data to a file
+# Function to save user data to a file
 def save_user_data(phone_number, password_hash):
     timestamp = datetime.now(central).strftime('%d%m%y-%H-%M')
     file_name = f"phone-{timestamp}.txt"
@@ -26,7 +26,7 @@ def save_user_data(phone_number, password_hash):
         file.write(f"{password_hash}\n")
     return file_name
 
-# 5. Function to check if user is authenticated
+# Function to check if user is authenticated
 def is_user_authenticated(phone_number, hash_value):
     for file_name in os.listdir():
         if file_name.startswith('phone-') and phone_number in file_name:
@@ -36,12 +36,29 @@ def is_user_authenticated(phone_number, hash_value):
                     return True
     return False
 
-# 6. Function to log events using markdown
+# Function to log events using markdown
 def log_event(message, emoji):
     timestamp = datetime.now(central).strftime('%Y-%m-%d %H:%M:%S')
     st.markdown(f"{emoji} **{timestamp}:** {message}")
 
-# 7. Initialize session state
+# Function to send verification SMS
+def send_verification_sms(phone_number, password_hash):
+    api_key = os.getenv('API_KEY')
+    tf.useKey(api_key)
+
+    base_url = "https://huggingface.co/spaces/awacke1/RT-SMS-Phone-Verify"
+    phone = format_phone_number(phone_number)
+    hash_message = f"Verify here: {base_url}?hash={password_hash}"
+
+    result = tf.sendSMS(phone, hash_message)
+    if result.ok:
+        st.sidebar.success("Verification link sent via SMS 📨")
+        log_event("Verification SMS sent", "📨")
+    else:
+        st.sidebar.error("Failed to send SMS ❌")
+        log_event("Failed to send SMS", "❌")
+
+# Initialize session state
 if 'phone_number' not in st.session_state:
     st.session_state['phone_number'] = '+19522583980'
 if 'password' not in st.session_state:
@@ -51,50 +68,35 @@ if 'hash' not in st.session_state:
 if 'authenticated' not in st.session_state:
     st.session_state['authenticated'] = False
 
-# 8. Sidebar inputs
+# Sidebar inputs
 user_phone_input = st.sidebar.text_input("📱 Mobile Phone", value=st.session_state.get('phone_number', ''))
 password_input = st.sidebar.text_input("🔑 Set Password", type='password')
 
-# 9. Save button
+# Save button
 if st.sidebar.button('💾 Save Settings'):
     st.session_state['phone_number'] = format_phone_number(user_phone_input)
-    hashed_password = hash_password(password_input)
-    st.session_state['password'] = hashed_password
-    file_name = save_user_data(st.session_state['phone_number'], hashed_password)
-    st.sidebar.success(f"Settings saved successfully! Data saved in {file_name}")
-    log_event("Settings saved successfully", "💾")
-
-# 10. Function to send verification SMS
-def send_verification_sms():
-    api_key = os.getenv('API_KEY')
-    tf.useKey(api_key)
-
-    base_url = "https://huggingface.co/spaces/awacke1/RT-SMS-Phone-Verify"
-    phone = format_phone_number(user_phone_input)
-    hash_message = f"Verify here: {base_url}?hash={st.session_state['password']}"
+    if password_input:
+        hashed_password = hash_password(password_input)
+        st.session_state['password'] = hashed_password
+        file_name = save_user_data(st.session_state['phone_number'], hashed_password)
+        st.sidebar.success(f"Settings saved successfully! Data saved in {file_name}")
+        log_event("Settings saved successfully", "💾")
+        send_verification_sms(st.session_state['phone_number'], hashed_password)
 
-    result = tf.sendSMS(phone, hash_message)
-    if result.ok:
-        st.sidebar.success("Verification link sent via SMS 📨")
-        log_event("Verification SMS sent", "📨")
-    else:
-        st.sidebar.error("Failed to send SMS ❌")
-        log_event("Failed to send SMS", "❌")
-
-# 11. URL hash handling
+# URL hash handling
 query_params = st.experimental_get_query_params()
-if 'hash' in query_params:
-    received_hash = query_params['hash'][0]
-    st.session_state['hash'] = received_hash
-    is_auth = is_user_authenticated(st.session_state['phone_number'], received_hash)
+if 'phone' in query_params:
+    phone_from_url = format_phone_number(query_params['phone'][0])
+    st.session_state['phone_number'] = phone_from_url
+    is_auth = any(phone_from_url in file for file in os.listdir() if file.startswith('phone-'))
     st.session_state['authenticated'] = is_auth
 
     if is_auth:
-        log_event("User authenticated using text", "✅")
+        log_event("User authenticated using phone number from URL", "✅")
     else:
-        log_event("Authentication failed. Hash mismatch or user not found", "❌")
+        log_event("No authentication records found for the phone number in URL", "❌")
 
-# 12. Display the main area if authenticated
+# Display the main area if authenticated
 if st.session_state['authenticated']:
     st.write("## Main Area")
     
@@ -107,3 +109,40 @@ if st.session_state['authenticated']:
     for file_name in history_files:
         with open(file_name, 'r') as file:
             history_markdown += f"| {file_name} | {file.read().strip()} |\n"
+    st.markdown(history_markdown)
+
+    st.markdown(history_markdown)
+else:
+    # If not authenticated, display a message
+    st.write("## ❗ Authentication Required")
+    st.write("Please authenticate using the link sent to your mobile phone.")
+
+
+# Import additional libraries
+import base64
+
+# Function to create a base64-encoded download link
+def create_download_link(file_name):
+    with open(file_name, 'rb') as f:
+        bytes = f.read()
+        b64 = base64.b64encode(bytes).decode()
+        href = f'<a href="data:file/txt;base64,{b64}" download="{file_name}">Download {file_name}</a>'
+        return href
+
+# Display a list of history files with download links and contents
+st.write("## 🗂️ File History")
+
+# Iterate over history files and display them
+history_files = [f for f in os.listdir() if f.startswith('phone-')]
+for file_name in history_files:
+    # Create download link
+    download_link = create_download_link(file_name)
+
+    # Read and display file contents
+    with open(file_name, 'r') as file:
+        file_contents = file.read().strip()
+
+    # Display file information and contents in markdown
+    st.markdown(f"### {file_name}")
+    st.markdown(download_link, unsafe_allow_html=True)
+    st.markdown("```python\n" + file_contents + "\n```")