app.py

import streamlit as st
import google.generativeai as genai
import os
from dotenv import load_dotenv

load_dotenv()
# Configure the API key
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

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"},
]

def create_model_with_search(use_search):
    """
    Crée un modèle Gemini avec ou sans recherche Google
    
    Args:
        use_search (bool): Active ou désactive la recherche Google
    
    Returns:
        genai.GenerativeModel: Modèle Gemini configuré
    """
    tools = "google_search_retrieval" if use_search else None
    
    return genai.GenerativeModel(
        'gemini-1.5-flash', 
        safety_settings=safety_settings,
        system_instruction="Tu es un assistant intelligent. ton but est d'assister au mieux que tu peux. tu as été créé par Aenir et tu t'appelles Mariam",
        tools=tools
    )

def role_to_streamlit(role):
    if role == "model":
        return "assistant"
    else:
        return role

# Ajouter un radiobox pour la recherche Google
st.sidebar.header("Paramètres de recherche")
use_google_search = st.sidebar.radio(
    "Recherche Google", 
    options=[
        "Désactivée", 
        "Activée"
    ],
    index=0  # Par défaut, recherche désactivée
)

# Convertir le radiobox en booléen
search_enabled = use_google_search == "Activée"

# Créer le modèle initial avec les paramètres du radiobox
model = create_model_with_search(search_enabled)

# Add a Gemini Chat history object to Streamlit session state
if "chat" not in st.session_state:
    st.session_state.chat = model.start_chat(history=[])

# Display Form Title
st.title("Mariam AI!")

# File upload section
uploaded_file = st.file_uploader("Télécharger un fichier (image/document)", type=['jpg','mp4','mp3', 'jpeg', 'png', 'pdf', 'txt'])

# Display chat messages from history above current input box
for message in st.session_state.chat.history:
    with st.chat_message(role_to_streamlit(message.role)):
        st.markdown(message.parts[0].text)

# Function to handle file upload with Gemini
def process_uploaded_file(file):
    if file is not None:
        # Save the uploaded file temporarily
        with open(os.path.join("temp", file.name), "wb") as f:
            f.write(file.getbuffer())
        
        # Upload the file to Gemini
        try:
            gemini_file = genai.upload_file(os.path.join("temp", file.name))
            return gemini_file
        except Exception as e:
            st.error(f"Erreur lors du téléchargement du fichier : {e}")
            return None

# Accept user's next message, add to context, resubmit context to Gemini
if prompt := st.chat_input("Hey?"):
    # Recréer le modèle avec les paramètres de recherche actuels
    model = create_model_with_search(search_enabled)
    
    # Réinitialiser la conversation avec le nouveau modèle
    st.session_state.chat = model.start_chat(history=st.session_state.chat.history)
    
    # Process any uploaded file
    uploaded_gemini_file = None
    if uploaded_file:
        uploaded_gemini_file = process_uploaded_file(uploaded_file)
    
    # Display user's last message
    st.chat_message("user").markdown(prompt)

    # Send user entry to Gemini with optional file
    try:
        if uploaded_gemini_file:
            # If a file is uploaded, include it in the context
            response = st.session_state.chat.send_message([uploaded_gemini_file, "\n\n", prompt])
        else:
            # Normal text-only conversation
            response = st.session_state.chat.send_message(prompt)

        # Display last response
        with st.chat_message("assistant"):
            st.markdown(response.text)
    
    except Exception as e:
        st.error(f"Erreur lors de l'envoi du message : {e}")

# Create temp directory if it doesn't exist
os.makedirs("temp", exist_ok=True)