Update app.py

app.py

@@ -21,6 +21,9 @@ safety_settings = [
     {"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
@@ -73,20 +76,25 @@ st.title("Mariam AI - Chat Intelligent")
 if "chat" not in st.session_state:
     st.session_state.chat = model.start_chat(history=[])
 
-# Créer trois colonnes pour les boutons d'upload
-col1, col2, col3 = st.columns([1, 1, 8])
-
-# Bouton pour les images
-with col1:
-    uploaded_images = st.file_uploader("", type=["jpg", "jpeg", "png", "gif"], 
-                                     accept_multiple_files=True, key="images")
-    st.markdown("📸")
-
-# Bouton pour les autres fichiers
-with col2:
-    uploaded_files = st.file_uploader("", type=["txt", "pdf"], 
-                                    accept_multiple_files=True, key="files")
-    st.markdown("📁")
+# CSS personnalisé pour le conteneur d'upload
+st.markdown("""
+    <style>
+    .stButton>button {
+        height: 40px;
+        padding: 0 10px;
+    }
+    .upload-container {
+        display: flex;
+        align-items: center;
+        gap: 10px;
+        margin-bottom: 10px;
+    }
+    .custom-file-upload {
+        display: inline-block;
+        cursor: pointer;
+    }
+    </style>
+""", unsafe_allow_html=True)
 
 # Afficher l'historique des messages
 for message in st.session_state.chat.history:
@@ -97,8 +105,28 @@ for message in st.session_state.chat.history:
                 if hasattr(part, 'image'):
                     st.image(part.image)
 
-# Zone de saisie du chat
-if prompt := st.chat_input("Que puis-je faire pour vous ?"):
+# Créer un conteneur pour la zone de saisie et les boutons d'upload
+input_container = st.container()
+
+with input_container:
+    col1, col2, col3 = st.columns([8, 1, 1])
+    
+    with col3:
+        uploaded_files = st.file_uploader("", type=["txt", "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)
+    
+    with col1:
+        prompt = st.chat_input("Que puis-je faire pour vous ?")
+
+if prompt:
     content = [prompt]
     temp_files = []