Spaces:
Sleeping
Sleeping
import os | |
import hashlib | |
import streamlit as st | |
import textflowsms as tf | |
from datetime import datetime | |
# Central Time Zone Adjustment | |
import pytz | |
central = pytz.timezone('US/Central') | |
# 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 | |
def hash_password(password): | |
return hashlib.sha256(password.encode()).hexdigest() | |
# 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" | |
with open(file_name, 'w') as file: | |
file.write(f"{password_hash}\n") | |
return file_name | |
# 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: | |
with open(file_name, 'r') as file: | |
stored_hash = file.readline().strip() | |
if stored_hash == hash_value: | |
return True | |
return False | |
# Initialize session state | |
if 'phone_number' not in st.session_state: | |
st.session_state['phone_number'] = '+19522583980' # Default phone number | |
if 'password' not in st.session_state: | |
st.session_state['password'] = '' | |
if 'hash' not in st.session_state: | |
st.session_state['hash'] = '' | |
if 'authenticated' not in st.session_state: | |
st.session_state['authenticated'] = False | |
# 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 | |
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') | |
# 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) | |
# 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") | |
# Display the main area if authenticated | |
if st.session_state['authenticated']: | |
st.write("## Main Area") | |
# [Display main area content] | |
else: | |
if st.sidebar.button('π¨ Send Verify Request'): | |
send_verification_sms() | |
# Display history for the specific phone number | |
st.write("## π Verification History") | |
history_files = [f for f in os.listdir() if f.startswith('phone-') and st.session_state['phone_number'] in f] | |
for file_name in history_files: | |
with open(file_name, 'r') as file: | |
st.write(f"{file_name}: {file.read()}") | |