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-{phone_number}-{timestamp}.txt"
+    file_name = f"phone-{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(f'phone-{phone_number}') and file_name.endswith('.txt'):
+        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:
@@ -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}?phone={phone}&hash={password_hash}"
+    hash_message = f"Verify here: {base_url}?hash={password_hash}"
 
     result = tf.sendSMS(phone, hash_message)
     if result.ok:
@@ -58,44 +58,52 @@ def send_verification_sms(phone_number, password_hash):
         st.sidebar.error("Failed to send SMS ❌")
         log_event("Failed to send SMS", "❌")
 
-# 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.")
+# 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 and 'hash' in query_params:
+if 'phone' in query_params:
     phone_from_url = format_phone_number(query_params['phone'][0])
-    hash_from_url = query_params['hash'][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_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", "✅")
+    if is_auth:
+        log_event("User authenticated using phone number from URL", "✅")
     else:
-        st.error(f"Validation failed. Phone: {phone_from_url}, Hash: {hash_from_url}")
-        log_event("Validation failed using URL parameters", "❌")
+        log_event("No authentication records found for the phone number in URL", "❌")
 
 # Display the main area if authenticated
-if st.session_state.get('authenticated', False):
+if st.session_state['authenticated']:
     st.write("## Main Area")
+    
     # Display history for the specific phone number
-    st.write("## 📜 Verification History")
-    history_files = [f for f in os.listdir() if f.startswith(f'phone-{st.session_state["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:
@@ -103,7 +111,38 @@ if st.session_state.get('authenticated', False):
             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 login to view your file history.")
+    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```")