Update app.py

app.py

@@ -1,19 +1,38 @@
-from flask import Flask, render_template, request, jsonify, session, redirect
+from flask import Flask, render_template, request, jsonify, session, redirect, url_for, send_file
 import google.generativeai as genai
 import PyPDF2
 import os
 import re
+import json
 from datetime import datetime
+import io
+from twilio.rest import Client
+import base64
+import uuid
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+from reportlab.lib import colors
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle
 
 app = Flask(__name__)
 
-app.secret_key = '688ed745a74bdd7ac238f5b50f4104fb87d6774b8b0a4e06e7e18ac5ed0fa31c'  # Add this for session management
-upload_base = os.getenv('UPLOAD_DIR')
+app.secret_key = '688ed745a74bdd7ac238f5b50f4104fb87d6774b8b0a4e06e7e18ac5ed0fa31c'
+upload_base = os.getenv('UPLOAD_DIR', './uploads')
 upload_folder = os.path.join(upload_base, 'uploads')
 
 app.config['UPLOAD_FOLDER'] = upload_folder
 os.makedirs(upload_folder, exist_ok=True)
 
+# Twilio Configuration
+ACCOUNT_SID = 'AC490e071f8d01bf0df2f03d086c788d87'
+AUTH_TOKEN = '224b23b950ad5a4052aba15893fdf083'
+TWILIO_FROM = 'whatsapp:+14155238886'
+TWILIO_TO = 'whatsapp:+917559355282'  # Hardcoded number as requested
+
+# Initialize Twilio client
+twilio_client = Client(ACCOUNT_SID, AUTH_TOKEN)
+
 # Gemini API Configuration
 genai.configure(api_key="AIzaSyD54ejbjVIVa-F3aD_Urnp8m1EFLUGR__I")
 
@@ -29,8 +48,12 @@ model = genai.GenerativeModel(
     generation_config=generation_config,
 )
 
+# Patient database (in-memory for demo, use a real database in production)
+patients_db = {}
+
 
 def extract_text_from_pdf(pdf_file):
+    """Extract text from uploaded PDF file"""
     try:
         pdf_reader = PyPDF2.PdfReader(pdf_file)
         text = ""
@@ -69,6 +92,238 @@ def extract_care_plan_format(pdf_text):
         return None
 
 
