added policy document ocr display; changed rules.txt

app.py

@@ -108,9 +108,13 @@ def main():
                         (item for item in response if item.get("stage") == "SUGGEST"), None
                     )['response']
                     suggestion = suggestion.split('<POLICY_PITCH>')[-1].split('</POLICY_PITCH>')[0]
+                    text = next(
+                        (item for item in response if item.get("stage") == "OCR"), None
+                    )['response']
                     # Store results
                     all_analyses.append({
                         'name': uploaded_file.name,
+                        'text' : text,
                         'analysis' : analysis,
                         'suggestion' : suggestion
                     })
@@ -122,6 +126,8 @@ def main():
         with tab1:
             for idx, analysis in enumerate(all_analyses):
                 with st.expander(f"### Policy {idx + 1}: {analysis['name']}"):
+                    with st.container():
+                        st.markdown(f'Policy Document : {analysis["text"]}')
                     with st.container():
                         st.markdown(re.sub(r'\<\/?(GOOD|AVERAGE|BAD|FINAL_VERDICT)\>', '', analysis['analysis']))
                     with st.container():

policy_analyser/prompts/health/rules.txt

@@ -1,5 +1,21 @@
 Extract the following data points from the policy document and apply rules on them:
-Parameters : ["Sum Insured", "Adults", "Children", "Insured_Address_City", "Room Rent Limit", "Deductible", "Sublimits", "Copay", "PED Waiting Period", "Thirty Day Waiting Period", "Specific Illness Waiting Period", "Maternity Benefits", "Pre_Post_Hospitalization", "Daycare_Procedures", "Restoration_Benefit", "Preventive_Annual_Health_Checkup", "Ambulance_Cover"]
+- "Sum Insured" : Sum insured in policy
+- "Adults" : no. of adults covered in policy
+- "Children" : no. of children covered in policy
+- "Insured_Address_City" : Insured person city of residence
+- "Room Rent Limit" : limit on hospital room rent
+- "Deductible" : deductible in policy
+- "Sublimits" : Sublimit or amount caps on different diseases
+- "Copay" : copayment
+- "PED Waiting Period" : Waiting period on pre-existing diseases
+- "Thirty Day Waiting Period" : 30-day waiting period
+- "Specific Illness Waiting Period" : waiting period on specific illnesses/diseases
+- "Maternity Benefits" : benefits on maternity illnesses or cases
+- "Pre_Post_Hospitalization" : pre & pos-hospitalization benefits
+- "Daycare_Procedures" : if daycare procedures & treatments are covered
+- "Restoration_Benefit" : if sum insured is restored after it gets exhausted
+- "Preventive_Annual_Health_Checkup" : if free annual health checkup is covered
+- "Ambulance_Cover" : cover on ambulance charges of hospitalization
 
 IF Insured_Address_City in Top_Cities:
     Is_Top_City = true