Create app.py

app.py

@@ -0,0 +1,41 @@
+import gradio as gr
+import google.generativeai as genai
+import os
+
+# Configurez l'API Gemini
+genai.configure(api_key="VOTRE_CLÉ_API")  # Remplacez "VOTRE_CLÉ_API" par votre clé API Gemini
+
+# Définissez le modèle
+model = genai.GenerativeModel("gemini-1.5-flash")
+
+# Chemin vers le fichier PDF (assurez-vous qu'il est correct)
+PDF_FILE_PATH = "livre.pdf"
+
+def process_pdf(prompt):
+    """Traite le fichier PDF local et génère une réponse basée sur l'invite."""
+    try:
+        # Assurez-vous que le fichier PDF existe
+        if not os.path.exists(PDF_FILE_PATH):
+            raise FileNotFoundError(f"Le fichier PDF n'a pas été trouvé : {PDF_FILE_PATH}")
+
+        # Générez le contenu à l'aide du modèle Gemini
+        result = model.generate_content([
+            genai.upload_file(PDF_FILE_PATH),
+            "\n\n",
+            prompt
+        ])
+        return result.text
+    except Exception as e:
+        return f"Une erreur s'est produite : {e}"
+
+# Créez l'interface Gradio
+iface = gr.Interface(
+    fn=process_pdf,
+    inputs=gr.Textbox(label="Prompt", placeholder="Posez une question sur le PDF..."),
+    outputs=gr.Textbox(label="Résultat"),
+    title="Assistant PDF avec Gemini 1.5 Flash",
+    description="Posez des questions sur le fichier livre.pdf déjà présent dans le répertoire.",
+)
+
+# Lancez l'application
+iface.launch()