Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 import google.generativeai as genai
 import os
 
-# Configurez l'API Gemini
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])  # Remplacez "VOTRE_CLÉ_API" par votre clé API Gemini
+# Configuration de l'API Gemini
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 safety_settings = [
     {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
     {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
@@ -11,39 +11,46 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
-# Définissez le modèle
-model = genai.GenerativeModel("gemini-2.0-flash-exp",safety_settings=safety_settings)
+# Définition du modèle
+model = genai.GenerativeModel("gemini-2.0-flash-exp", safety_settings=safety_settings)
 
-# Chemin vers le fichier PDF (assurez-vous qu'il est correct)
+# Chemin vers le fichier PDF
 PDF_FILE_PATH = "okoumba-nkoghe-elo_compress.pdf"
-
 sample_file = genai.upload_file(PDF_FILE_PATH)
-def process_pdf(prompt):
-    """Traite le fichier PDF local et génère une réponse basée sur l'invite."""
+
+def process_input(prompt, image):
+    """Traite le texte et l'image optionnelle et génère une réponse."""
     try:
-        # Assurez-vous que le fichier PDF existe
+        content = [sample_file, "\n\n", prompt + " répond en français."]
+        
+        # Ajout de l'image au contenu si elle est fournie
+        if image is not None:
+            img_file = genai.upload_file(image)
+            content.insert(1, img_file)
+        
+        # Génération du contenu
+        result = model.generate_content(content)
         
-        # Générez le contenu à l'aide du modèle Gemini
-        result = model.generate_content([
-            sample_file,
-            "\n\n",
-            prompt + " répond en français."
-        ])
-
         print(prompt)
-        #genai.delete_file(sample_file.name)
         return result.text
     except Exception as e:
-        return "Une erreur s'est produite : "
+        return f"Une erreur s'est produite : {str(e)}"
 
-# Créez l'interface Gradio
+# Création de l'interface Gradio
 iface = gr.Interface(
-    fn=process_pdf,
-    inputs=gr.Textbox(label="Question", placeholder="Posez une question sur le roman...",lines=4),
-    outputs=gr.Textbox(label="Résultat"),
+    fn=process_input,
+    inputs=[
+        gr.Textbox(label="Question", placeholder="Posez une question sur le roman...", lines=4),
+        gr.Image(label="Image (facultative)", type="filepath")
+    ],
+    outputs=gr.Textbox(label="Résultat"),
     title="Mariam elo la fille du soleil",
-    description=" Triché !!!! On va encore triché haannnn "
+    description="Triché !!!! On va encore triché haannnn",
+    examples=[
+        ["Qui est le personnage principal ?", None],
+        ["Résume moi l'histoire", None]
+    ]
 )
 
-# Lancez l'application
+# Lancement de l'application
 iface.launch()