Update app.py
Browse files
app.py
CHANGED
@@ -36,11 +36,6 @@ model = genai.GenerativeModel(model_name="gemini-1.5-flash-latest",
|
|
36 |
system_instruction=system_instruction, safety_settings=safety_settings)
|
37 |
|
38 |
|
39 |
-
code_civil = "data/CODE_CIVIL_I_-_2023 (1).pdf"
|
40 |
-
code_civil_II = "data/CODE_civil_II_-_2.pdf"
|
41 |
-
Code_de_procedure_pénale = "data/Code_de_proc._pénale-2023.pdf"
|
42 |
-
code_penal = "data/NOUVEAU_CODE_PENAL_2024.pdf"
|
43 |
-
|
44 |
uploaded_files = []
|
45 |
def upload_if_needed(pathname: str) -> list[str]:
|
46 |
path = Path(pathname)
|
@@ -69,28 +64,17 @@ def extract_pdf_pages(pathname: str) -> list[str]:
|
|
69 |
print(f"Erreur: Fichier PDF '{pathname}' introuvable.")
|
70 |
return parts
|
71 |
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
"parts": extract_pdf_pages(
|
80 |
-
|
81 |
-
|
82 |
-
|
83 |
-
"parts": extract_pdf_pages(Code_de_procedure_pénale)
|
84 |
-
},
|
85 |
-
{
|
86 |
-
"role": "user",
|
87 |
-
"parts": extract_pdf_pages(code_penal)
|
88 |
-
},
|
89 |
-
{ "role": "model",
|
90 |
-
"parts": [""]
|
91 |
-
},
|
92 |
-
])
|
93 |
-
|
94 |
|
95 |
def respond(user_input):
|
96 |
convo.send_message(user_input)
|
|
|
36 |
system_instruction=system_instruction, safety_settings=safety_settings)
|
37 |
|
38 |
|
|
|
|
|
|
|
|
|
|
|
39 |
uploaded_files = []
|
40 |
def upload_if_needed(pathname: str) -> list[str]:
|
41 |
path = Path(pathname)
|
|
|
64 |
print(f"Erreur: Fichier PDF '{pathname}' introuvable.")
|
65 |
return parts
|
66 |
|
67 |
+
# Fonction pour charger tous les PDFs du dossier data
|
68 |
+
def load_pdfs_from_data():
|
69 |
+
data_dir = "data"
|
70 |
+
pdf_files = [f for f in os.listdir(data_dir) if f.endswith('.pdf')]
|
71 |
+
convo = model.start_chat()
|
72 |
+
for pdf_file in pdf_files:
|
73 |
+
pdf_path = os.path.join(data_dir, pdf_file)
|
74 |
+
convo.send_message({"role": "user", "parts": extract_pdf_pages(pdf_path)})
|
75 |
+
return convo
|
76 |
+
|
77 |
+
convo = load_pdfs_from_data()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
78 |
|
79 |
def respond(user_input):
|
80 |
convo.send_message(user_input)
|