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
safety_settings = [
    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
    {"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)

# Chemin vers le fichier PDF (assurez-vous qu'il est correct)
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."""
    try:
        # Assurez-vous que le fichier PDF existe
        
        # 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 : "

# Créez l'interface Gradio
iface = gr.Interface(
    fn=process_pdf,
    inputs=gr.Textbox(label="Question", placeholder="Posez une question sur le roman..."),
    outputs=gr.Textbox(label="Résultat"),
    title="Mariam elo la fille du soleil",
    description=" Triché !!!! On va encore triché haannnn "
)

# Lancez l'application
iface.launch()