Upload 3 files

app.py

@@ -0,0 +1,395 @@
+import os
+import json
+import logging
+import tempfile
+import re
+from dotenv import load_dotenv
+from flask import Flask, request, jsonify, render_template, send_file
+from groq import Groq
+import google.generativeai as genai
+import pandas as pd
+from datetime import datetime, date, time
+from apscheduler.schedulers.background import BackgroundScheduler
+from twilio.rest import Client
+import io
+
+# Load environment variables
+load_dotenv()
+
+app = Flask(__name__)
+
+# Configure API keys
+GROQ_API_KEY = os.getenv('GROQ_API_KEY', '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')
+GEMINI_API_KEY = os.getenv('GEMINI_API_KEY', 'AIzaSyCizcswP6vlKDMdB3HRAtVi2JbifOpbPvA')
+
+# Initialize clients
+groq_client = Groq(api_key=GROQ_API_KEY)
+genai.configure(api_key=GEMINI_API_KEY)
+MODEL_NAME = "gemini-1.5-flash"
+
+# Initialize scheduler
+scheduler = BackgroundScheduler()
+scheduler.start()
+
+# Logger setup
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Add back the Twilio configuration
+ACCOUNT_SID = 'AC490e071f8d01bf0df2f03d086c788d87'
+AUTH_TOKEN = '224b23b950ad5a4052aba15893fdf083'
+TWILIO_FROM = 'whatsapp:+14155238886'
+PATIENT_PHONE = 'whatsapp:+917559355282'
+
+# Initialize Twilio client
+twilio_client = Client(ACCOUNT_SID, AUTH_TOKEN)
+
+
+def send_whatsapp_message(message):
+    """Send WhatsApp notification"""
+    try:
+        msg = twilio_client.messages.create(
+            from_=TWILIO_FROM,
+            body=message,
+            to=PATIENT_PHONE
+        )
+        logger.info(f"WhatsApp notification sent: {msg.sid}")
+        return True
+    except Exception as e:
+        logger.error(f"Failed to send WhatsApp notification: {str(e)}")
+        return False
+
+
+
+# define defaults for descriptive slots
+DESCRIPTIVE_SLOTS = {
+    'Morning':   time(hour= 8, minute=0),
+    'Afternoon': time(hour=13, minute=0),
+    'Evening':   time(hour=18, minute=0),
+    'Night':     time(hour=21, minute=0),
+}
+
+def parse_time_slot(time_str):
+    """
+    Try to parse “hh:mm AM/PM”, else map descriptive slots,
+    else return None.
+    """
+    try:
+        return datetime.strptime(time_str, '%I:%M %p')
+    except ValueError:
+        # fallback to descriptor map
+        slot = DESCRIPTIVE_SLOTS.get(time_str.title())
+        if slot:
+            # combine with today’s date just for hour/minute extraction
+            return datetime.combine(date.today(), slot)
+        else:
+            logger.warning(f"Unrecognized time format: '{time_str}'")
+            return None
+
+def schedule_notifications(schedule_data):
+    """Schedule notifications for medications, meals, and activities."""
+    try:
+        scheduler.remove_all_jobs()
+
+        for section, prefix, key_map in [
+            ('medication_schedule', 'med',   {'time': 'Scheduled Time', 'fields': ['Meal', 'Medication Name', 'Dosage']}),
+            ('meal_schedule',       'meal',  {'time': 'Time',           'fields': ['Meal', 'Details']}),
+            ('activity_schedule',   'activity', {'time': 'Time',       'fields': ['Activity', 'Duration', 'Notes']}),
+        ]:
+            for item in schedule_data.get(section, []):
+                time_str = item[key_map['time']]
+                time_obj = parse_time_slot(time_str)
+                if not time_obj:
+                    # skip invalid entries
+                    continue
+
+                # build the message dynamically
+                if section == 'medication_schedule':
+                    message = (
+                        f"🔔 Medication Reminder\n"
+                        f"Time: {time_str}\n"
+                        f"Meal: {item['Meal']}\n"
+                        f"Medication: {item['Medication Name']}\n"
+                        f"Dosage: {item.get('Dosage', 'As prescribed')}"
+                    )
+                elif section == 'meal_schedule':
+                    message = (
+                        f"🍽 Meal Reminder\n"
+                        f"Time: {time_str}\n"
+                        f"Meal: {item['Meal']}\n"
+                        f"Details: {item.get('Details', '')}"
+                    )
+                else:  # activity_schedule
+                    message = (
+                        f"🏃 Activity Reminder\n"
+                        f"Time: {time_str}\n"
+                        f"Activity: {item['Activity']}\n"
+                        f"Duration: {item.get('Duration', '')}\n"
+                        f"Notes: {item.get('Notes', '')}"
+                    )
+
+                job_id = f"{prefix}_{item.get(key_map['fields'][0], '').replace(' ', '_')}_{time_str}"
+                scheduler.add_job(
+                    send_whatsapp_message,
+                    trigger='cron',
+                    args=[message],
+                    hour=time_obj.hour,
+                    minute=time_obj.minute,
+                    id=job_id,
+                    replace_existing=True
+                )
+
+        return True
+
+    except Exception as e:
+        logger.error(f"Failed to schedule notifications: {e}")
+        return False
+
+
+
+# ✅ Function to Extract JSON from Response
+def extract_json(text):
+    json_match = re.search(r'\{.*\}', text, re.DOTALL)
+    if json_match:
+        return json_match.group(0)
+    return None
+
+
+# ✅ Function to Transcribe Audio Using Groq
+def transcribe_audio(audio_file):
+    try:
+        temp_dir = tempfile.mkdtemp()
+        temp_path = os.path.join(temp_dir, "temp_audio.mp3")
+        audio_file.save(temp_path)
+
+        # Transcribe audio
+        with open(temp_path, "rb") as file:
+            transcription = groq_client.audio.transcriptions.create(
+                file=(temp_path, file.read()),
+                model="whisper-large-v3-turbo",
+                response_format="json",
+                language="en",
+                temperature=0.0
+            )
+
+        # Cleanup temporary files
+        os.remove(temp_path)
+        os.rmdir(temp_dir)
+
+        logger.info(f"Transcription successful: {transcription.text[:100]}...")
+        return transcription.text
+
+    except Exception as e:
+        logger.error(f"Transcription error: {str(e)}")
+        return None
+
+
+# ✅ Function to Generate Care Plan Using Gemini
+def generate_care_plan(conversation_text):
+    try:
+        model = genai.GenerativeModel(MODEL_NAME)
+        prompt = f"""
+            Based on the following conversation transcript between a doctor and a patient, generate a structured care plan in *valid JSON format*.
+
+            *JSON Format:*
+            {{
+                "medication_schedule": [{{"Medication Name": ""}}],
+                "meal_schedule": [{{"Time": "", "Meal": "", "Details": ""}}],
+                "activity_schedule": [{{"Time": "", "Activity": "", "Duration": "", "Notes": ""}}]
+            }}
+
+            *Fixed Meal Times:*
+            - Breakfast: 7:00 AM
+            - Snack: 10:00 AM
+            - Lunch: 1:00 PM
+            - Snack: 5:00 PM
+            - Dinner: 8:00 PM
+
+            Ensure medications align with meals. Adjust exercise based on health conditions.
+
+            *Conversation Transcript:*
+            {conversation_text}
+
+            *Output Strictly JSON. Do NOT add explanations or extra text.*
+        """
+
+        response = model.generate_content(prompt)
+        raw_response = response.text
+        logger.info(f"Raw Gemini Response: {raw_response[:100]}...")
+
+        json_text = extract_json(raw_response)
+        if json_text:
+            return json.loads(json_text)
+
+        logger.error("No valid JSON found in response.")
+        return create_default_care_plan()
+
+    except Exception as e:
+        logger.error(f"Care plan generation error: {str(e)}")
+        return create_default_care_plan()
+
+
+# ✅ Function to Create a Default Care Plan (Fallback)
+def create_default_care_plan():
+    return {
+        "medication_schedule": [],
+        "meal_schedule": [
+            {"Time": "7:00 AM", "Meal": "Breakfast", "Details": ""},
+            {"Time": "1:00 PM", "Meal": "Lunch", "Details": ""},
+            {"Time": "8:00 PM", "Meal": "Dinner", "Details": ""}
+        ],
+        "activity_schedule": []
+    }
+
+
+# ✅ Route: Homepage
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+
+# ✅ Route: Generate Care Plan
+@app.route("/generate_care_plan", methods=["POST"])
+def generate_care_plan_route():
+    try:
+        if 'audio' not in request.files:
+            return jsonify({"error": "No audio file provided"}), 400
+
+        audio_file = request.files['audio']
+        if not audio_file:
+            return jsonify({"error": "Empty audio file"}), 400
+
+        transcript = transcribe_audio(audio_file)
+        if not transcript:
+            return jsonify({"error": "Transcription failed"}), 500
+
+        care_plan = generate_care_plan(transcript)
+        return jsonify(care_plan)
+
+    except Exception as e:
+        logger.error(f"Error in generate_care_plan_route: {str(e)}")
+        return jsonify({"error": str(e)}), 500
+
+
+# ✅ Route: Generate Medication Schedule
+@app.route('/generate_schedule', methods=['POST'])
+def generate_schedule():
+    try:
+        data = request.get_json()
+        medicines = data.get("medicines", [])
+        schedule_list = []
+
+        meal_times = {
+            3: [("Breakfast", "7:00 AM"), ("Lunch", "1:00 PM"), ("Dinner", "8:00 PM")],
+            2: [("Breakfast", "7:00 AM"), ("Dinner", "8:00 PM")],
+            1: [("Dinner", "8:00 PM")]
+        }
+
+        for med in medicines:
+            medication_name = med.get("medication_name", "")
+            frequency = int(med.get("frequency", 1))
+            dosage = med.get("dosage", "")
+
+            for meal, time in meal_times.get(frequency, []):
+                schedule_list.append({
+                    "Medication Name": medication_name,
+                    "Dosage": dosage,
+                    "Meal": meal,
+                    "Scheduled Time": time
+                })
+
+        # Schedule notifications
+        schedule_data = {
+            'medication_schedule': schedule_list,
+            'meal_schedule': data.get('meal_schedule', []),
+            'activity_schedule': data.get('activity_schedule', [])
+        }
+        notifications_scheduled = schedule_notifications(schedule_data)
+
+        return jsonify({
+            "schedule_list": schedule_list,
+            "notifications_scheduled": notifications_scheduled
+        })
+
+    except Exception as e:
+        logger.error(f"Schedule generation error: {str(e)}")
+        return jsonify({"error": str(e)}), 500
+
+
+@app.route('/download_schedule', methods=['POST'])
+def download_schedule():
+    try:
+        data = request.json
+
+        # Create Excel writer object
+        output = io.BytesIO()
+        with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
+            workbook = writer.book
+
+            # Convert medication schedule to DataFrame and write to Excel
+            if data.get('medication_schedule'):
+                med_df = pd.DataFrame(data['medication_schedule'])
+                med_df = med_df.sort_values('Time') if 'Time' in med_df.columns else med_df
+                med_df.to_excel(writer, sheet_name='Medication Schedule', index=False)
+
+                worksheet = writer.sheets['Medication Schedule']
+                header_format = workbook.add_format({
+                    'bold': True,
+                    'bg_color': '#3498db',
+                    'font_color': 'white'
+                })
+
+                for col_num, value in enumerate(med_df.columns.values):
+                    worksheet.write(0, col_num, value, header_format)
+                    worksheet.set_column(col_num, col_num, 15)  # Set column width
+
+            # Write meal schedule
+            if data.get('meal_schedule'):
+                meal_df = pd.DataFrame(data['meal_schedule'])
+                meal_df = meal_df.sort_values('Time')
+                meal_df.to_excel(writer, sheet_name='Meal Schedule', index=False)
+
+                worksheet = writer.sheets['Meal Schedule']
+                header_format = workbook.add_format({
+                    'bold': True,
+                    'bg_color': '#2ecc71',
+                    'font_color': 'white'
+                })
+
+                for col_num, value in enumerate(meal_df.columns.values):
+                    worksheet.write(0, col_num, value, header_format)
+                    worksheet.set_column(col_num, col_num, 15)
+
+            # Write activity schedule
+            if data.get('activity_schedule'):
+                activity_df = pd.DataFrame(data['activity_schedule'])
+                activity_df = activity_df.sort_values('Time')
+                activity_df.to_excel(writer, sheet_name='Activity Schedule', index=False)
+
+                worksheet = writer.sheets['Activity Schedule']
+                header_format = workbook.add_format({
+                    'bold': True,
+                    'bg_color': '#e74c3c',
+                    'font_color': 'white'
+                })
+
+                for col_num, value in enumerate(activity_df.columns.values):
+                    worksheet.write(0, col_num, value, header_format)
+                    worksheet.set_column(col_num, col_num, 15)
+
+        output.seek(0)
+        return send_file(
+            output,
+            mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+            as_attachment=True,
+            download_name='daily_schedule.xlsx'
+        )
+
+    except Exception as e:
+        logger.error(f"Error generating Excel file: {str(e)}")
+        return jsonify({"error": str(e)}), 500
+
+
+# ✅ Run Flask App
+if __name__ == '__main__':
+    app.run(debug=True, threaded=True)