+def determine_patient_status(original_plan, updated_plan, feedback):
+    """Determine patient status based on care plan comparison and feedback"""
+    # Keywords indicating condition worsening
+    emergency_keywords = [
+        "severe chest pain", "heart attack", "shortness of breath",
+        "dizziness", "loss of consciousness", "extreme pain",
+        "sudden weakness", "confusion", "slurred speech", "severe headache",
+        "difficulty breathing", "severe shortness of breath", "wheezing",
+        "severe abdominal pain", "persistent vomiting", "uncontrolled bleeding",
+        "severe allergic reaction", "anaphylaxis", "immediate medical attention",
+        "emergency", "call 911", "urgent care", "hospital", "critical"
+    ]
+    
+    deteriorating_keywords = [
+        "worsening", "increased pain", "not improving", "deteriorating",
+        "elevated", "higher", "more frequent", "concerning", "monitor closely",
+        "decline", "decreased function", "less able", "more difficult"
+    ]
+    
+    improvement_keywords = [
+        "improving", "better", "reduced", "lower", "less pain", "increased function",
+        "healing", "recovery", "progress", "stable", "maintained", "consistent",
+        "well-controlled", "responsive", "good progress"
+    ]
+    
+    original_plan_lower = original_plan.lower() if original_plan else ""
+    updated_plan_lower = updated_plan.lower()
+    feedback_lower = feedback.lower()
+    
+    # Check for emergency keywords first - higher priority in feedback
+    if any(keyword in feedback_lower for keyword in emergency_keywords):
+        return "emergency"
+    
+    # Check for emergency keywords in updated plan
+    if any(keyword in updated_plan_lower for keyword in emergency_keywords):
+        return "emergency"
+    
+    # Check for deteriorating keywords
+    if any(keyword in feedback_lower for keyword in deteriorating_keywords) or \
+       any(keyword in updated_plan_lower for keyword in deteriorating_keywords):
+        return "deteriorating"
+    
+    # Check for improvement keywords
+    if any(keyword in feedback_lower for keyword in improvement_keywords) or \
+       any(keyword in updated_plan_lower for keyword in improvement_keywords):
+        return "improving"
+    
+    # Default to stable if no clear indicators
+    return "stable"
+
+
+def generate_care_plan_pdf(patient_info, care_plan_text, status):
+    """Generate a PDF of the care plan with improved styling"""
+    buffer = io.BytesIO()
+    
+    # Create a PDF document
+    doc = SimpleDocTemplate(buffer, pagesize=letter)
+    styles = getSampleStyleSheet()
+    
+    # Create custom styles
+    title_style = ParagraphStyle(
+        'Title',
+        parent=styles['Heading1'],
+        fontSize=20,
+        alignment=1,  # Center alignment
+        spaceAfter=20,
+        textColor=colors.navy
+    )
+    
+    heading_style = ParagraphStyle(
+        'Heading',
+        parent=styles['Heading2'],
+        fontSize=14,
+        spaceAfter=10,
+        spaceBefore=15,
+        textColor=colors.darkblue
+    )
+    
+    normal_style = ParagraphStyle(
+        'Normal',
+        parent=styles['Normal'],
+        fontSize=11,
+        spaceAfter=5,
+        leading=14
+    )
+    
+    bullet_style = ParagraphStyle(
+        'Bullet',
+        parent=styles['Normal'],
+        fontSize=11,
+        spaceAfter=3,
+        leftIndent=20,
+        leading=14
+    )
+    
+    # Status color mapping
+    status_colors = {
+        'emergency': colors.red,
+        'deteriorating': colors.orange,
+        'improving': colors.green,
+        'stable': colors.blue
+    }
+    
+    status_style = ParagraphStyle(
+        'Status',
+        parent=styles['Heading2'],
+        fontSize=14,
+        spaceBefore=10,
+        spaceAfter=10,
+        textColor=status_colors.get(status, colors.black)
+    )
+    
+    # Build the document content
+    story = []
+    
+    # Title
+    story.append(Paragraph("Patient Care Plan", title_style))
+    story.append(Spacer(1, 10))
+    
+    # Status indicator
+    status_text = {
+        'emergency': "EMERGENCY - IMMEDIATE ACTION REQUIRED",
+        'deteriorating': "HIGH RISK - Condition Deteriorating",
+        'improving': "LOW RISK - Condition Improving",
+        'stable': "STABLE - Maintain Current Care"
+    }
+    
+    story.append(Paragraph(status_text.get(status, "Status: Unknown"), status_style))
+    story.append(Spacer(1, 15))
+    
+    # Patient Information Table
+    patient_data = [
+        ["Patient Name:", patient_info.get('name', 'N/A')],
+        ["Age:", patient_info.get('age', 'N/A')],
+        ["Gender:", patient_info.get('gender', 'N/A')],
+        ["Generated Date:", datetime.now().strftime("%Y-%m-%d %H:%M")]
+    ]
+    
+    patient_table = Table(patient_data, colWidths=[120, 300])
+    patient_table.setStyle(TableStyle([
+        ('GRID', (0, 0), (-1, -1), 0.5, colors.grey),
+        ('BACKGROUND', (0, 0), (0, -1), colors.lightgrey),
+        ('FONTNAME', (0, 0), (0, -1), 'Helvetica-Bold'),
+        ('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
+        ('LEFTPADDING', (0, 0), (-1, -1), 6),
+        ('RIGHTPADDING', (0, 0), (-1, -1), 6),
+        ('TOPPADDING', (0, 0), (-1, -1), 4),
+        ('BOTTOMPADDING', (0, 0), (-1, -1), 4),
+    ]))
+    
+    story.append(patient_table)
+    story.append(Spacer(1, 20))
+    
+    # Care Plan Content
+    story.append(Paragraph("Care Plan Details:", heading_style))
+    story.append(Spacer(1, 10))
+    
+    # Process care plan text by sections
+    sections = re.findall(r'([A-Z][A-Z\s]+):(.*?)(?=\n[A-Z][A-Z\s]+:|$)', care_plan_text, re.DOTALL)
+    
+    for section_title, section_content in sections:
+        story.append(Paragraph(section_title + ":", heading_style))
+        
+        # Process bullet points if they exist
+        bullet_points = re.findall(r'[-•]\s*(.*?)(?=[-•]|\n|$)', section_content.strip())
+        
+        if bullet_points:
+            for point in bullet_points:
+                if point.strip():
+                    story.append(Paragraph(f"• {point.strip()}", bullet_style))
+        else:
+            # If no bullet points, add the content as a paragraph
+            cleaned_content = section_content.strip()
+            if cleaned_content:
+                story.append(Paragraph(cleaned_content, normal_style))
+        
+        story.append(Spacer(1, 5))
+    
+    # Build the PDF
+    doc.build(story)
+    buffer.seek(0)
+    return buffer
+
+
+def send_whatsapp_care_plan(patient_name, care_plan_text, status):
+    """Send care plan via WhatsApp using Twilio with improved formatting"""
+    try:
+        # Status emoji
+        status_emoji = {
+            'emergency': "🚨 EMERGENCY",
+            'deteriorating': "⚠️ HIGH RISK",
+            'improving': "✅ IMPROVING",
+            'stable': "🟦 STABLE"
+        }
+        
+        # Format message for WhatsApp with improved structure
+        message = f"*Care Plan Update for {patient_name}*\n\n"
+        message += f"*Status: {status_emoji.get(status, 'Unknown')}*\n\n"
+        
+        # Extract and format sections
+        sections = re.findall(r'([A-Z][A-Z\s]+):(.*?)(?=\n[A-Z][A-Z\s]+:|$)', care_plan_text, re.DOTALL)
+        
+        for section_title, section_content in sections:
+            message += f"*{section_title}:*\n"
+            
+            # Process bullet points if they exist
+            bullet_points = re.findall(r'[-•]\s*(.*?)(?=[-•]|\n|$)', section_content.strip())
+            
+            if bullet_points:
+                for point in bullet_points:
+                    if point.strip():
+                        message += f"• {point.strip()}\n"
+            else:
+                # If no bullet points, add the content as a paragraph
+                cleaned_content = section_content.strip()
+                if cleaned_content:
+                    message += f"{cleaned_content}\n"
+            
+            message += "\n"
+        
+        # Send WhatsApp message using hardcoded number
+        twilio_client.messages.create(
+            from_=TWILIO_FROM,
+            body=message,
+            to=TWILIO_TO
+        )
+        return True
+    except Exception as e:
+        print(f"Error sending WhatsApp message: {e}")
+        return False
+
+
 @app.route('/')
 def index():
     return render_template('index.html')
@@ -83,29 +338,41 @@ def switch_role():
 
 @app.route('/doctor_dashboard')
 def doctor_dashboard():
-    if session.get('role') != 'doctor':
-        return redirect('/')
     return render_template('doctor_dashboard.html')
 
 
+@app.route('/patient_details/<patient_id>')
+def patient_details(patient_id):
+    patient = patients_db.get(patient_id)
+    if not patient:
+        return redirect('/doctor_dashboard')
+    return render_template('doctor_dashboard.html', selected_patient=patient)
+
+
 @app.route('/submit_feedback', methods=['POST'])
 def submit_feedback():
     try:
+        # Get patient information
+        name = request.form.get('name', '')
+        age = request.form.get('age', '')
+        gender = request.form.get('gender', '')
         feedback = request.form.get('feedback', '')
+        
         care_plan_text = ""
         care_plan_format = None
-
+        
         if 'care_plan_pdf' in request.files:
             pdf_file = request.files['care_plan_pdf']
             if pdf_file.filename != '':
                 care_plan_text = extract_text_from_pdf(pdf_file)
                 care_plan_format = extract_care_plan_format(care_plan_text)
-
+        
         # If no format is found in the PDF, use a default format
         if not care_plan_format:
             care_plan_format = """
 ASSESSMENT:
-[Assessment details]
+- [Assessment details]
+
 DAILY CARE PLAN:
 Morning:
 - [Morning activities]
@@ -113,95 +380,221 @@ Afternoon:
 - [Afternoon activities]
 Evening:
 - [Evening activities]
+
 MEDICATIONS:
 - [Medication details]
+
 ADDITIONAL RECOMMENDATIONS:
 - [Recommendations]
+
 FOLLOW-UP:
-[Follow-up details]
+- [Follow-up details]
             """
-
+        
         # Define emergency keywords to check for severe symptoms
         emergency_keywords = [
             # Cardiovascular
             "severe chest pain", "heart attack", "shortness of breath",
             "dizziness", "loss of consciousness", "extreme pain",
-
+            
             # Neurological
             "sudden weakness", "confusion", "slurred speech", "severe headache",
-
+            
             # Respiratory
             "difficulty breathing", "severe shortness of breath", "wheezing", "respiratory distress",
-
+            
             # Gastrointestinal
             "severe abdominal pain", "persistent vomiting", "uncontrolled bleeding",
-
+            
             # Others
             "severe allergic reaction", "anaphylaxis"
         ]
-
+        
         feedback_lower = feedback.lower()
         is_emergency = any(keyword in feedback_lower for keyword in emergency_keywords)
-
+        
         if is_emergency:
-            # Store emergency notification
-            emergency_data = {
-                'patient_feedback': feedback,
-                'timestamp': datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-                'status': 'urgent'
-            }
-            # In a real application, you would store this in a database
-            # For now, we'll use a global variable (not recommended for production)
-            if not hasattr(app, 'emergency_notifications'):
-                app.emergency_notifications = []
-            app.emergency_notifications.append(emergency_data)
-
             # If emergency symptoms are detected, instruct immediate emergency response
             emergency_message = (
-                "Emergency symptoms detected. Please call emergency services immediately at 104/108/109/112."
+                "ASSESSMENT:\n"
+                "- Emergency symptoms detected. Immediate medical attention required.\n\n"
+                "EMERGENCY ACTION PLAN:\n"
+                "- Call emergency services immediately at 104/108/109/112.\n"
+                "- Do not delay seeking medical help.\n"
+                "- Do not eat or drink anything until evaluated by medical professionals.\n\n"
+                "FOLLOW-UP:\n"
+                "- Immediate hospitalization may be required.\n"
+                "- Contact your primary physician as soon as possible.\n"
             )
+            
+            # Create a patient record with emergency status
+            patient_id = str(uuid.uuid4())
+            patients_db[patient_id] = {
+                'id': patient_id,
+                'name': name,
+                'age': age,
+                'gender': gender,
+                'feedback': feedback,
+                'original_plan': care_plan_text,
+                'updated_plan': emergency_message,
+                'status': 'emergency',
+                'timestamp': datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            }
+            
+            # Generate PDF for emergency instructions
+            patient_info = {
+                'name': name,
+                'age': age,
+                'gender': gender
+            }
+            pdf_buffer = generate_care_plan_pdf(patient_info, emergency_message, 'emergency')
+            pdf_base64 = base64.b64encode(pdf_buffer.getvalue()).decode('utf-8')
+            
+            # Send emergency WhatsApp message
+            send_whatsapp_care_plan(name, emergency_message, 'emergency')
+            
             return jsonify({
                 'success': True,
-                'updated_plan': emergency_message
+                'updated_plan': emergency_message,
+                'patient_id': patient_id,
+                'status': 'emergency',
+                'pdf_data': pdf_base64
             })
-
+        
         # Prepare a prompt for Gemini AI for a perfect, attractively formatted day care plan.
         prompt = f"""
+Patient Information:
+Name: {name}
+Age: {age}
+Gender: {gender}
+
 Patient Care Plan Update Request:
 Current Symptoms and Feedback: {feedback}
 Current Care Plan (extracted from PDF): {care_plan_text}
+
 Based on the patient's feedback and current care plan, please provide an updated, perfect daily care plan in the exact following format:
 {care_plan_format}
+
 Please ensure the following:
 - The response is well-structured with clear section headings.
-- Use bullet points for list items without including any asterisk (*) symbols.
+- Use bullet points for list items.
 - Format the text attractively and professionally.
-- Remove any extraneous symbols.
+- Be specific with recommendations based on the patient's feedback.
+- Include specific times for medications if applicable.
+- Provide clear follow-up instructions.
 """
         # Get response from Gemini AI
         response = model.generate_content(prompt)
         # Remove any asterisk symbols from the response text
         updated_plan = response.text.replace("*", "")
-
+        
+        # Determine patient status
+        status = determine_patient_status(care_plan_text, updated_plan, feedback)
+        
+        # Generate PDF for downloading
+        patient_info = {
+            'name': name,
+            'age': age,
+            'gender': gender
+        }
+        pdf_buffer = generate_care_plan_pdf(patient_info, updated_plan, status)
+        pdf_base64 = base64.b64encode(pdf_buffer.getvalue()).decode('utf-8')
+        
+        # Create a new patient record or update existing
+        patient_id = str(uuid.uuid4())
+        patients_db[patient_id] = {
+            'id': patient_id,
+            'name': name,
+            'age': age,
+            'gender': gender,
+            'feedback': feedback,
+            'original_plan': care_plan_text,
+            'updated_plan': updated_plan,
+            'status': status,
+            'timestamp': datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        }
+        
+        # Send care plan via WhatsApp
+        whatsapp_sent = send_whatsapp_care_plan(name, updated_plan, status)
+        
         return jsonify({
             'success': True,
-            'updated_plan': updated_plan
+            'updated_plan': updated_plan,
+            'pdf_data': pdf_base64,
+            'patient_id': patient_id,
+            'status': status,
+            'whatsapp_sent': whatsapp_sent
         })
-
+    
     except Exception as e:
+        print(f"Error: {str(e)}")
         return jsonify({
             'success': False,
             'error': str(e)
         })
 
 
+@app.route('/download_pdf/<patient_id>')
+def download_pdf(patient_id):
+    if patient_id not in patients_db:
+        return "Patient not found", 404
+    
+    patient = patients_db[patient_id]
+    pdf_buffer = generate_care_plan_pdf({
+        'name': patient['name'],
+        'age': patient['age'],
+        'gender': patient['gender']
+    }, patient['updated_plan'], patient['status'])
+    
+    return send_file(
+        pdf_buffer,
+        as_attachment=True,
+        download_name=f"care_plan_{patient['name'].replace(' ', '_')}.pdf",
+        mimetype='application/pdf'
+    )
+
+
 @app.route('/get_emergency_notifications')
 def get_emergency_notifications():
-    if session.get('role') != 'doctor':
-        return jsonify({'success': False, 'error': 'Unauthorized'})
-
-    notifications = getattr(app, 'emergency_notifications', [])
-    return jsonify({'success': True, 'notifications': notifications})
+    # Filter patients with emergency status
+    emergency_patients = [p for p in patients_db.values() if p['status'] == 'emergency']
+    
+    notifications = []
+    for patient in emergency_patients:
+        notifications.append({
+            'patient_id': patient['id'],
+            'patient_name': patient['name'],
+            'patient_age': patient['age'],
+            'patient_gender': patient['gender'],
+            'patient_feedback': patient['feedback'],
+            'timestamp': patient['timestamp'],
+            'status': patient['status']
+        })
+    
+    return jsonify({
+        'success': True,
+        'notifications': notifications
+    })
+
+
+@app.route('/get_patients')
+def get_patients():
+    # Return all patients for the doctor dashboard
+    return jsonify({
+        'success': True,
+        'patients': list(patients_db.values())
+    })
+
+
+@app.route('/get_patient/<patient_id>')
+def get_patient(patient_id):
+    if patient_id not in patients_db:
+        return jsonify({'success': False, 'error': 'Patient not found'})
+    
+    return jsonify({
+        'success': True,
+        'patient': patients_db[patient_id]
+    })
 
 
 if __name__ == '__main__':