Docfile commited on
Commit
6f5c7c6
·
verified ·
1 Parent(s): 1f49d4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -17,19 +17,19 @@ model = genai.GenerativeModel("gemini-2.0-flash-exp",safety_settings=safety_sett
17
  # Chemin vers le fichier PDF (assurez-vous qu'il est correct)
18
  PDF_FILE_PATH = "okoumba-nkoghe-elo_compress.pdf"
19
 
 
20
  def process_pdf(prompt):
21
  """Traite le fichier PDF local et génère une réponse basée sur l'invite."""
22
  try:
23
  # Assurez-vous que le fichier PDF existe
24
- if not os.path.exists(PDF_FILE_PATH):
25
- raise FileNotFoundError(f"Le fichier PDF n'a pas été trouvé : {PDF_FILE_PATH}")
26
-
27
  # Générez le contenu à l'aide du modèle Gemini
28
  result = model.generate_content([
29
- genai.upload_file(PDF_FILE_PATH),
30
  "\n\n",
31
  prompt
32
  ])
 
33
  return result.text
34
  except Exception as e:
35
  return f"Une erreur s'est produite : {e}"
 
17
  # Chemin vers le fichier PDF (assurez-vous qu'il est correct)
18
  PDF_FILE_PATH = "okoumba-nkoghe-elo_compress.pdf"
19
 
20
+ sample_file = genai.upload_file(PDF_FILE_PATH)
21
  def process_pdf(prompt):
22
  """Traite le fichier PDF local et génère une réponse basée sur l'invite."""
23
  try:
24
  # Assurez-vous que le fichier PDF existe
25
+
 
 
26
  # Générez le contenu à l'aide du modèle Gemini
27
  result = model.generate_content([
28
+ sample_file,
29
  "\n\n",
30
  prompt
31
  ])
32
+ #genai.delete_file(sample_file.name)
33
  return result.text
34
  except Exception as e:
35
  return f"Une erreur s'est produite : {e}"