Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,6 +4,10 @@ import os
|
|
| 4 |
from dotenv import load_dotenv
|
| 5 |
|
| 6 |
load_dotenv()
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
# Configure the API key
|
| 8 |
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
| 9 |
|
|
@@ -14,24 +18,8 @@ safety_settings = [
|
|
| 14 |
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
|
| 15 |
]
|
| 16 |
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
Crée un modèle Gemini avec ou sans recherche Google
|
| 20 |
-
|
| 21 |
-
Args:
|
| 22 |
-
use_search (bool): Active ou désactive la recherche Google
|
| 23 |
-
|
| 24 |
-
Returns:
|
| 25 |
-
genai.GenerativeModel: Modèle Gemini configuré
|
| 26 |
-
"""
|
| 27 |
-
tools = "google_search_retrieval" if use_search else None
|
| 28 |
-
|
| 29 |
-
return genai.GenerativeModel(
|
| 30 |
-
'gemini-1.5-flash',
|
| 31 |
-
safety_settings=safety_settings,
|
| 32 |
-
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",
|
| 33 |
-
tools=tools
|
| 34 |
-
)
|
| 35 |
|
| 36 |
def role_to_streamlit(role):
|
| 37 |
if role == "model":
|
|
@@ -39,23 +27,6 @@ def role_to_streamlit(role):
|
|
| 39 |
else:
|
| 40 |
return role
|
| 41 |
|
| 42 |
-
# Ajouter un radiobox pour la recherche Google
|
| 43 |
-
st.sidebar.header("Paramètres de recherche")
|
| 44 |
-
use_google_search = st.sidebar.radio(
|
| 45 |
-
"Recherche Google",
|
| 46 |
-
options=[
|
| 47 |
-
"Désactivée",
|
| 48 |
-
"Activée"
|
| 49 |
-
],
|
| 50 |
-
index=0 # Par défaut, recherche désactivée
|
| 51 |
-
)
|
| 52 |
-
|
| 53 |
-
# Convertir le radiobox en booléen
|
| 54 |
-
search_enabled = use_google_search == "Activée"
|
| 55 |
-
|
| 56 |
-
# Créer le modèle initial avec les paramètres du radiobox
|
| 57 |
-
model = create_model_with_search(search_enabled)
|
| 58 |
-
|
| 59 |
# Add a Gemini Chat history object to Streamlit session state
|
| 60 |
if "chat" not in st.session_state:
|
| 61 |
st.session_state.chat = model.start_chat(history=[])
|
|
@@ -88,12 +59,6 @@ def process_uploaded_file(file):
|
|
| 88 |
|
| 89 |
# Accept user's next message, add to context, resubmit context to Gemini
|
| 90 |
if prompt := st.chat_input("Hey?"):
|
| 91 |
-
# Recréer le modèle avec les paramètres de recherche actuels
|
| 92 |
-
model = create_model_with_search(search_enabled)
|
| 93 |
-
|
| 94 |
-
# Réinitialiser la conversation avec le nouveau modèle
|
| 95 |
-
st.session_state.chat = model.start_chat(history=st.session_state.chat.history)
|
| 96 |
-
|
| 97 |
# Process any uploaded file
|
| 98 |
uploaded_gemini_file = None
|
| 99 |
if uploaded_file:
|
|
|
|
| 4 |
from dotenv import load_dotenv
|
| 5 |
|
| 6 |
load_dotenv()
|
| 7 |
+
|
| 8 |
+
|
| 9 |
+
tools = "google_search_retrieval"
|
| 10 |
+
|
| 11 |
# Configure the API key
|
| 12 |
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
| 13 |
|
|
|
|
| 18 |
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
|
| 19 |
]
|
| 20 |
|
| 21 |
+
model = genai.GenerativeModel('gemini-2.0-flash-exp',
|
| 22 |
+
safety_settings=safety_settings,tools=tools, 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")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
def role_to_streamlit(role):
|
| 25 |
if role == "model":
|
|
|
|
| 27 |
else:
|
| 28 |
return role
|
| 29 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
# Add a Gemini Chat history object to Streamlit session state
|
| 31 |
if "chat" not in st.session_state:
|
| 32 |
st.session_state.chat = model.start_chat(history=[])
|
|
|
|
| 59 |
|
| 60 |
# Accept user's next message, add to context, resubmit context to Gemini
|
| 61 |
if prompt := st.chat_input("Hey?"):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 62 |
# Process any uploaded file
|
| 63 |
uploaded_gemini_file = None
|
| 64 |
if uploaded_file:
|