Update app.py

app.py

@@ -6,29 +6,30 @@ import textwrap
 from datetime import datetime, timedelta
 import streamlit as st
 
-# Set your API key for DeepSeek
+# Set your API key
 OPENROUTER_API_KEY = "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"
 
-# Load the patient data from the URL
+# Load your patient data
 dataset_url = "https://huggingface.co/spaces/chrisaldikaraharja/AutomatedFollowUpSystem/resolve/main/PatientData%20-%20Sheet1.csv"
-try:
-    data = pd.read_csv(dataset_url)
-    st.success("Patient data loaded successfully!")
-    st.write(data.head())  # Display the first 5 rows of the dataset
-except Exception as e:
-    st.error(f"Error loading patient data: {e}")
-
-# Function to generate follow-up message for the patient
-def generate_follow_up_message(patient_id):
+data = pd.read_csv(dataset_url)  # Loading the patient data from the URL
+st.write("Patient data loaded successfully.")
+st.write(data.head())  # Displaying the first few rows of the dataset
+
+# Mailjet API credentials
+MAILJET_API_KEY = 'f746efdd7af5c96a033ddb90a78a5704'
+MAILJET_SECRET_KEY = '5ee6955f792909206669fb10e9910b57'
+
+def generate_follow_up_message(patient_name):
     try:
-        # Extract patient data based on patient ID
-        patient_data = data[data['Patient ID'] == patient_id]
+        # Filter data by patient name
+        patient_data = data[data['Patient Name'].str.contains(patient_name, case=False, na=False)]
+        
         if patient_data.empty:
-            return f"Patient with ID {patient_id} not found.", None
+            return f"Patient with name {patient_name} not found.", None, None
         
         patient_data = patient_data.iloc[0]
 
-        patient_name = patient_data['Patient Name']
+        patient_id = patient_data['Patient ID']
         health_condition = patient_data['Health Condition']
         last_visit_date = patient_data['Last Visit Date']
         patient_email = patient_data['Email']
@@ -51,6 +52,8 @@ def generate_follow_up_message(patient_id):
             headers={
                 "Authorization": f"Bearer {OPENROUTER_API_KEY}",
                 "Content-Type": "application/json",
+                "HTTP-Referer": "<YOUR_SITE_URL>",  # Optional. Site URL for rankings
+                "X-Title": "<YOUR_SITE_NAME>",       # Optional. Site title for rankings
             },
             data=json.dumps({
                 "model": "deepseek/deepseek-chat:free",
@@ -67,32 +70,31 @@ def generate_follow_up_message(patient_id):
         if response.status_code == 200:
             follow_up_message = response.json()['choices'][0]['message']['content']
             formatted_message = "\n".join(textwrap.wrap(follow_up_message, width=80))
-            return formatted_message, patient_email
+            return formatted_message, patient_email, patient_id
         else:
-            return f"Failed to generate message. Error: {response.status_code}", None
+            return f"Failed to generate message. Error: {response.status_code}", None, None
 
     except Exception as e:
-        return f"An error occurred: {e}", None
+        return f"An error occurred: {e}", None, None
 
 # Streamlit UI for patient search and message sending
 st.title("Patient Follow-Up Reminder System")
 
-# Input boxes for patient name and patient ID
+# Input box for patient name
 patient_name_input = st.text_input("Enter Patient Name")
-patient_id_input = st.number_input("Enter Patient ID", min_value=1)
 
-if patient_name_input and patient_id_input:
+if patient_name_input:
     # Generate the follow-up message based on the input
-    follow_up_message, patient_email = generate_follow_up_message(patient_id_input)
+    follow_up_message, patient_email, patient_id = generate_follow_up_message(patient_name_input)
     
     if "Patient not found" not in follow_up_message:
         st.write("**Generated Message:**")
         st.write(follow_up_message)
+        st.write(f"**Patient ID:** {patient_id}")
         
         if st.button("Send Reminder"):
-            # Code to send the email to the patient's email (e.g., using SMTP, or via another API like SendGrid or Twilio)
+            # Send the email using Mailjet
             try:
-                # Replace with your email sending functionality (e.g., using SMTP or any email service)
                 send_email(patient_email, follow_up_message)
                 st.success("Message sent successfully!")
             except Exception as e:
@@ -100,9 +102,39 @@ if patient_name_input and patient_id_input:
     else:
         st.warning(follow_up_message)
 
-# Placeholder function to send the email
+# Function to send email using Mailjet
 def send_email(recipient_email, message):
-    # Use an email service like SMTP, SendGrid, etc., to send the email
-    # This is a placeholder; you can replace it with actual email sending code
-    print(f"Sending message to {recipient_email}")
-    print(message)
+    # Mailjet API endpoint
+    url = 'https://api.mailjet.com/v3.1/send'
+
+    # Email details
+    from_email = 'christ10aldika@gmail.com'  # Sender email
+    subject = 'Appointment Reminder'
+
+    # Prepare the message body
+    message_body = f"""
+    Dear {recipient_email.split('@')[0]},\n\n
+    {message}\n\n
+    If you have any questions or concerns, please reach out to us at sentramedika@gmail.com.
+    """
+
+    # Email payload
+    payload = {
+        "Messages": [
+            {
+                "From": {"Email": from_email, "Name": "Christ Aldika"},  # Sender name and email
+                "To": [{"Email": recipient_email, "Name": recipient_email}],  # Recipient email
+                "Subject": subject,
+                "TextPart": message_body,
+            }
+        ]
+    }
+    # Send the email request
+    response = requests.post(url, json=payload, auth=(MAILJET_API_KEY, MAILJET_SECRET_KEY))
+
+    if response.status_code == 200:
+        print("Email sent successfully via Mailjet!")
+        print("Response:", response.json())
+    else:
+        print(f"Failed to send email. Status Code: {response.status_code}")
+        print("Response:", response.text)