Spaces:
Sleeping
Sleeping
import os | |
import hashlib | |
import streamlit as st | |
import textflowsms as tf | |
from datetime import datetime | |
import pytz | |
# Central Time Zone Adjustment | |
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 | |
# 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}") | |
# 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: | |
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) | |
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) | |
# URL hash handling | |
query_params = st.experimental_get_query_params() | |
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 phone number from URL", "β ") | |
else: | |
log_event("No authentication records found for the phone number in URL", "β") | |
# 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") | |
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: | |
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```") |