Update app.py

app.py

@@ -8,8 +8,20 @@ load_dotenv()
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 
-# Initialize the Generative Model
-model = genai.GenerativeModel('gemini-1.5-flash')
+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"},
+]
+
+
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+
+model = 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")
+
+
 
 # Function to get response from the model
 # Gemini uses 'model' for assistant; Streamlit uses 'assistant'
@@ -27,7 +39,7 @@ if "chat" not in st.session_state:
     st.session_state.chat = model.start_chat(history=[])
 
 # Display Form Title
-st.title("Chat with Google Gemini-1.5-flash!")
+st.title("Mariam AI!")
 
 # Display chat messages from history above current input box
 for message in st.session_state.chat.history:
@@ -35,7 +47,7 @@ for message in st.session_state.chat.history:
         st.markdown(message.parts[0].text)
 
 # Accept user's next message, add to context, resubmit context to Gemini
-if prompt := st.chat_input("I possess a well of knowledge. What would you like to know?"):
+if prompt := st.chat_input("Hey?"):
     # Display user's last message
     st.chat_message("user").markdown(prompt)