Spaces:
Sleeping
Sleeping
File size: 3,087 Bytes
d5a6a33 9b159d4 1db4ff7 188c1ca 2ca54a6 331c814 9b159d4 b16530f 1db4ff7 ee96e70 1db4ff7 436aefa 1db4ff7 436aefa 586d97d ff703b4 5ae354b 1db4ff7 d5a6a33 586d97d 1db4ff7 586d97d 1db4ff7 2ca54a6 586d97d b16530f 586d97d b16530f 586d97d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
import streamlit as st
import google.generativeai as genai
import os
from dotenv import load_dotenv
load_dotenv()
tools = "google_search_retrieval"
# 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"},
]
model = genai.GenerativeModel('gemini-2.0-flash-exp',
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")
def role_to_streamlit(role):
if role == "model":
return "assistant"
else:
return role
# 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?"):
# 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) |