Update app.py

app.py

@@ -72,10 +72,6 @@ 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
 st.markdown("""
     <style>
@@ -89,13 +85,24 @@ st.markdown("""
         gap: 10px;
         margin-bottom: 10px;
     }
-    .custom-file-upload {
-        display: inline-block;
-        cursor: pointer;
+    .icon-container {
+        display: flex;
+        align-items: center;
+        gap: 10px;
+    }
+    .stFileUploader {
+        margin-bottom: 0;
+    }
+    .stFileUploader > div {
+        margin-bottom: 0 !important;
     }
     </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)):
@@ -109,22 +116,26 @@ for message in st.session_state.chat.history:
 input_container = st.container()
 
 with input_container:
-    col1, col2, col3 = st.columns([8, 1, 1])
-    
-    with col3:
-        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)
+    col1, col2 = st.columns([0.9, 0.1])
     
     with col1:
-        prompt = st.chat_input("Que puis-je faire pour vous ?")
+        # Conteneur pour l'input et les icônes
+        input_col1, input_col2, input_col3 = st.columns([0.85, 0.075, 0.075])
+        
+        with input_col1:
+            prompt = st.chat_input("Que puis-je faire pour vous ?")
+        
+        with input_col2:
+            uploaded_images = st.file_uploader("", type=["jpg", "jpeg", "png", "gif"], 
+                                             accept_multiple_files=True, key="images",
+                                             label_visibility="collapsed")
+            st.markdown('<div class="icon-container">📸</div>', unsafe_allow_html=True)
+        
+        with input_col3:
+            uploaded_files = st.file_uploader("", type=["txt","mp4","mp3","pdf"], 
+                                            accept_multiple_files=True, key="files", 
+                                            label_visibility="collapsed")
+            st.markdown('<div class="icon-container">📁</div>', unsafe_allow_html=True)
 
 if prompt:
     content = [prompt]