Update app.py

app.py

@@ -2,12 +2,11 @@ import streamlit as st
 import google.generativeai as genai
 import os
 from dotenv import load_dotenv
+import http.client
+import json
 
 load_dotenv()
 
-
-tools = "google_search_retrieval"
-
 # Configure the API key
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
@@ -18,8 +17,57 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
-model = genai.GenerativeModel('gemini-2.0-flash-exp',tools='code_execution' ,
-                               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")
+model = genai.GenerativeModel('gemini-2.0-flash-exp', tools='code_execution',
+                            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 perform_web_search(query):
+    conn = http.client.HTTPSConnection("google.serper.dev")
+    payload = json.dumps({"q": query})
+    headers = {
+        'X-API-KEY': '9b90a274d9e704ff5b21c0367f9ae1161779b573',
+        'Content-Type': 'application/json'
+    }
+    try:
+        conn.request("POST", "/search", payload, headers)
+        res = conn.getresponse()
+        data = json.loads(res.read().decode("utf-8"))
+        return data
+    except Exception as e:
+        st.error(f"Erreur lors de la recherche web : {e}")
+        return None
+    finally:
+        conn.close()
+
+def format_search_results(data):
+    if not data:
+        return "Aucun résultat trouvé"
+    
+    result = ""
+    
+    # Knowledge Graph
+    if 'knowledgeGraph' in data:
+        kg = data['knowledgeGraph']
+        result += f"### {kg.get('title', '')}\n"
+        result += f"*{kg.get('type', '')}*\n\n"
+        result += f"{kg.get('description', '')}\n\n"
+    
+    # Organic Results
+    if 'organic' in data:
+        result += "### Résultats principaux:\n"
+        for item in data['organic'][:3]:  # Limit to top 3 results
+            result += f"- **{item['title']}**\n"
+            result += f"  {item['snippet']}\n"
+            result += f"  [Lien]({item['link']})\n\n"
+    
+    # People Also Ask
+    if 'peopleAlsoAsk' in data:
+        result += "### Questions fréquentes:\n"
+        for item in data['peopleAlsoAsk'][:2]:  # Limit to top 2 questions
+            result += f"- **{item['question']}**\n"
+            result += f"  {item['snippet']}\n\n"
+    
+    return result
 
 def role_to_streamlit(role):
     if role == "model":
@@ -27,17 +75,24 @@ def role_to_streamlit(role):
     else:
         return role
 
-# Add a Gemini Chat history object to Streamlit session state
+# Add chat and settings to session state
 if "chat" not in st.session_state:
     st.session_state.chat = model.start_chat(history=[])
+if "web_search" not in st.session_state:
+    st.session_state.web_search = False
 
 # Display Form Title
 st.title("Mariam AI!")
 
+# Settings section
+with st.sidebar:
+    st.title("Paramètres")
+    st.session_state.web_search = st.toggle("Activer la recherche web", value=st.session_state.web_search)
+
 # File upload section
-uploaded_file = st.file_uploader("Télécharger un fichier (image/document)", type=['jpg','mp4','mp3', 'jpeg', 'png', 'pdf', 'txt'])
+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
+# Display chat messages
 for message in st.session_state.chat.history:
     with st.chat_message(role_to_streamlit(message.role)):
         st.markdown(message.parts[0].text)
@@ -45,11 +100,8 @@ for message in st.session_state.chat.history:
 # 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
@@ -57,33 +109,37 @@ def process_uploaded_file(file):
             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
+# Chat input and processing
 if prompt := st.chat_input("Hey?"):
-    # Process any uploaded file
     uploaded_gemini_file = None
     if uploaded_file:
         uploaded_gemini_file = process_uploaded_file(uploaded_file)
     
-    # Display user's last message
+    # Display user message
     st.chat_message("user").markdown(prompt)
-
-    # Send user entry to Gemini with optional file
+    
     try:
+        # Perform web search if enabled
+        web_results = None
+        if st.session_state.web_search:
+            with st.spinner("Recherche web en cours..."):
+                web_results = perform_web_search(prompt)
+                if web_results:
+                    formatted_results = format_search_results(web_results)
+                    prompt = f"""Question: {prompt}\n\nRésultats de recherche web:\n{formatted_results}\n\nPourrais-tu analyser ces informations et me donner une réponse complète?"""
+        
+        # Send message to Gemini
         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
-            print(prompt)
-            print("---------")
             response = st.session_state.chat.send_message(prompt)
-            print(response.text)
-        # Display last response
+        
+        # Display assistant 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
+# Create temp directory
 os.makedirs("temp", exist_ok=True)