Update app.py

app.py

@@ -72,6 +72,10 @@ 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>
@@ -85,24 +89,24 @@ st.markdown("""
         gap: 10px;
         margin-bottom: 10px;
     }
-    .icon-container {
-        display: flex;
-        align-items: center;
-        gap: 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 */
     }
-    .stFileUploader {
-        margin-bottom: 0;
+    .input-container > div {
+        flex-grow: 1; /* Permettre à la zone de saisie de prendre l'espace disponible */
     }
-    .stFileUploader > div {
-        margin-bottom: 0 !important;
+    .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)):
@@ -116,27 +120,46 @@ for message in st.session_state.chat.history:
 input_container = st.container()
 
 with input_container:
-    col1, col2 = st.columns([0.9, 0.1])
-    
-    with col1:
-        # 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)
-
+    # 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)
+            
 if prompt:
     content = [prompt]
     temp_files = []