Update app.py

app.py

@@ -3,22 +3,22 @@ import hashlib
 import streamlit as st
 import textflowsms as tf
 from datetime import datetime
-
-# Central Time Zone Adjustment
 import pytz
+
+# 1. Central Time Zone Adjustment
 central = pytz.timezone('US/Central')
 
-# Function to format phone number
+# 2. 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
 
-# Function to hash a password
+# 3. Function to hash a password
 def hash_password(password):
     return hashlib.sha256(password.encode()).hexdigest()
 
-# Function to save user data to a file
+# 4. 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
 
-# Function to check if user is authenticated
+# 5. 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,9 +36,14 @@ def is_user_authenticated(phone_number, hash_value):
                     return True
     return False
 
-# Initialize session state
+# 6. 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
 if 'phone_number' not in st.session_state:
-    st.session_state['phone_number'] = '+19522583980'  # Default phone number
+    st.session_state['phone_number'] = '+19522583980'
 if 'password' not in st.session_state:
     st.session_state['password'] = ''
 if 'hash' not in st.session_state:
@@ -46,66 +51,59 @@ if 'hash' not in st.session_state:
 if 'authenticated' not in st.session_state:
     st.session_state['authenticated'] = False
 
-# Sidebar inputs
+# 8. 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')
 
-# Save button
+# 9. 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}")
-    # Record save event with timestamp
-    st.session_state['save_time'] = datetime.now(central).strftime('%Y-%m-%d %H:%M:%S')
-
+    log_event("Settings saved successfully", "💾")
 
-# Function to send verification SMS
+# 10. Function to send verification SMS
 def send_verification_sms():
-    
-    # Load the API key from an environment variable
     api_key = os.getenv('API_KEY')
     tf.useKey(api_key)
-    #tf.sendSMS("+19522583980", "Test 2")
 
     base_url = "https://huggingface.co/spaces/awacke1/RT-SMS-Phone-Verify"
-    # Correct the sendSMS function call
-    #phone=st.session_state['phone_number']
-    phone=user_phone_input
-    st.write('Sending SMS to phone number ' + phone)
-    
-    hashmessage=f"Verify here: {base_url}?hash={st.session_state['password']}"
-    st.write('Hash message: ' + hashmessage)
-    
-    result = tf.sendSMS(phone, hashmessage)
-    
-    if(result.ok):
-        print(result.data)
-        st.sidebar.success("Verification link sent via SMS")
-    else:
-      print(result.message)
+    phone = format_phone_number(user_phone_input)
+    hash_message = f"Verify here: {base_url}?hash={st.session_state['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", "❌")
 
-# URL hash handling
+# 11. URL hash handling
 query_params = st.experimental_get_query_params()
 if 'hash' in query_params:
-    st.session_state['hash'] = query_params['hash'][0]
-    st.session_state['authenticated'] = is_user_authenticated(st.session_state['phone_number'], st.session_state['hash'])
-    if st.session_state['authenticated']:
-        st.write("✅ User has authenticated using text")
+    received_hash = query_params['hash'][0]
+    st.session_state['hash'] = received_hash
+    is_auth = is_user_authenticated(st.session_state['phone_number'], received_hash)
+    st.session_state['authenticated'] = is_auth
+
+    if is_auth:
+        log_event("User authenticated using text", "✅")
+    else:
+        log_event("Authentication failed. Hash mismatch or user not found", "❌")
 
-# Display the main area if authenticated
+# 12. Display the main area if authenticated
 if st.session_state['authenticated']:
     st.write("## Main Area")
-        
+    
     # Display history for the specific phone number
-    st.write("## 📜 Authenticated!  Below is Your Verification History")
+    st.write("## 📜 Authenticated! Below is Your Verification History")
     history_files = [f for f in os.listdir() if f.startswith('phone-') and st.session_state['phone_number'] in f]
+    
+    # Create a markdown table for history
+    history_markdown = "| Filename | Hash Value |\n| --- | --- |\n"
     for file_name in history_files:
         with open(file_name, 'r') as file:
-            st.write(f"{file_name}: {file.read()}")
-
-else:
-    if st.sidebar.button('📨 Send Verify Request'):
-        send_verification_sms()
+            history_markdown += f"| {file_name} | {file.read().strip()} |\n"