Update app.py

app.py

@@ -652,7 +652,7 @@ def doctor_dashboard():
          logger.info("Accessed doctor dashboard, setting role to doctor.")
     return render_template('doctor_dashboard.html')
 
-
+# --- START: Modified submit_feedback function (includes optimized prompt logic) ---
 @app.route('/submit_feedback', methods=['POST'])
 def submit_feedback():
     ai_enabled = bool(model) # Check if model is initialized
@@ -681,7 +681,6 @@ def submit_feedback():
              age = None # Store as None if not provided
 
         care_plan_text = "" # This will store the extracted text from PDF
-        care_plan_format = None # This will store the detected format
 
         if 'care_plan_pdf' in request.files:
             pdf_file = request.files['care_plan_pdf']
@@ -694,25 +693,35 @@ def submit_feedback():
                 logger.info(f"Processing uploaded PDF: {pdf_file.filename}")
                 care_plan_text = extract_text_from_pdf(pdf_file)
 
-                # If extraction resulted in an error message, set format to None
-                if "[Error extracting PDF text" in care_plan_text or "[No readable text found" in care_plan_text or "[PDF Content Unavailable" in care_plan_text:
-                     care_plan_format = None
-                     logger.warning(f"PDF text extraction failed or empty: {care_plan_text}")
-                else:
-                    care_plan_format = extract_care_plan_format(care_plan_text)
-                    logger.info(f"Extracted text length: {len(care_plan_text)}. Format found: {care_plan_format is not None}")
+                # If extraction resulted in an error message, the variable will contain it.
+                # This error message will be passed to the AI as "Previous Care Plan Details".
+                logger.info(f"Extracted text length: {len(care_plan_text)}. Begins with: {care_plan_text[:100]}...")
             else:
                  logger.info("No PDF file uploaded or file is empty.")
 
-        # Determine the initial status based on feedback and original plan
+        # Determine the initial status based on feedback and original plan text
         # Pass "" for updated_plan initially, as it hasn't been generated yet for status check
         initial_status = determine_patient_status(care_plan_text, "", feedback)
         logger.info(f"Initial status determined based on feedback/original plan: {initial_status}")
 
-
         generated_plan_text = "" # This will store the AI-generated or fallback plan
         final_status_to_save = initial_status # Start with initial status
 
+        # --- Define the NEW, FIXED Care Plan Structure for the AI Output ---
+        # Removed 'Emergency Contacts' and 'Red Flags' sections, added 'PHYSICAL ACTIVITY AND EXERCISE'
+        required_care_plan_structure = """
+PATIENT INFORMATION:
+ASSESSMENT:
+DAILY CARE PLAN:
+Morning:
+Afternoon:
+Night:
+MEDICATIONS:
+DIET AND HYDRATION:
+PHYSICAL ACTIVITY AND EXERCISE:
+SYMPTOM MANAGEMENT:
+"""
+
         # Only generate AI plan if AI is enabled AND status isn't immediate emergency based on feedback
         # If feedback triggers "emergency", the generated_plan_text is a fixed emergency plan.
         if final_status_to_save == 'emergency':
@@ -725,82 +734,69 @@ def submit_feedback():
                 f"- Gender: {gender}\n\n"
                 "ASSESSMENT:\n"
                 f"- Emergency symptoms reported: {feedback}. Immediate medical attention required.\n\n"
-                "EMERGENCY ACTION PLAN:\n"
-                "- *Call emergency services immediately* at your local emergency number (e.g., 104/108/109/112).\n"
-                "- Do not delay seeking medical help. If possible, have someone stay with the patient.\n"
-                "- If conscious, help the patient into a comfortable position (e.g., upright for breathing difficulties, on back with legs elevated for shock).\n"
-                "- Do not give food or drink until evaluated by medical professionals.\n"
-                "- Prepare relevant medical history, medication list, and previous care plan if available for emergency responders.\n"
-                "- Follow *all* instructions from emergency responders.\n\n"
-                 "RED FLAGS / WHEN TO SEEK HELP:\n"
-                 "- *Any* worsening of reported emergency symptoms requires urgent re-evaluation by medical professionals.\n"
-                 "- Do not attempt to manage severe symptoms at home once emergency signs are present.\n\n"
-                "FOLLOW-UP:\n"
-                "- Immediate hospitalization or urgent medical evaluation is necessary.\n"
-                "- Inform your primary physician/care team as soon as medically stable.\n"
-                "- A new care plan will be developed after the emergency situation is resolved and evaluated by medical professionals.\n"
+                "DAILY CARE PLAN:\n" # Using the new structure headings
+                "Morning:\n"
+                "- Seek immediate medical evaluation.\n"
+                "Afternoon:\n"
+                "- Follow instructions from emergency services.\n"
+                "Night:\n"
+                "- Remain under medical supervision if advised.\n\n"
+                "MEDICATIONS:\n"
+                "- Specific medicines and dosage usage based on symptoms (ex : for fever paramcetamol 1-1-1)\n\n"
+                "DIET AND HYDRATION:\n"
+                "- Do not consume food or drink unless advised by medical professionals.\n\n"
+                "PHYSICAL ACTIVITY AND EXERCISE:\n"
+                "- Rest completely and avoid any physical activity until evaluated by medical professionals.\n\n"
+                "SYMPTOM MANAGEMENT:\n"
+                "- Focus on seeking professional help. Do not attempt home management of emergency symptoms.\n"
             )
