Update app.py

app.py

@@ -114,16 +114,11 @@ def initialize_gemini():
     except Exception as e:
         raise gr.Error(f"Error initializing Gemini: {str(e)}")
 
-def create_prompt(extracted_text: str) -> str:
-    data_to_extract = {
-        "tribunal": "",
-        "numero_rg": "",
-        "date_ordonnance": "",
-        "demandeurs": [],
-        "defendeurs": [],
-        "avocats_demandeurs": [],
-        "avocats_defendeurs": []
-    }
+def create_prompt(extracted_text: str, path_to_data_to_extract: str) -> str:
+
+    # load data to extract
+    with open(path_to_data_to_extract, 'r', encoding='utf-8') as file:
+        data_to_extract = json.load(file)
 
     prompt = f"""Tu es un assistant juridique expert en analyse de documents judiciaires français.
     Je vais te fournir le contenu d'un document judiciaire extrait d'un PDF.