Update app.py

app.py

@@ -2,116 +2,28 @@ import streamlit as st
 import pandas as pd
 import pywhatkit
 import time
-import pyautogui
-import os
-from datetime import datetime
 from pyvirtualdisplay import Display  # Import PyVirtualDisplay
 
 # Initialize a virtual display (required for headless environments)
 display = Display(visible=0, size=(800, 600))
 display.start()  # Start the virtual display
 
-# Set page configuration
-st.set_page_config(page_title="WhatsApp Instant Messenger", page_icon="📱", layout="centered")
-
-# Custom CSS for styling
-st.markdown(
-    """
-    <style>
-        /* Background Color */
-        .stApp {
-            background-color: #802b57;
-        }
-
-        /* Title and Labels */
-        h1, h2, h3, label {
-            color: white !important;
-            font-weight: bold !important;
-        }
-
-        /* Success Messages */
-        .stSuccess {
-            color: white !important;
-            font-weight: bold !important;
-        }
-
-        /* Error Messages */
-        .stError {
-            color: red !important;
-            font-weight: bold !important;
-        }
-
-        /* Warning Messages */
-        .stWarning {
-            color: orange !important;
-            font-weight: bold !important;
-        }
-
-        /* Spinner Text */
-        .stSpinner > div > div {
-            color: white !important;
-            font-weight: bold !important;
-        }
-
-        /* Button Styling */
-        .stButton button {
-            background-color: #25D366;
-            color: white;
-            font-weight: bold;
-        }
-
-        /* File Uploader Label */
-        .stFileUploader label {
-            color: white !important;
-            font-weight: bold !important;
-        }
-                /* Title, Labels, and All Messages */
-        h1, h2, h3, label, p, .stSuccess, .stError, .stWarning, .stInfo, .stSpinner > div > div {
-            color: white !important;
-            font-weight: bold !important;
-        }
-
-        /* Button Styling */
-        .stButton button {
-            background-color: #25D366;
-            color: white;
-            font-weight: bold;
-        }
-
-        /* File Uploader Label */
-        .stFileUploader label {
-            color: white !important;
-            font-weight: bold !important;
-        }
-
-        /* Progress Bar Text */
-        .stProgress > div > div > div {
-            color: white !important;
-            font-weight: bold !important;
-        }
-    </style>
-    """,
-    unsafe_allow_html=True,
-)
-
-# Function to send WhatsApp message instantly and press enter
+# Function to send WhatsApp message instantly
 def send_whatsapp_message(phone_number, message):
     try:
         # Clean phone number to ensure proper format
         phone_number = str(phone_number).strip()
         if not phone_number.startswith('+'):
             phone_number = '+' + phone_number
+
         # Use pywhatkit to open WhatsApp Web with the message
         pywhatkit.sendwhatmsg_instantly(
             phone_no=phone_number,
             message=message,
             wait_time=15,  # Wait time for WhatsApp Web to load
-            tab_close=False  # Keep tab open
+            tab_close=True  # Close the tab after sending the message
         )
-        
-        time.sleep(2)  # Small delay
-        pyautogui.press('enter')  # Press Enter to send
-        
+        time.sleep(2)  # Small delay to ensure the message is sent
         return True, None
     except Exception as e:
         return False, str(e)
@@ -120,84 +32,68 @@ def send_whatsapp_message(phone_number, message):
 st.title("WhatsApp Instant Messenger 📱")
 st.markdown("**Send instant WhatsApp messages to multiple recipients.**")
 
-# Manual Input Option
-st.subheader("**Send Single Message**")
-phone_number = st.text_input("**Enter Phone Number (with country code, e.g., +1234567890):**", key="single_phone")
-message = st.text_area("**Enter Message:**", key="single_message")
-if st.button("Send Now", key="send_single"):
-    if phone_number and message:
-        with st.spinner("Sending message..."):
-            success, error = send_whatsapp_message(phone_number, message)
-            if success:
-                st.success(f"Message sent successfully to {phone_number}!")
-            else:
-                st.error(f"Failed to send message: {error}")
-    else:
-        st.warning("Please enter both phone number and message.")
-
 # File Upload Option