-            # For emergency, the generated plan text *is* the final plan to save/send.
             # Status remains 'emergency', which was already set as final_status_to_save.
 
 
         elif ai_enabled: # AI is enabled and initial status is not emergency
-            # Define or get the format template
-            if not care_plan_format or not care_plan_format.strip():
-                logger.info("Using default care plan format as extraction failed or returned empty.")
-                care_plan_format = """
-PATIENT INFORMATION:
-- Name: [Patient Name]
-- Age: [Age]
-- Gender: [Gender] (no other parameter other than these three should be there , not even contact number i want only three Name)
-ASSESSMENT:
-- [Summary of patient's current condition based on feedback and previous plan]
-DAILY CARE PLAN:
-Morning:
-- [Morning activities/medications/checks]
-Afternoon:
-- [Afternoon activities/medications/checks]
-Night:
-- [Night activities/medications/sleep instructions/checks]
-MEDICATIONS:
-- [List of medications, dosage, frequency, and time, including PRN (as needed) instructions] {hey please give it by general sense dont response like i am not physician etc , give , if you dont know still appropriate medicates pelase give)}
-- DIET AND HYDRATION 
-- Physical activity and exercise
-SYMPTOM MANAGEMENT:
-- [Short instructions for managing specific symptoms (e.g., pain, nausea, shortness of breath), non-pharmacological interventions]
-                """ # Enhanced default format
+            logger.info("AI is enabled and status is not emergency. Generating plan via AI.")
 