requirements.txt

@@ -0,0 +1,7 @@
+Flask
+python-dotenv
+groq
+google-generativeai
+pandas
+APScheduler
+twilio

templates/index.html

@@ -0,0 +1,1093 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <title>Medical Care Plan Generator</title>
+  <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.3/dist/css/bootstrap.min.css" rel="stylesheet" integrity="sha384-QWTKZyjpPEjISv5WaRU9OFeRpok6YctnYmDr5pNlyT2bRjXh0JMhjY6hW+ALEwIH" crossorigin="anonymous">
+  <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.4/css/all.min.css">
+  <style>
+    body {
+      font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+      max-width: 1200px;
+      margin: 0 auto;
+      padding: 20px;
+      background-color: #f5f5f5;
+    }
+    .container {
+      background-color: white;
+      padding: 30px;
+      border-radius: 12px;
+      box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
+      margin-bottom: 30px;
+    }
+    h1, h2, h3 {
+      color: #2c3e50;
+    }
+    h1 {
+      text-align: center;
+      margin-bottom: 30px;
+    }
+    .section {
+      margin-bottom: 40px;
+    }
+    .upload-section {
+      text-align: center;
+      margin: 30px 0;
+      padding: 20px;
+      background-color: #f8f9fa;
+      border-radius: 8px;
+    }
+    #uploadForm {
+      margin: 20px 0;
+    }
+    .results-container {
+      margin-top: 30px;
+    }
+    table {
+      width: 100%;
+      border-collapse: collapse;
+      margin-bottom: 20px;
+    }
+    th, td {
+      padding: 12px;
+      text-align: left;
+      border: 1px solid #ddd;
+    }
+    th {
+      background-color: #3498db;
+      color: white;
+    }
+    button {
+      background-color: #3498db;
+      color: white;
+      padding: 12px 24px;
+      border: none;
+      border-radius: 6px;
+      cursor: pointer;
+      font-size: 16px;
+      transition: background-color 0.3s;
+    }
+    button:hover {
+      background-color: #2980b9;
+    }
+    .loading {
+      text-align: center;
+      margin: 20px 0;
+      display: none;
+    }
+    .error {
+      color: #e74c3c;
+      padding: 10px;
+      background-color: #fde8e8;
+      border-radius: 4px;
+      margin: 10px 0;
+    }
+    /* Print styles */
+    @media print {
+      body {
+        padding: 0;
+        margin: 0;
+      }
+      .container {
+        box-shadow: none;
+        padding: 20px;
+      }
+      .upload-section,
+      #uploadForm,
+      #loading,
+      #downloadPDF,
+      #notificationControls,
+      .btn {
+        display: none !important;
+      }
+      table {
+        width: 100%;
+        border-collapse: collapse;
+        margin-bottom: 20px;
+        page-break-inside: avoid;
+      }
+      th, td {
+        border: 1px solid #000;
+        padding: 8px;
+        text-align: left;
+      }
+      th {
+        background-color: #f0f0f0 !important;
+        -webkit-print-color-adjust: exact;
+      }
+      h3 {
+        margin-top: 20px;
+        margin-bottom: 10px;
+      }
+    }
+    .loading-overlay {
+        display: none;
+        position: fixed;
+        top: 0;
+        left: 0;
+        width: 100%;
+        height: 100%;
+        background: rgba(0, 0, 0, 0.5);
+        z-index: 1000;
+    }
+
+    .loading-content {
+        position: absolute;
+        top: 50%;
+        left: 50%;
+        transform: translate(-50%, -50%);
+        background: white;
+        padding: 30px;
+        border-radius: 10px;
+        text-align: center;
+    }
+
+    .spinner {
+        width: 50px;
+        height: 50px;
+        border: 5px solid #f3f3f3;
+        border-top: 5px solid #3498db;
+        border-radius: 50%;
+        animation: spin 1s linear infinite;
+        margin: 0 auto 20px;
+    }
+
+    @keyframes spin {
+        0% { transform: rotate(0deg); }
+        100% { transform: rotate(360deg); }
+    }
+
+    .loading-text {
+        color: #2c3e50;
+        font-size: 18px;
+        margin-top: 15px;
+    }
+
+    .progress-steps {
+        margin-top: 15px;
+        text-align: left;
+    }
+
+    .step {
+        margin: 8px 0;
+        color: #666;
+    }
+
+    .step.active {
+        color: #3498db;
+        font-weight: bold;
+    }
+
+    .step.completed {
+        color: #2ecc71;
+    }
+
+    .care-plan-container {
+        margin-top: 30px;
+    }
+
+    .table-responsive {
+        margin-bottom: 30px;
+        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        border-radius: 8px;
+        overflow: hidden;
+    }
+
+    .table thead {
+        background-color: #3498db;
+        color: white;
+    }
+
+    .table tbody tr:nth-child(even) {
+        background-color: #f8f9fa;
+    }
+
+    .table tbody tr:hover {
+        background-color: #e9ecef;
+    }
+
+    h3 {
+        color: #2c3e50;
+        margin-bottom: 20px;
+        padding-bottom: 10px;
+        border-bottom: 2px solid #3498db;
+    }
+
+    td[contenteditable="true"] {
+        padding: 5px;
+        border: 1px solid #ffc107;
+        border-radius: 4px;
+    }
+
+    .edit-btn, .save-btn {
+        padding: 2px 8px;
+        margin: 0 2px;
+    }
+
+    .save-btn {
+        background-color: #28a745;
+        border-color: #28a745;
+    }
+
+    .edit-btn:hover {
+        background-color: #0056b3;
+    }
+
+    .save-btn:hover {
+        background-color: #218838;
+    }
+
+    @keyframes slideIn {
+        from { transform: translateX(100%); opacity: 0; }
+        to { transform: translateX(0); opacity: 1; }
+    }
+
+    @keyframes slideOut {
+        from { transform: translateX(0); opacity: 1; }
+        to { transform: translateX(100%); opacity: 0; }
+    }
+
+    .notification-status {
+        border-left: 4px solid;
+        margin-top: 20px;
+    }
+
+    .alert-success.notification-status {
+        border-left-color: #28a745;
+    }
+
+    .alert-warning.notification-status {
+        border-left-color: #ffc107;
+    }
+
+    .alert-danger.notification-status {
+        border-left-color: #dc3545;
+    }
+
+    .meal-activity-section {
+        margin-top: 30px;
+    }
+
+    .action-buttons {
+        display: flex;
+        gap: 10px;
+        margin-top: 20px;
+        justify-content: center;
+    }
+
+    .action-buttons button {
+        flex: 1;
+        max-width: 200px;
+    }
+
+    .add-item-btn {
+        margin-top: 10px;
+        background-color: #17a2b8;
+    }
+
+    #notificationControls {
+        margin-top: 20px;
+        padding: 15px;
+        border: 1px solid #ddd;
+        border-radius: 8px;
+        background-color: #f8f9fa;
+    }
+  </style>
+</head>
+<body>
+  <div class="loading-overlay" id="loadingOverlay">
+    <div class="loading-content">
+        <div class="spinner"></div>
+        <div class="loading-text">Processing your audio file...</div>
+        <div class="progress-steps">
+            <div class="step" id="step1">1. Uploading audio file...</div>
+            <div class="step" id="step2">2. Transcribing audio...</div>
+            <div class="step" id="step3">3. Generating care plan...</div>
+        </div>
+    </div>
+  </div>
+  <div class="container section">
+    <h1>Medical Care Plan Generator</h1>
+    <form id="uploadForm" class="mb-4">
+      <div class="mb-3">
+        <label for="audioFile" class="form-label">Upload Audio Recording</label>
+        <input type="file" id="audioFile" name="audio" accept="audio/*" class="form-control" required>
+      </div>
+      <button type="submit" class="btn btn-primary">Generate Care Plan</button>
+    </form>
+    <div id="loading" class="loading">
+      Processing audio and generating care plan...
+    </div>
+    <div id="carePlanResult"></div>
+  </div>
+
+  <!-- Medication Schedule Section -->
+  <div class="container section">
+    <h2 class="text-center mb-4">Patient Daily Schedule</h2>
+    <div id="medicationTableSection"></div>
+
+    <!-- Meal Schedule Section -->
+    <div id="mealScheduleSection" class="meal-activity-section"></div>
+
+    <!-- Activity Schedule Section -->
+    <div id="activityScheduleSection" class="meal-activity-section"></div>
+
+    <div class="action-buttons">
+      <button id="generateTimetableBtn" class="btn btn-primary mt-3" style="display:none;">
+        <i class="fas fa-calendar-check"></i> Generate Daily Timetables
+      </button>
+      <button id="downloadExcelBtn" class="btn btn-success mt-3" style="display:none;">
+        <i class="fas fa-file-excel"></i> Download Excel Schedule
+      </button>
+    </div>
+
+    <!-- Notification Controls Section -->
+    <div id="notificationControls" style="display:none;">
+      <h3><i class="fas fa-bell"></i> WhatsApp Notifications</h3>
+      <div class="form-check form-switch mb-3">
+        <input class="form-check-input" type="checkbox" id="enableNotifications" checked>
+        <label class="form-check-label" for="enableNotifications">Enable WhatsApp Reminders</label>
+      </div>
+      <div id="notificationStatus"></div>
+    </div>
+
+    <div id="timetableResult" class="mt-4"></div>
+  </div>
+
+  <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.3/dist/js/bootstrap.bundle.min.js"></script>
+  <script>
+    // Add a variable to store the care plan data
+    let carePlanData = null;
+    let allScheduleData = {};
+
+    function createTable(title, headers, data, editable = true, tableType = '') {
+        if (!data || data.length === 0) {
+            let html = `
+                <div class="mt-4">
+                    <h3>${title}</h3>
+                    <p class="text-muted">No data available</p>`;
+
+            if (tableType) {
+                html += `<button class="btn btn-sm add-item-btn" onclick="addNewItem('${tableType}')">
+                    <i class="fas fa-plus"></i> Add New Item
+                </button>`;
+            }
+
+            html += `</div>`;
+            return html;
+        }
+
+        let html = `<div class="mt-4"><h3>${title}</h3>`;
+        html += '<div class="table-responsive">';
+        const tableId = tableType ? ` id="${tableType}Table"` : '';
+        html += `<table class="table table-bordered"${tableId}>`;
+        html += '<thead class="table-primary"><tr>';
+
+        headers.forEach(header => {
+            html += `<th>${header}</th>`;
+        });
+
+        // Add Dosage column for medication table
+        if (tableType === 'medication') {
+            html += '<th>Dosage</th>';
+            html += '<th>Frequency per Day</th>';
+        }
+
+        if (editable) {
+            html += '<th>Actions</th>';
+        }
+
+        html += '</tr></thead><tbody>';
+
+        data.forEach((row, index) => {
+            html += '<tr>';
+            headers.forEach(header => {
+                const value = row[header] || '';
+                html += `<td contenteditable="false">${value}</td>`;
+            });
+
+            // Add dosage field for medication table
+            if (tableType === 'medication') {
+                const dosage = row['Dosage'] || '';
+                const frequency = row['frequency'] || 1;
+
+                html += `<td contenteditable="false">${dosage}</td>`;
+                html += `
+                    <td>
+                        <select class="form-select frequency-select">
+                            <option value="1" ${frequency == 1 ? 'selected' : ''}>Once daily (Dinner)</option>
+                            <option value="2" ${frequency == 2 ? 'selected' : ''}>Twice daily (Breakfast & Dinner)</option>
+                            <option value="3" ${frequency == 3 ? 'selected' : ''}>Three times daily (Breakfast, Lunch & Dinner)</option>
+                        </select>
+                    </td>
+                `;
+            }
+
+            if (editable) {
+                html += `
+                    <td>
+                        <button class="btn btn-sm btn-primary edit-btn" onclick="toggleEdit(this, '${tableType}', ${index})">
+                            <i class="fas fa-edit"></i> Edit
+                        </button>
+                        <button class="btn btn-sm btn-success save-btn" onclick="saveEdit(this, '${tableType}', ${index})" style="display:none;">
+                            <i class="fas fa-save"></i> Save
+                        </button>
+                        <button class="btn btn-sm btn-danger" onclick="deleteItem('${tableType}', ${index})">
+                            <i class="fas fa-trash"></i>
+                        </button>
+                    </td>
+                `;
+            }
+
+            html += '</tr>';
+        });
+
+        html += '</tbody></table></div>';
+
+        // Add button to add new items
+        if (tableType) {
+            html += `<button class="btn btn-sm add-item-btn" onclick="addNewItem('${tableType}')">
+                <i class="fas fa-plus"></i> Add New Item
+            </button>`;
+        }
+
+        html += '</div>';
+        return html;
+    }
+
+    function addNewItem(tableType) {
+        let newItem = {};
+
+        if (tableType === 'medication') {
+            newItem = { "Medication Name": "New Medication", "Dosage": "1 tablet", "frequency": 1 };
+            if (!carePlanData.medication_schedule) carePlanData.medication_schedule = [];
+            carePlanData.medication_schedule.push(newItem);
+            document.getElementById('medicationTableSection').innerHTML = createTable(
+                'Medication Schedule',
+                ["Medication Name"],
+                carePlanData.medication_schedule,
+                true,
+                'medication'
+            );
+        }
+        else if (tableType === 'meal') {
+            newItem = { "Time": "12:00 PM", "Meal": "New Meal", "Details": "Meal details" };
+            if (!carePlanData.meal_schedule) carePlanData.meal_schedule = [];
+            carePlanData.meal_schedule.push(newItem);
+            document.getElementById('mealScheduleSection').innerHTML = createTable(
+                'Meal Schedule',
+                ["Time", "Meal", "Details"],
+                carePlanData.meal_schedule,
+                true,
+                'meal'
+            );
+        }
+        else if (tableType === 'activity') {
+            newItem = { "Time": "5:00 PM", "Activity": "New Activity", "Duration": "30 min", "Notes": "Activity notes" };
+            if (!carePlanData.activity_schedule) carePlanData.activity_schedule = [];
+            carePlanData.activity_schedule.push(newItem);
+            document.getElementById('activityScheduleSection').innerHTML = createTable(
+                'Activity Schedule',
+                ["Time", "Activity", "Duration", "Notes"],
+                carePlanData.activity_schedule,
+                true,
+                'activity'
+            );
+        }
+
+        // Show generate button
+        document.getElementById('generateTimetableBtn').style.display = 'block';
+    }
+
+    function deleteItem(tableType, index) {
+        if (confirm('Are you sure you want to delete this item?')) {
+            if (tableType === 'medication' && carePlanData.medication_schedule) {
+                carePlanData.medication_schedule.splice(index, 1);
+                document.getElementById('medicationTableSection').innerHTML = createTable(
+                    'Medication Schedule',
+                    ["Medication Name"],
+                    carePlanData.medication_schedule,
+                    true,
+                    'medication'
+                );
+            }
+            else if (tableType === 'meal' && carePlanData.meal_schedule) {
+                carePlanData.meal_schedule.splice(index, 1);
+                document.getElementById('mealScheduleSection').innerHTML = createTable(
+                    'Meal Schedule',
+                    ["Time", "Meal", "Details"],
+                    carePlanData.meal_schedule,
+                    true,
+                    'meal'
+                );
+            }
+            else if (tableType === 'activity' && carePlanData.activity_schedule) {
+                carePlanData.activity_schedule.splice(index, 1);
+                document.getElementById('activityScheduleSection').innerHTML = createTable(
+                    'Activity Schedule',
+                    ["Time", "Activity", "Duration", "Notes"],
+                    carePlanData.activity_schedule,
+                    true,
+                    'activity'
+                );
+            }
+
+            // Regenerate timetables if they're already displayed
+            if (document.getElementById('timetableResult').innerHTML !== '') {
+                document.getElementById('generateTimetableBtn').click();
+            }
+        }
+    }
+
+    function toggleEdit(button, tableType, index) {
+        const row = button.closest('tr');
+        const cells = row.querySelectorAll('td:not(:last-child)');
+        const editBtn = button;
+        const saveBtn = row.querySelector('.save-btn');
+
+        // Don't make frequency selector cell editable for medication table
+        const editableCells = tableType === 'medication' ?
+            cells.slice(0, -1) : cells;
+
+        editableCells.forEach(cell => {
+            cell.contentEditable = "true";
+            cell.style.backgroundColor = "#fff3cd";
+        });
+
+        editBtn.style.display = "none";
+        saveBtn.style.display = "inline-block";
+    }
+
+    function saveEdit(button, tableType, index) {
+        const row = button.closest('tr');
+        const cells = row.querySelectorAll('td:not(:last-child)');
+        const editBtn = row.querySelector('.edit-btn');
+        const saveBtn = button;
+
+        // Don't process frequency selector cell for medication table
+        const editableCells = tableType === 'medication' ?
+            cells.slice(0, -1) : cells;
+
+        editableCells.forEach(cell => {
+            cell.contentEditable = "false";
+            cell.style.backgroundColor = "";
+        });
+
+        editBtn.style.display = "inline-block";
+        saveBtn.style.display = "none";
+
+        // Update the carePlanData based on the edited values
+        if (tableType === 'medication') {
+            const frequencySelect = row.querySelector('.frequency-select');
+            const frequency = frequencySelect ? parseInt(frequencySelect.value) : 1;
+
+            carePlanData.medication_schedule[index] = {
+                "Medication Name": cells[0].textContent.trim(),
+                "Dosage": cells[1].textContent.trim(),
+                "frequency": frequency
+            };
+        } else if (tableType === 'meal') {
+            carePlanData.meal_schedule[index] = {
+                Time: cells[0].textContent.trim(),
+                Meal: cells[1].textContent.trim(),
+                Details: cells[2].textContent.trim()
+            };
+        } else if (tableType === 'activity') {
+            carePlanData.activity_schedule[index] = {
+                Time: cells[0].textContent.trim(),
+                Activity: cells[1].textContent.trim(),
+                Duration: cells[2].textContent.trim(),
+                Notes: cells[3].textContent.trim()
+            };
+        }
+
+        // Regenerate timetables if they're already displayed
+        if (document.getElementById('timetableResult').innerHTML !== '') {
+            document.getElementById('generateTimetableBtn').click();
+        }
+    }
+
+    function updateLoadingStep(stepNumber) {
+        // Reset all steps
+        document.querySelectorAll('.step').forEach(step => {
+            step.classList.remove('active', 'completed');
+        });
+
+        // Mark completed steps
+        for (let i = 1; i < stepNumber; i++) {
+            document.getElementById(`step${i}`).classList.add('completed');
+        }
+
+        // Mark current step as active
+        document.getElementById(`step${stepNumber}`).classList.add('active');
+    }
+
+    document.getElementById('uploadForm').addEventListener('submit', async function(e) {
+        e.preventDefault();
+
+        const fileInput = document.getElementById('audioFile');
+        const loadingOverlay = document.getElementById('loadingOverlay');
+        const resultDiv = document.getElementById('carePlanResult');
+        const medicationTableSection = document.getElementById('medicationTableSection');
+        const mealScheduleSection = document.getElementById('mealScheduleSection');
+        const activityScheduleSection = document.getElementById('activityScheduleSection');
+
+        if (!fileInput.files.length) {
+            resultDiv.innerHTML = '<div class="alert alert-danger">Please select an audio file</div>';
+            return;
+        }
+
+        const formData = new FormData();
+        formData.append('audio', fileInput.files[0]);
+
+        loadingOverlay.style.display = 'flex';
+        resultDiv.innerHTML = '';
+        medicationTableSection.innerHTML = '';
+        mealScheduleSection.innerHTML = '';
+        activityScheduleSection.innerHTML = '';
+        updateLoadingStep(1);
+
+        try {
+            // Step 1: Upload started
+            updateLoadingStep(1);
+            await new Promise(resolve => setTimeout(resolve, 1000)); // Show upload step
+
+            // Step 2: Transcription
+            updateLoadingStep(2);
+            const response = await fetch('/generate_care_plan', {
+                method: 'POST',
+                body: formData
+            });
+
+            // Step 3: Generating plan
+            updateLoadingStep(3);
+            const data = await response.json();
+
+            if (data.error) {
+                resultDiv.innerHTML = `<div class="alert alert-danger">Error: ${data.error}</div>`;
+            } else {
+                // Store the care plan data globally
+                carePlanData = data;
+
+                // Display the care plan data
+                let html = `
+                    <div class="care-plan-container">
+                        <div class="alert alert-success">
+                            <h2><i class="fas fa-check-circle"></i> Care Plan Generated Successfully</h2>
+                        </div>
+                        <div class="card mb-4">
+                            <div class="card-header bg-info text-white">
+                                <h3 class="mb-0">Patient Information</h3>
+                            </div>
+                            <div class="card-body">
+                                <p><strong>Name:</strong> ${data.patient_name || 'N/A'}</p>
+                                <p><strong>Age:</strong> ${data.patient_age || 'N/A'}</p>
+                                <p><strong>Medical Condition:</strong> ${data.medical_condition || 'N/A'}</p>
+                            </div>
+                        </div>
+                `;
+
+                if (data.prescription_summary) {
+                    html += `
+                        <div class="card mb-4">
+                            <div class="card-header bg-primary text-white">
+                                <h3 class="mb-0">Prescription Summary</h3>
+                            </div>
+                            <div class="card-body">
+                                <p>${data.prescription_summary}</p>
+                            </div>
+                        </div>
+                    `;
+                }
+
+                // Transcription section
+                if (data.transcription) {
+                    html += `
+                        <div class="card mb-4">
+                            <div class="card-header bg-secondary text-white">
+                                <h3 class="mb-0">Audio Transcription</h3>
+                            </div>
+                            <div class="card-body">
+                                <p>${data.transcription}</p>
+                            </div>
+                        </div>
+                    `;
+                }
+
+                resultDiv.innerHTML = html;
+
+                // Create the medication table for setting frequencies
+                if (data.medication_schedule && data.medication_schedule.length > 0) {
+                    medicationTableSection.innerHTML = createTable(
+                        'Medication Schedule',
+                        ["Medication Name"],
+                        data.medication_schedule,
+                        true,
+                        'medication'
+                    );
+                }
+
+                // Create the meal schedule table
+                if (data.meal_schedule && data.meal_schedule.length > 0) {
+                    mealScheduleSection.innerHTML = createTable(
+                        'Meal Schedule',
+                        ["Time", "Meal", "Details"],
+                        data.meal_schedule,
+                        true,
+                        'meal'
+                    );
+                } else {
+                    mealScheduleSection.innerHTML = createTable(
+                        'Meal Schedule',
+                        ["Time", "Meal", "Details"],
+                        [],
+                        true,
+                        'meal'
+                    );
+                }
+
+                // Create the activity schedule table
+                if (data.activity_schedule && data.activity_schedule.length > 0) {
+                    activityScheduleSection.innerHTML = createTable(
+                        'Activity Schedule',
+                        ["Time", "Activity", "Duration", "Notes"],
+                        data.activity_schedule,
+                        true,
+                        'activity'
+                    );
+                } else {
+                    activityScheduleSection.innerHTML = createTable(
+                        'Activity Schedule',
+                        ["Time", "Activity", "Duration", "Notes"],
+                        [],
+                        true,
+                        'activity'
+                    );
+                }
+
+                document.getElementById('generateTimetableBtn').style.display = 'block';
+                document.getElementById('notificationControls').style.display = 'block';
+            }
+        } catch (error) {
+            resultDiv.innerHTML = `<div class="alert alert-danger">Error: ${error.message}</div>`;
+        } finally {
+            loadingOverlay.style.display = 'none';
+        }
+    });
+
+    document.getElementById('generateTimetableBtn').addEventListener('click', function() {
+        const timetableResult = document.getElementById('timetableResult');
+
+        if (!carePlanData) {
+            timetableResult.innerHTML = '<div class="alert alert-warning">No data available</div>';
+            return;
+        }
+
+        // Get medication data from the tables
+        const medicationTable = document.getElementById('medicationTable');
+        let medications = [];
+
+        if (medicationTable) {
+            const rows = medicationTable.querySelectorAll('tbody tr');
+            medications = Array.from(rows).map(row => {
+                const cells = row.querySelectorAll('td');
+                const frequencySelect = row.querySelector('.frequency-select');
+                const frequency = frequencySelect ? parseInt(frequencySelect.value) : 1;
+
+                return {
+                    name: cells[0].textContent.trim(),
+                    dosage: cells[1].textContent.trim() || '1 tablet',
+                    frequency: frequency
+                };
+            });
+        }
+
+        // Prepare data for schedule generation
+        const scheduleData = {
+            medicines: medications.map(med => ({
+                medication_name: med.name,
+                dosage: med.dosage,
+                frequency: med.frequency
+            })),
+            meal_schedule: carePlanData.meal_schedule || [],
+            activity_schedule: carePlanData.activity_schedule || []
+        };
+
+        allScheduleData = scheduleData;
+
+        // Send data to backend to generate schedule and set up notifications
+        fetch('/generate_schedule', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify(scheduleData)
+        })
+        .then(response => response.json())
+        .then(data => {
+            if (data.error) {
+                timetableResult.innerHTML = `<div class="alert alert-danger">Error: ${data.error}</div>`;
+                return;
+            }
+
+            // Update notification status
+            const notificationStatus = document.getElementById('notificationStatus');
+            if (data.notifications_scheduled) {
+                notificationStatus.innerHTML = `
+                    <div class="alert alert-success notification-status">
+                        <i class="fas fa-check-circle"></i> WhatsApp notifications have been scheduled successfully.
+                    </div>
+                `;
+            } else {
+                notificationStatus.innerHTML = `
+                    <div class="alert alert-warning notification-status">
+                        <i class="fas fa-exclamation-triangle"></i> WhatsApp notifications could not be scheduled.
+                    </div>
+                `;
+            }
+
+            // Update medication schedule with the generated data
+            if (data.schedule_list && data.schedule_list.length > 0) {
+                carePlanData.medication_schedule = data.schedule_list;
+            }
+
+            // Generate daily timetable
+            let html = `
+                <div class="alert alert-success">
+                    <h2><i class="fas fa-calendar-alt"></i> Daily Medication Schedule</h2>
+                </div>
+            `;
+
+            // Create daily schedule table
+            html += '<div class="table-responsive">';
+            html += '<table class="table table-bordered">';
+            html += '<thead class="table-primary"><tr><th>Time</th><th>Medications</th><th>Meal</th><th>Activities</th></tr></thead>';
+            html += '<tbody>';
+
+            // Create a combined schedule
+            let combinedSchedule = [];
+
+            // Add medication items
+            carePlanData.medication_schedule.forEach(med => {
+                combinedSchedule.push({
+                    time: med['Scheduled Time'],
+                    type: 'medication',
+                    name: med['Medication Name'],
+                    details: med['Dosage'],
+                    meal: med['Meal']
+                });
+            });
+
+            // Add meal items
+            if (carePlanData.meal_schedule) {
+                carePlanData.meal_schedule.forEach(meal => {
+                    combinedSchedule.push({
+                        time: meal['Time'],
+                        type: 'meal',
+                        name: meal['Meal'],
+                        details: meal['Details']
+                    });
+                });
+            }
+
+            // Add activity items
+            if (carePlanData.activity_schedule) {
+                carePlanData.activity_schedule.forEach(activity => {
+                    combinedSchedule.push({
+                        time: activity['Time'],
+                        type: 'activity',
+                        name: activity['Activity'],
+                        details: activity['Duration'],
+                        notes: activity['Notes']
+                    });
+                });
+            }
+
+            // Sort the combined schedule by time
+            combinedSchedule.sort((a, b) => {
+                // Convert time strings to Date objects for comparison
+                const timeA = new Date(`01/01/2023 ${a.time}`);
+                const timeB = new Date(`01/01/2023 ${b.time}`);
+                return timeA - timeB;
+            });
+
+            // Group items by time
+            const scheduledItems = {};
+            combinedSchedule.forEach(item => {
+                if (!scheduledItems[item.time]) {
+                    scheduledItems[item.time] = {
+                        medications: [],
+                        meals: [],
+                        activities: []
+                    };
+                }
+
+                if (item.type === 'medication') {
+                    scheduledItems[item.time].medications.push({
+                        name: item.name,
+                        details: item.details,
+                        meal: item.meal
+                    });
+                } else if (item.type === 'meal') {
+                    scheduledItems[item.time].meals.push({
+                        name: item.name,
+                        details: item.details
+                    });
+                } else if (item.type === 'activity') {
+                    scheduledItems[item.time].activities.push({
+                        name: item.name,
+                        details: item.details,
+                        notes: item.notes
+                    });
+                }
+            });
+
+            // Create table rows
+            Object.keys(scheduledItems).sort((a, b) => {
+                const timeA = new Date(`01/01/2023 ${a}`);
+                const timeB = new Date(`01/01/2023 ${b}`);
+                return timeA - timeB;
+            }).forEach(time => {
+                const items = scheduledItems[time];
+
+                html += '<tr>';
+                html += `<td class="fw-bold">${time}</td>`;
+
+                // Medications column
+                html += '<td>';
+                if (items.medications.length > 0) {
+                    items.medications.forEach(med => {
+                        html += `<div class="mb-1"><strong>${med.name}</strong> (${med.details})</div>`;
+                    });
+                } else {
+                    html += '-';
+                }
+                html += '</td>';
+
+                // Meals column
+                html += '<td>';
+                if (items.meals.length > 0) {
+                    items.meals.forEach(meal => {
+                        html += `<div class="mb-1"><strong>${meal.name}</strong>`;
+                        if (meal.details) {
+                            html += `<br><small>${meal.details}</small>`;
+                        }
+                        html += '</div>';
+                    });
+                } else if (items.medications.some(med => med.meal)) {
+                    const meal = items.medications[0].meal;
+                    html += `<div class="mb-1"><strong>${meal}</strong></div>`;
+                } else {
+                    html += '-';
+                }
+                html += '</td>';
+
+                // Activities column
+                html += '<td>';
+                if (items.activities.length > 0) {
+                    items.activities.forEach(activity => {
+                        html += `<div class="mb-1"><strong>${activity.name}</strong> (${activity.details})`;
+                        if (activity.notes) {
+                            html += `<br><small>${activity.notes}</small>`;
+                        }
+                        html += '</div>';
+                    });
+                } else {
+                    html += '-';
+                }
+                html += '</td>';
+
+                html += '</tr>';
+            });
+
+            html += '</tbody></table></div>';
+
+            // Show download button
+            html += `
+                <div class="action-buttons">
+                    <button id="printScheduleBtn" class="btn btn-info">
+                        <i class="fas fa-print"></i> Print Schedule
+                    </button>
+                </div>
+            `;
+
+            timetableResult.innerHTML = html;
+            document.getElementById('downloadExcelBtn').style.display = 'block';
+
+            // Add event listener for print button
+            document.getElementById('printScheduleBtn').addEventListener('click', function() {
+                window.print();
+            });
+        })
+        .catch(error => {
+            timetableResult.innerHTML = `<div class="alert alert-danger">Error: ${error.message}</div>`;
+        });
+    });
+
+    // Event listener for downloading Excel file
+    document.getElementById('downloadExcelBtn').addEventListener('click', function() {
+        if (!carePlanData) {
+            alert('No data available for download');
+            return;
+        }
+
+        // Prepare the complete data for download
+        const downloadData = {
+            medication_schedule: carePlanData.medication_schedule || [],
+            meal_schedule: carePlanData.meal_schedule || [],
+            activity_schedule: carePlanData.activity_schedule || []
+        };
+
+        // Request Excel file from server
+        fetch('/download_schedule', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify(downloadData)
+        })
+        .then(response => {
+            if (!response.ok) {
+                throw new Error('Server error');
+            }
+            return response.blob();
+        })
+        .then(blob => {
+            const url = window.URL.createObjectURL(blob);
+            const a = document.createElement('a');
+            a.style.display = 'none';
+            a.href = url;
+            a.download = 'daily_schedule.xlsx';
+            document.body.appendChild(a);
+            a.click();
+            window.URL.revokeObjectURL(url);
+        })
+        .catch(error => {
+            alert(`Error downloading the file: ${error.message}`);
+        });
+    });
+
+    // Listen for changes to notification toggle
+    document.getElementById('enableNotifications').addEventListener('change', function() {
+        const notificationStatus = document.getElementById('notificationStatus');
+        if (this.checked) {
+            notificationStatus.innerHTML = `
+                <div class="alert alert-success notification-status">
+                    <i class="fas fa-check-circle"></i> WhatsApp notifications are enabled.
+                </div>
+            `;
+            // Regenerate schedule to enable notifications
+            if (allScheduleData && Object.keys(allScheduleData).length > 0) {
+                document.getElementById('generateTimetableBtn').click();
+            }
+        } else {
+            notificationStatus.innerHTML = `
+                <div class="alert alert-warning notification-status">
+                    <i class="fas fa-bell-slash"></i> WhatsApp notifications are disabled.
+                </div>
+            `;
+        }
+    });
+  </script>
+</body>
+</html>