Update app.py

app.py

@@ -72,41 +72,24 @@ model = genai.GenerativeModel('gemini-1.5-flash',
 st.set_page_config(page_title="Mariam - Assistant IA", page_icon="🤖")
 st.title("Mariam AI - Chat Intelligent")
 
-# Initialiser l'historique de chat
-if "chat" not in st.session_state:
-    st.session_state.chat = model.start_chat(history=[])
-
-# CSS personnalisé pour le conteneur d'upload
+# CSS personnalisé 
 st.markdown("""
     <style>
-    .stButton>button {
-        height: 40px;
-        padding: 0 10px;
+    .stFileUploader {
+        margin-bottom: 10px;
     }
     .upload-container {
         display: flex;
         align-items: center;
         gap: 10px;
-        margin-bottom: 10px;
-    }
-    .custom-file-upload {
-        display: inline-block;
-        cursor: pointer;
-    }
-    .input-container {
-        display: flex; /* Utiliser flexbox pour aligner les éléments */
-        align-items: center; /* Aligner verticalement les éléments au centre */
-        gap: 10px; /* Ajouter un espace entre les éléments */
-    }
-    .input-container > div {
-        flex-grow: 1; /* Permettre à la zone de saisie de prendre l'espace disponible */
-    }
-    .input-container .st-chat-input {
-        flex-grow: 1; /* Permettre à la zone de saisie de prendre l'espace disponible */
     }
     </style>
 """, unsafe_allow_html=True)
 
+# Initialiser l'historique de chat
+if "chat" not in st.session_state:
+    st.session_state.chat = model.start_chat(history=[])
+
 # Afficher l'historique des messages
 for message in st.session_state.chat.history:
     with st.chat_message(role_to_streamlit(message.role)):
@@ -120,68 +103,38 @@ for message in st.session_state.chat.history:
 input_container = st.container()
 
 with input_container:
-    # Utiliser une seule ligne pour la zone de saisie et les boutons
-    prompt = st.chat_input("Que puis-je faire pour vous ?", key="chat_input")
-
-    # Conteneur pour les icônes d'upload
-    upload_icons_container = st.container()
-    with upload_icons_container:
-        col1, col2 = st.columns(2)  # Colonnes pour les icônes d'upload
-
-        with col1:
-            uploaded_files = st.file_uploader("", type=["txt", "mp4", "mp3", "pdf"],
-                                                accept_multiple_files=True, key="files",
-                                                label_visibility="collapsed")
-            st.markdown("📁", unsafe_allow_html=True)
-
-        with col2:
-            uploaded_images = st.file_uploader("", type=["jpg", "jpeg", "png", "gif"],
-                                                 accept_multiple_files=True, key="images",
-                                                 label_visibility="collapsed")
-            st.markdown("📸", unsafe_allow_html=True)
-
-# Appliquer le style au conteneur
-st.markdown(f"""
-    <style>
-        div[data-testid='stVerticalBlock']:has(div.uploadedFile) {{
-            flex-direction: row !important;
-            align-items: center;
-            gap: 10px;
-        }}
-        .stVerticalBlock {{
-            gap: 0rem;
-        }}
-        div[data-testid='stHorizontalBlock'] {{
-            align-items: center;
-        }}
-        div[data-testid='stTextInput'] {{
-            padding-bottom: 0;
-        }}
-    </style>
-""", unsafe_allow_html=True)
-            
+    col1, col2 = st.columns([0.85, 0.15])
+    
+    with col1:
+        prompt = st.chat_input("Que puis-je faire pour vous ?")
+    
+    with col2:
+        # Upload unique pour tous les types de fichiers
+        uploaded_files = st.file_uploader("📁", 
+                                          type=["txt","mp4","mp3","pdf", "jpg", "jpeg", "png", "gif"], 
+                                          accept_multiple_files=True)
+
 if prompt:
     content = [prompt]
     temp_files = []
     
     try:
-        # Traitement des images
-        if uploaded_images:
-            for img_file in uploaded_images:
-                if allowed_file(img_file.name):
-                    image = Image.open(img_file)
-                    content.append(image)
-                    st.chat_message("user").image(image)
-
-        # Traitement des autres fichiers
+        # Traitement des fichiers
         if uploaded_files:
             for file in uploaded_files:
                 if allowed_file(file.name):
-                    with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.name)[1]) as temp_file:
-                        temp_file.write(file.getvalue())
-                        temp_files.append(temp_file.name)
-                        uploaded_file = upload_and_process_file(temp_file.name)
-                        content.append(uploaded_file)
+                    # Si c'est une image
+                    if file.type.startswith('image/'):
+                        image = Image.open(file)
+                        content.append(image)
+                        st.chat_message("user").image(image)
+                    else:
+                        # Pour les autres types de fichiers
+                        with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.name)[1]) as temp_file:
+                            temp_file.write(file.getvalue())
+                            temp_files.append(temp_file.name)
+                            uploaded_file = upload_and_process_file(temp_file.name)
+                            content.append(uploaded_file)
 
         # Afficher le message utilisateur
         st.chat_message("user").markdown(prompt)