Update app.py

app.py

@@ -1,134 +1,128 @@
 import os
 import pandas as pd
 import requests
-import json
 import textwrap
 from datetime import datetime, timedelta
 import streamlit as st
 
-# Set your API key
+# -----------------------------
+# API Keys
+# -----------------------------
 OPENROUTER_API_KEY = "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"
+MAILJET_API_KEY = "f746efdd7af5c96a033ddb90a78a5704"
+MAILJET_SECRET_KEY = "5ee6955f792909206669fb10e9910b57"
 
-# Load your patient data
+# -----------------------------
+# Load Patient Data
+# -----------------------------
 dataset_url = "https://huggingface.co/spaces/chrisaldikaraharja/AutomatedFollowUpSystem/resolve/main/PatientData%20-%20Sheet1.csv"
-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
+data = pd.read_csv(dataset_url)
+st.write("✅ Patient data loaded successfully.")
+st.write(data.head())
 
-# Mailjet API credentials
-MAILJET_API_KEY = 'f746efdd7af5c96a033ddb90a78a5704'
-MAILJET_SECRET_KEY = '5ee6955f792909206669fb10e9910b57'
-
-# Function to send email using Mailjet
+# -----------------------------
+# Mailjet Email Sender
+# -----------------------------
 def send_email(recipient_email, follow_up_message):
-    # Mailjet API endpoint
-    url = 'https://api.mailjet.com/v3.1/send'
-
-    # Sender email and details
-    from_email = 'christ10aldika@gmail.com'  # Sender email
-    subject = 'Appointment Reminder'
+    """Send an email via Mailjet API."""
+    url = "https://api.mailjet.com/v3.1/send"
+    from_email = "christ10aldika@gmail.com"
+    subject = "Appointment Reminder"
 
-    # Email payload
     payload = {
         "Messages": [
             {
-                "From": {"Email": from_email, "Name": "Christ Aldika"},  # Sender name and email
-                "To": [{"Email": recipient_email, "Name": recipient_email}],  # Recipient email
+                "From": {"Email": from_email, "Name": "Christ Aldika"},
+                "To": [{"Email": recipient_email, "Name": recipient_email}],
                 "Subject": subject,
-                "TextPart": follow_up_message,  # Use the generated follow-up message
+                "TextPart": follow_up_message,
             }
         ]
     }
 
-    # 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())
+        return True, "Email sent successfully!"
     else:
-        print(f"Failed to send email. Status Code: {response.status_code}")
-        print("Response:", response.text)
+        return False, f"Failed to send email. Status: {response.status_code}, Response: {response.text}"
 
+# -----------------------------
+# Follow-Up Message Generator
+# -----------------------------
 def generate_follow_up_message(patient_name):
+    """Generate a follow-up message for a patient using OpenRouter API."""
     try:
-        # 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 name {patient_name} not found.", None, None
-        
-        patient_data = patient_data.iloc[0]
+        patient_data = data[data["Patient Name"].str.contains(patient_name, case=False, na=False)]
 
-        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']
+        if patient_data.empty:
+            return f"Patient with name '{patient_name}' not found.", None, None
 
-        # Convert last visit date to a datetime object
-        last_visit_date_obj = datetime.strptime(last_visit_date, '%Y-%m-%d')
+        patient_data = patient_data.iloc[0]
+        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"]
 
-        # Add 7 days to the last visit date to calculate the next appointment date
+        # Date calculations
+        last_visit_date_obj = datetime.strptime(last_visit_date, "%Y-%m-%d")
         next_appointment_date = last_visit_date_obj + timedelta(days=7)
+        next_appointment_date_str = next_appointment_date.strftime("%Y-%m-%d")
+
+        # Prompt for AI
+        prompt = (
+            f"Write a polite, detailed follow-up message for {patient_name}, who has {health_condition} "
+            f"and last visited the doctor on {last_visit_date}. Remind them of their next appointment "
+            f"on {next_appointment_date_str} and encourage them to contact the clinic at sentramedika@gmail.com "
+            f"if they have concerns. Keep the message polite and informative, around 2 sentences."
+        )
 
-        # Format the next appointment date
-        next_appointment_date_str = next_appointment_date.strftime('%Y-%m-%d')
-
-        # Replace the placeholders in the prompt
-        prompt = f"Write a polite, detailed follow-up message for {patient_name}, who has {health_condition} and last visited the doctor on {last_visit_date}. Remind them of their next appointment on {next_appointment_date_str} and encourage them to contact the clinic at sentramedika@gmail.com if they have concerns. Keep the message polite and informative, around 2 sentences."
-
-        # Make a POST request to the DeepSeek API
+        # API request to OpenRouter
         response = requests.post(
             url="https://openrouter.ai/api/v1/chat/completions",
             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
+                "HTTP-Referer": "https://your-site.com",  
+                "X-Title": "Patient Follow-Up Reminder System",
             },
-            data=json.dumps({
-                "model": "deepseek/deepseek-chat:free",
-                "messages": [
-                    {
-                        "role": "user",
-                        "content": prompt
-                    }
-                ],
-            })
+            json={
+                "model": "deepseek/deepseek-r1:free",  # ✅ Correct model
+                "messages": [{"role": "user", "content": prompt}]
+            }
         )
 
-        # Check if the request was successful
-        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, patient_id
+        if response.ok:
+            try:
+                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, patient_id
+            except (KeyError, IndexError):
+                return "Unexpected API response format.", None, None
         else:
-            return f"Failed to generate message. Error: {response.status_code}", None, None
+            return f"Failed to generate message. Error: {response.status_code} - {response.text}", None, None
 
     except Exception as e:
         return f"An error occurred: {e}", None, None
 
-# Streamlit UI for patient search and message sending
-st.title("Patient Follow-Up Reminder System")
-
-# Input box for patient name
+# -----------------------------
+# Streamlit UI
+# -----------------------------
+st.title("💌 Patient Follow-Up Reminder System")
 patient_name_input = st.text_input("Enter Patient Name")
 
 if patient_name_input:
-    # Generate the follow-up message based on the input
     follow_up_message, patient_email, patient_id = generate_follow_up_message(patient_name_input)
-    
-    if "Patient not found" not in follow_up_message:
+
+    if patient_email:
         st.write("**Generated Message:**")
         st.write(follow_up_message)
         st.write(f"**Patient ID:** {patient_id}")
-        
+
         if st.button("Send Reminder"):
-            # Send the email using Mailjet
-            try:
-                send_email(patient_email, follow_up_message)
-                st.success("Message sent successfully!")
-            except Exception as e:
-                st.error(f"Failed to send email: {e}")
+            success, message = send_email(patient_email, follow_up_message)
+            if success:
+                st.success(message)
+            else:
+                st.error(message)
     else:
-        st.warning(follow_up_message)
+        st.warning(follow_up_message)