Update app.py

app.py

@@ -14,9 +14,24 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
-model = genai.GenerativeModel('gemini-2.0-flash-exp', 
-                               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")
+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-2.0-flash-exp', 
+        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":
@@ -24,6 +39,23 @@ def role_to_streamlit(role):
     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=[])
@@ -56,6 +88,12 @@ def process_uploaded_file(file):
 
 # 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: