Update app.py

app.py

@@ -14,23 +14,32 @@ safety_settings = [
 # Définition du modèle
 model = genai.GenerativeModel("gemini-2.0-flash-exp", safety_settings=safety_settings)
 
-# Chemin vers le fichier PDF
-PDF_FILE_PATH = "okoumba-nkoghe-elo_compress.pdf"
-sample_file = genai.upload_file(PDF_FILE_PATH)
+# Chargement de tous les fichiers PDF du dossier "data"
+DATA_FOLDER = "data"
+pdf_files = [
+    os.path.join(DATA_FOLDER, file)
+    for file in os.listdir(DATA_FOLDER)
+    if file.lower().endswith(".pdf")
+]
+uploaded_files = [genai.upload_file(pdf_file) for pdf_file in pdf_files]
 
 def process_input(prompt, image):
     """Traite le texte et l'image optionnelle et génère une réponse."""
     try:
-        content = [sample_file, "\n\n", prompt + " répond en français."]
-        
-        # Ajout de l'image au contenu si elle est fournie
+        # Préparer le contenu en ajoutant tous les fichiers PDF
+        content = uploaded_files.copy()
+
+        # Si une image est fournie, l'ajouter au contenu
         if image is not None:
             img_file = genai.upload_file(image)
-            content.insert(1, img_file)
-        
+            content.append(img_file)
+
+        # Ajout du prompt avec l'instruction de répondre en français
+        content.append("\n\n")
+        content.append(prompt + " répond en français.")
+
         # Génération du contenu
         result = model.generate_content(content)
-        
         print(prompt)
         return result.text
     except Exception as e:
@@ -44,13 +53,13 @@ iface = gr.Interface(
         gr.Image(label="Image (facultative)", type="filepath")
     ],
     outputs=gr.Textbox(label="Résultat"),
-    title="Mariam elo la fille du soleil",
+    title="Mariam eco",
     description="Triché !!!! On va encore triché haannnn",
     examples=[
-        ["Qui est le personnage principal ?", None],
-        ["Résume moi l'histoire", None]
+        ["Cest quoi un mass media", None],
+        ["Cest quoi l'entrepreneuriat", None]
     ]
 )
 
 # Lancement de l'application
-iface.launch()
+iface.launch()