Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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 |
-
|
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}"
|