Update app.py

app.py

@@ -21,7 +21,7 @@ def hash_password(password):
 # 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"
+    file_name = f"phone-{phone_number}-{timestamp}.txt"
     with open(file_name, 'w') as file:
         file.write(f"{password_hash}\n")
     return file_name
@@ -29,7 +29,7 @@ def save_user_data(phone_number, password_hash):
 # 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:
+        if file_name.startswith(f'phone-{phone_number}') and file_name.endswith('.txt'):
             with open(file_name, 'r') as file:
                 stored_hash = file.readline().strip()
                 if stored_hash == hash_value:
@@ -48,7 +48,7 @@ def send_verification_sms(phone_number, password_hash):
 
     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}"
+    hash_message = f"Verify here: {base_url}?phone={phone}&hash={password_hash}"
 
     result = tf.sendSMS(phone, hash_message)
     if result.ok:
@@ -58,52 +58,44 @@ def send_verification_sms(phone_number, password_hash):
         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)
+# Sidebar inputs for login
+phone_input = st.sidebar.text_input("Phone Number")
+password_input = st.sidebar.text_input("Password", type="password")
+
+# Button to handle login
+if st.sidebar.button("Login"):
+    phone_formatted = format_phone_number(phone_input)
+    password_hashed = hash_password(password_input)
+
+    # Check if user is authenticated
+    if is_user_authenticated(phone_formatted, password_hashed):
+        st.session_state['authenticated'] = True
+        st.session_state['phone_number'] = phone_formatted
+        st.success("Logged in successfully!")
+    else:
+        st.error("Invalid phone number or password.")
 
 # URL hash handling
 query_params = st.experimental_get_query_params()
-if 'phone' in query_params:
+if 'phone' in query_params and 'hash' 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
+    hash_from_url = query_params['hash'][0]
 
-    if is_auth:
-        log_event("User authenticated using phone number from URL", "✅")
+    if is_user_authenticated(phone_from_url, hash_from_url):
+        st.session_state['authenticated'] = True
+        st.session_state['phone_number'] = phone_from_url
+        log_event("User authenticated using URL parameters", "✅")
     else:
-        log_event("No authentication records found for the phone number in URL", "❌")
+        st.error(f"Validation failed. Phone: {phone_from_url}, Hash: {hash_from_url}")
+        log_event("Validation failed using URL parameters", "❌")
 
 # Display the main area if authenticated
-if st.session_state['authenticated']:
+if st.session_state.get('authenticated', False):
     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]
-    
+    st.write("## 📜 Verification History")
+    history_files = [f for f in os.listdir() if f.startswith(f'phone-{st.session_state["phone_number"]}')]
+
     # Create a markdown table for history
     history_markdown = "| Filename | Hash Value |\n| --- | --- |\n"
     for file_name in history_files:
@@ -111,38 +103,7 @@ if st.session_state['authenticated']:
             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```")
+    st.write("Please login to view your file history.")