+            # --- MODIFIED AND OPTIMIZED AI PROMPT (This is the key change) ---
             prompt = f"""
-You are a helpful and highly structured AI assistant generating updated patient care plans.
-Your task is to create a NEW, comprehensive, and well-structured daily care plan based on the provided patient's personal information, their current symptoms/feedback, and their previous care plan (if available).
-Patient Information:
+You are an expert AI assistant specialized in generating concise, structured patient care plans.
+
+Based on the following Patient Information, Patient Feedback, and Previous Care Plan (if available), generate a NEW care plan.
+
+--- Patient Information ---
 Name: {name}
 Age: {age if age is not None else 'N/A'}
 Gender: {gender}
-Patient Feedback/Symptoms Update:
+
+--- Patient Feedback/Symptoms Update ---
 {feedback}
-Previous Care Plan Details (if available):
+
+--- Previous Care Plan Details (if available) ---
 {care_plan_text if care_plan_text and "[No readable text found" not in care_plan_text and "[Error extracting PDF text" not in care_plan_text and "[PDF Content Unavailable" not in care_plan_text else "No previous care plan provided or could not be read."}
-Instructions:
-1. Generate the updated care plan strictly using the exact following format template.
-2. Populate each section of the template based on the patient's information, their *latest feedback/symptoms*, and integrate relevant, SAFE, and appropriate elements from the previous plan if they are still applicable and helpful given the feedback.
-3. Prioritize addressing the issues raised in the patient feedback.
-4. Be specific, actionable, and realistic in your recommendations (e.g., specify *what* activities, *when* to take medications and *what" medications and diet to follow).
-5. Ensure the language is clear, empathetic, and easy for a patient or caregiver to understand. Avoid overly technical jargon where possible.
-7. Do NOT include any introductory phrases (like "Here is the updated plan:", "Based on your feedback...") or concluding sentences outside the plan structure. Provide ONLY the structured content that fits within the template.
-8. If the previous care plan was unavailable or unreadable, create the plan based solely on the patient information and feedback, still following the template.
-9. Ensure the plan is medically sound and reflects standard care principles. also give specific medications rexomemndations based on patient symptoms (ex : paracetamol for fever etc)
-10. If the feedback indicates significant improvement, the plan should reflect this (e.g., adjusting activity levels up, noting successful symptom management) while still including monitoring and red flags.
-11. Review the feedback and previous plan carefully to determine the most likely current STATUS (e.g., Emergency, Deteriorating, Improving, Stable) and ensure the plan content aligns with that status, especially the RED FLAGS section. ensure plan does not exist whatsapp message character limit
-12. kindly, do not produce emergency contact section drop it okay,also in any section dont give like not provided, not known,need to be added like stufff, cosnider youself super intellgient and real physician
-Care Plan Format Template (hey only give in text format do not use tbales okay ,a lso ensure that plan is short and no section great than 4 sentences okay):
-{care_plan_format}
-also ensure that do not give the follwong section :
-Emergency Contacts:
-Primary Caregiver: (Not provided)
-Doctor's Name: (Not provided)
-Hospital Contact: (Not provided) (it is useless abd vague okay and also in any section do not say like doctor or caretaker will provide)
+
+--- Instructions for Generating the New Care Plan ---
+1. Generate the care plan using ONLY the exact section headings and sub-sections listed in the "Required Care Plan Structure" below. Maintain the order.
+2. Fill in the details for each section based on the provided information. Prioritize addressing the latest feedback while incorporating relevant, SAFE, and appropriate details from the previous plan.
+3. Be specific, actionable, and tailored to the patient's situation as described.
+4. Keep the content concise. Aim for a few sentences or bullet points per sub-section (e.g., Morning, Afternoon, Night) and per main section (e.g., ASSESSMENT, MEDICATIONS, DIET, EXERCISE, SYMPTOM MANAGEMENT). Do NOT exceed 4 sentences/points per section/sub-section if possible, but prioritize clarity and necessary instructions.
+5. Regarding MEDICATIONS:
+    - If the previous plan listed specific prescribed medications, include them with dosage and timing if possible.
+    - If specific medications are not available in the previous plan, list the *types* of medications commonly used for the patient's condition (if implied by the feedback/previous plan, e.g., "Blood pressure medication", "Pain relief").
+    - You MAY mention common, over-the-counter examples for specific symptoms if the feedback implies them (e.g., "Paracetamol for fever or mild pain"), but *always* emphasize consulting a doctor for specific prescriptions and dosage. Avoid recommending prescription drugs you are not explicitly told about.
+    - *Crucially, do NOT make up prescription drug names or dosages.*
+    - *Crucially, do NOT use phrases like "Medications to be determined by physician", "Doctor will provide", "Not provided", "(Unknown)", etc.* List what is known or infer likely categories.
+6. For PHYSICAL ACTIVITY AND EXERCISE, provide concrete examples or recommendations suitable for someone with the patient's condition and current status (e.g., "Gentle walking 15 mins daily", "Light stretching exercises", "Avoid strenuous activity"). Tailor this to the patient's feedback (e.g., if deteriorating, recommend rest).
+7. Do NOT include sections like "Red Flags", "Emergency Contacts", "Contact Number", "Doctor's Name", "Hospital Contact", or "Primary Caregiver".
+8. Do NOT use any vague placeholders like "(To be added)", "(Not provided)", or "(Unknown)" anywhere in the plan. If information is truly unavailable and you cannot infer a reasonable default (like suggesting a type of medication), omit the specific detail rather than using a placeholder.
+9. Do NOT include any introductory phrases (e.g., "Here is the plan", "Based on your feedback") or concluding remarks outside the plan structure. Provide ONLY the structured content.
+10. Ensure the plan's tone and content are appropriate for the patient's determined status (Deteriorating, Improving, Stable). For deteriorating status, emphasize caution and monitoring. For improving, suggest gradual progression if appropriate.
+11. Review the feedback and previous plan carefully to determine the most likely current STATUS (e.g., Emergency, Deteriorating, Improving, Stable) and ensure the plan content aligns with that status.
+
+--- Required Care Plan Structure (Use ONLY these exact sections and sub-sections in this order) ---
+{required_care_plan_structure}
 """
+            # --- END OF MODIFIED AI PROMPT ---
+
+
             logger.info("Sending prompt to AI model...")
 
             try:
