import streamlit as st import google.generativeai as genai import os from dotenv import load_dotenv from PIL import Image import tempfile import time import ssl # Charger les variables d'environnement load_dotenv() # Configurer la clé API genai.configure(api_key=os.getenv("GOOGLE_API_KEY")) # Paramètres de sécurité 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"}, ] def role_to_streamlit(role): return "assistant" if role == "model" else role def upload_and_process_file(file_path): max_retries = 3 retry_delay = 2 for attempt in range(max_retries): try: if not os.path.exists(file_path): raise FileNotFoundError(f"Le fichier {file_path} n'existe pas") file_size = os.path.getsize(file_path) if file_size == 0: raise ValueError(f"Le fichier {file_path} est vide") uploaded_file = genai.upload_file(path=file_path) timeout = 300 start_time = time.time() while uploaded_file.state.name == "PROCESSING": if time.time() - start_time > timeout: raise TimeoutError("Timeout pendant le traitement du fichier") time.sleep(10) uploaded_file = genai.get_file(uploaded_file.name) if uploaded_file.state.name == "FAILED": raise ValueError(f"Échec du traitement: {uploaded_file.state.name}") return uploaded_file except Exception as e: if attempt < max_retries - 1: time.sleep(retry_delay * (attempt + 1)) else: raise def allowed_file(filename): ALLOWED_EXTENSIONS = {'txt','mp4','mp3','pdf', 'png', 'jpg', 'jpeg', 'gif'} return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS # Initialiser le modèle 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") # Configuration de la page Streamlit st.set_page_config(page_title="Mariam - Assistant IA", page_icon="🤖") st.title("Mariam AI - Chat Intelligent") # CSS personnalisé st.markdown(""" """, unsafe_allow_html=True) # Initialiser l'historique de chat if "chat" not in st.session_state: st.session_state.chat = model.start_chat(history=[]) # Conteneur principal avec marge en bas main_container = st.container() main_container.markdown('