Update app.py

app.py

@@ -132,14 +132,18 @@ def pdf_to_images(pdf_path, dpi=300, output_format='JPEG'):
 # -- START -- set up run variables
 
 system_msg = """
-You are kheops an AI assistant,you an accountant expert powered by kheops Team with computer vision.
-AI knowledge cutoff: April 2024
-
-Built-in vision capabilities:
-- extract text from image
-- describe images
-- analyze image contents
-- logical problem-solving requiring machine vision
+You are a data extraction assistant specialized in French real estate documents. Your task is to extract key information and ensure the validity of the data according to the following rules:
+1. Extract the following values: "Total_a_payer", "Fond_travaux_alur", "Part_charges_previsionnelles", and "Part_autres_travaux".
+2. Ensure that "Total_a_payer" is equal to the sum of "Fond_travaux_alur", "Part_charges_previsionnelles", and "Part_autres_travaux". If the amounts do not match, return a JSON with an error message.
+3. Additionally, extract the property owner’s name, address or lot number, and the period of the document.
+4. Return the extracted data in JSON format.
+
+### Validation Rule:
+- Total_a_payer = Fond_travaux_alur + Part_charges_previsionnelles + Part_autres_travaux.
+
+### Expected output:
+- A JSON containing all the extracted values if the validation passes.
+- A JSON with an error message if the validation fails.
 """.strip()
 
 
@@ -147,7 +151,6 @@ Built-in vision capabilities:
 Sachant que Total à payer doit etre egal à Fond travaux alur + Part charges prévisionnelles+  Part autres travaux - le solde précédent"""
 # The user message
 user_msg = """
-Sachant que Total à payer = Fond travaux alur + Part charges prévisionnelles+  Part autres travaux - le solde précédent
 fournit les informations suivante sous format json uniquement:
 -Total à payer 
 -Fond travaux loi alur et non pas le fond de participation
@@ -161,10 +164,6 @@ fournit les informations suivante sous format json uniquement:
 -la reference:
 - date du document
 - date limit du payement
-information : des enterprise de gestion immo envoi ces appel de fond sont envoyer a l'entreprise de gestion immobiliére KALIZ GESTION et ton role c'est un agent qui extrait les données et tu me donnent un fichier json a la fin  
-attention : 
-- Fond travaux loi alur et non pas le fond de participation
-- Total à payer a debité et non pas credit
 """.strip()