Spaces:
Sleeping
Sleeping
File size: 3,045 Bytes
d5a6a33 1db4ff7 188c1ca 2ca54a6 331c814 586d97d b16530f 1db4ff7 ee96e70 1db4ff7 436aefa 1db4ff7 436aefa 586d97d 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 |
import streamlit as st
import google.generativeai as genai
import os
from dotenv import load_dotenv
load_dotenv()
# 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-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")
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', '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) |