Update app.py

app.py

@@ -76,10 +76,10 @@ st.title("Mariam AI - Chat Intelligent")
 if "chat" not in st.session_state:
     st.session_state.chat = model.start_chat(history=[])
 
-# CSS personnalisé (amélioré)
+# CSS personnalisé (simplifié)
 st.markdown("""
     <style>
-    /* Conteneur principal pour la zone de saisie et les uploads */
+    /* Conteneur principal pour la zone de saisie et l'upload */
     .input-area {
         display: flex;
         align-items: center;
@@ -92,14 +92,7 @@ st.markdown("""
         flex-grow: 1; /* Permet à la zone de saisie de prendre l'espace disponible */
     }
 
-    /* Style pour les conteneurs d'upload */
-    .upload-container {
-        display: flex;
-        align-items: center;
-        gap: 5px;
-    }
-
-    /* Style pour les icônes (ajustez la taille si nécessaire) */
+    /* Style pour l'icône (ajustez la taille si nécessaire) */
     .upload-icon {
         font-size: 1.5em;
     }
@@ -115,37 +108,27 @@ for message in st.session_state.chat.history:
                 if hasattr(part, 'image'):
                     st.image(part.image)
 
-# Créer le conteneur principal pour la zone de saisie et les uploads
+# Créer le conteneur principal pour la zone de saisie et l'upload
 input_area = st.container()
 
 with input_area:
     # Zone de saisie
     prompt = st.chat_input("Que puis-je faire pour vous ?", key="chat_input")
 
-    # Colonnes pour les icônes d'upload (utiliser plus de colonnes pour un espacement plus fin)
-    ucol1, ucol2, ucol3, ucol4 = st.columns([1,1,1,1])
+    # Colonnes pour l'icône et l'uploader (réduit à 2 colonnes)
+    ucol1, ucol2 = st.columns([1, 5])  # Ajustez la largeur relative si nécessaire
 
     with ucol1:
         # Icône d'upload de fichiers
         st.markdown("<span class='upload-icon'>📁</span>", unsafe_allow_html=True)
 
     with ucol2:
-        # Upload de fichiers
-        uploaded_files = st.file_uploader("", type=["txt", "mp4", "mp3", "pdf"],
-                                            accept_multiple_files=True, key="files",
-                                            label_visibility="collapsed")
-
-    with ucol3:
-        # Icône d'upload d'images
-        st.markdown("<span class='upload-icon'>📸</span>", unsafe_allow_html=True)
-
-    with ucol4:
-        # Upload d'images
-        uploaded_images = st.file_uploader("", type=["jpg", "jpeg", "png", "gif"],
-                                            accept_multiple_files=True, key="images",
-                                            label_visibility="collapsed")
-
-# Appliquer les styles aux éléments
+        # Upload de fichiers (tous types)
+        uploaded_file = st.file_uploader("", type=["txt", "mp4", "mp3", "pdf", "png", "jpg", "jpeg", "gif"],
+                                        accept_multiple_files=True, key="files",
+                                        label_visibility="collapsed")
+
+# Appliquer les styles aux éléments (simplifié)
 st.markdown(f"""
     <style>
         div[data-testid='stChatInput'] {{
@@ -171,23 +154,23 @@ if 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
-        if uploaded_files:
-            for file in uploaded_files:
+        # Traitement des fichiers uploadés
+        if uploaded_file:
+            for file in uploaded_file:
                 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)
+                    file_extension = os.path.splitext(file.name)[1].lower()
+                    if file_extension in ['.jpg', '.jpeg', '.png', '.gif']:
+                        # Si c'est une image, la traiter comme une image
+                        image = Image.open(file)
+                        content.append(image)
+                        st.chat_message("user").image(image)
+                    else:
+                        # Sinon, la traiter comme un fichier
+                        with tempfile.NamedTemporaryFile(delete=False, suffix=file_extension) 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)