-st.subheader("**Send Multiple Messages**")
-st.markdown("**Upload an Excel file with columns:**")
 uploaded_file = st.file_uploader("Upload your Excel file", type=["xls", "xlsx"])
 if uploaded_file:
     try:
         # Read the Excel file
-        if uploaded_file.name.endswith('.xls'):
-            df = pd.read_excel(uploaded_file, engine='xlrd', dtype={"Phone Numbers": str})
-        elif uploaded_file.name.endswith('.xlsx'):
-            df = pd.read_excel(uploaded_file, engine='openpyxl', dtype={"Phone Numbers": str})
-        
+        df = pd.read_excel(uploaded_file, dtype={"Phone Numbers": str})
+
         st.success("File uploaded successfully!", icon="✅")
+
         # Validate required columns
-        required_columns = {"Phone Numbers", "Message"}
+        required_columns = {"Phone Numbers", "Message", "Date"}
         if not required_columns.issubset(df.columns):
             missing = required_columns - set(df.columns)
             st.error(f"Error: The file is missing columns: {', '.join(missing)}")
         else:
             # Send Messages
             if st.button("Send All Messages Now"):
-                # Create containers for displaying progress
-                status_container = st.empty()
-                progress_bar = st.progress(0)
-                result_container = st.empty()
-                
-                status_container.info("Preparing to send messages...")
                 successful_sends = []
                 failed_sends = []
-                total_rows = len(df)
-                
+
                 for index, row in df.iterrows():
-                    current_row = index + 1
-                    progress = current_row / total_rows
-                    status_container.info(f"Processing row {current_row}/{total_rows}")
-                    progress_bar.progress(progress)
-                    
-                    phone_raw = str(row["Phone Numbers"])
-                    message = str(row["Message"])
-                    
-                    if not message:
-                        failed_sends.append((phone_raw, "Empty message"))
+                    phone_raw = str(row["Phone Numbers"]).strip()
+                    message = str(row["Message"]).strip()
+                    date_str = str(row["Date"]).strip()
+
+                    if not phone_raw or not message or not date_str:
+                        failed_sends.append((phone_raw, "Missing data"))
                         continue
-                    
-                    success, error = send_whatsapp_message(phone_raw, message)
-                    
-                    if success:
-                        successful_sends.append((phone_raw, message))
-                    else:
-                        failed_sends.append((phone_raw, error))
-                    
-                    time.sleep(1)
-                
-                status_container.empty()
-                result_text = f"Completed: {len(successful_sends)} successful, {len(failed_sends)} failed"
-                
+
+                    try:
+                        # Parse the timestamp
+                        scheduled_time = pd.to_datetime(date_str, format="%d-%m-%Y %H:%M:%S", errors="coerce")
+                        if pd.isna(scheduled_time):
+                            raise ValueError("Invalid date format")
+
+                        current_time = datetime.now()
+                        delay = (scheduled_time - current_time).total_seconds()
+
+                        if delay > 0:
+                            st.write(f"Scheduling message to {phone_raw} at {scheduled_time.strftime('%d-%m-%Y %H:%M:%S')}...")
+                            time.sleep(delay)  # Wait until the scheduled time
+
+                        # Send the message
+                        success, error = send_whatsapp_message(phone_raw, message)
+                        if success:
+                            successful_sends.append((phone_raw, message))
+                        else:
+                            failed_sends.append((phone_raw, error))
+
+                    except Exception as e:
+                        failed_sends.append((phone_raw, f"Error: {str(e)}"))
+
+                # Display results
                 if successful_sends:
-                    result_container.success(result_text)
-                elif failed_sends:
-                    result_container.error(result_text)
-                else:
-                    result_container.warning("No messages were processed")
+                    st.success(f"Successfully sent {len(successful_sends)} messages.")
+                if failed_sends:
+                    st.error(f"Failed to send {len(failed_sends)} messages.")
+                    for phone, error in failed_sends:
+                        st.write(f"To: {phone}, Error: {error}")
+
     except Exception as e:
         st.error(f"Error processing the file: {str(e)}")
+
 # Stop the virtual display when done
 display.stop()