Docfile commited on
Commit
0438389
·
verified ·
1 Parent(s): 2848170

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -27
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
- convo = model.start_chat(history=[
73
- {
74
- "role": "user",
75
- "parts": extract_pdf_pages(code_civil)
76
- },
77
- {
78
- "role": "user",
79
- "parts": extract_pdf_pages(code_civil_II)
80
- },
81
- {
82
- "role": "user",
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)