@@ -809,22 +805,18 @@ Hospital Contact: (Not provided) (it is useless abd vague okay and also in any s
                 generated_plan_text = response.text.strip()
 
                 # Remove markdown code block formatting if present
-                if generated_plan_text.startswith('```') and generated_plan_text.endswith('```'):
-                    # Find the first newline after ``` to potentially strip language name
+                if generated_plan_text.startswith('```'):
                     first_newline_after_code = generated_plan_text.find('\n')
                     if first_newline_after_code != -1:
-                        # Check if there's a language name before the newline
-                        potential_lang = generated_plan_text[3:first_newline_after_code].strip()
-                        if re.match(r'^[a-zA-Z0-9]+$', potential_lang): # Simple check for language name
-                             generated_plan_text = generated_plan_text[first_newline_after_code:].strip()
-                        else:
-                             # No language name or unexpected format, just strip ```
-                             generated_plan_text = generated_plan_text[3:].strip()
+                         # Check if there's a language name before the newline
+                         potential_lang = generated_plan_text[3:first_newline_after_code].strip()
+                         if re.match(r'^[a-zA-Z0-9]+$', potential_lang): # Simple check for language name
+                              generated_plan_text = generated_plan_text[first_newline_after_code:].strip()
+                         else:
+                              generated_plan_text = generated_plan_text[3:].strip()
                     else:
-                         # Handle case where ``` is on the last line without newline
-                         generated_plan_text = generated_plan_text[3:].strip()
+                          generated_plan_text = generated_plan_text[3:].strip()
 
-                    # Strip ending ```
                     if generated_plan_text.endswith('```'):
                         generated_plan_text = generated_plan_text[:-3].strip()
 
@@ -832,8 +824,6 @@ Hospital Contact: (Not provided) (it is useless abd vague okay and also in any s
                 logger.info(f"AI Response received. Length: {len(generated_plan_text)}")
 
                 # Re-determine the final status using the generated plan as well.
-                # This is important because the AI might infer severity the keyword matching missed initially,
-                # or the generated plan text itself might contain explicit strong status indicators.
                 final_status_to_save = determine_patient_status(care_plan_text, generated_plan_text, feedback)
                 logger.info(f"Final status determined after AI generation: {final_status_to_save}")
 
@@ -872,6 +862,7 @@ Hospital Contact: (Not provided) (it is useless abd vague okay and also in any s
 
 
         # Create and store patient record in the database
+        # Ensure original_plan is stored even if it contains error messages from PDF extraction
         new_patient = Patient(
             name=name,
             age=age,
@@ -889,15 +880,12 @@ Hospital Contact: (Not provided) (it is useless abd vague okay and also in any s
         logger.info(f"Patient {patient_id} added to DB with status: {final_status_to_save}.")
 
         # Generate PDF for downloading using the stored data
-        # Note: We pass the patient object directly to the PDF generator for simplicity
-        # and to include feedback in the PDF
         pdf_buffer = generate_care_plan_pdf(new_patient.to_dict(), new_patient.updated_plan, new_patient.status)
         pdf_buffer.seek(0) # Ensure buffer is at the start before base64 encoding
         pdf_base64 = base64.b64encode(pdf_buffer.getvalue()).decode('utf-8')
         logger.info("PDF generated and base64 encoded.")
 
         # Send care plan via WhatsApp (using the final saved data)
-        # Note: We pass the patient object directly to the WhatsApp function
         whatsapp_sent, whatsapp_message = send_whatsapp_care_plan(new_patient.to_dict(), new_patient.updated_plan, new_patient.status)
         logger.info(f"WhatsApp message attempt sent: {whatsapp_sent}, message: {whatsapp_message}")
 
@@ -910,7 +898,8 @@ Hospital Contact: (Not provided) (it is useless abd vague okay and also in any s
             'status': new_patient.status, # Return the final determined status
             'whatsapp_sent': whatsapp_sent,
             'whatsapp_message': whatsapp_message,
-            'ai_error': not ai_enabled or (ai_error_message is not None) # Indicate if AI was not enabled or failed
+            'ai_error': not ai_enabled or (ai_error_message is not None), # Indicate if AI was not enabled or failed
+            'ai_error_message': ai_error_message # Include the specific error message if applicable
         })
 
     except Exception as e:
@@ -920,6 +909,9 @@ Hospital Contact: (Not provided) (it is useless abd vague okay and also in any s
             'success': False,
             'error': f'An unexpected server error occurred: {str(e)}'
         }), 500
+# --- END: Modified submit_feedback function ---
+
+
 
 # --- New routes for Doctor Dashboard actions ---