first commit

src/app.py

@@ -208,11 +208,11 @@ if query:
             # 2. Reranking des documents pour améliorer la pertinence
             if docs:
                 rerank_result = st.session_state.reranker.rerank(query, passages)
+
+                # Si le reranker a bien retourné un résultat
                 if rerank_result is not None:
                     st.session_state.last_rerank = rerank_result.model_dump()
                     st.code(rerank_result)
-                else:
-                    st.error("Le rerank n'a pas produit de résultat valide.")
 
                     # Vérification si un feedback est requis
                     if rerank_result.feedback_required:
@@ -220,12 +220,9 @@ if query:
                             "role": "assistant",
                             "content": rerank_result.message
                         })
-                        #save_conversation_to_mongo(st.session_state.messages)
-                        st.rerun() 
-                         # Utiliser st.rerun() au lieu de continue
-                    
+                        st.rerun()  # Arrête ici pour laisser l'utilisateur reformuler
+
                     # Filtrage et réorganisation des documents selon le reranking
-                   
                     if rerank_result.top_indexes:
                         reranked_docs = [docs[i] for i in rerank_result.top_indexes if i < len(docs)]
                         docs = reranked_docs
@@ -236,6 +233,16 @@ if query:
                         })
                         st.stop()
 
+                # Si le reranker a échoué
+                else:
+                    st.error("❌ Le reranker a échoué et n'a renvoyé aucun résultat.")
+                    st.session_state.messages.append({
+                        "role": "assistant",
+                        "content": "Une erreur est survenue pendant le tri des documents. Veuillez réessayer."
+                    })
+                    st.stop()
+
+
                     
             # 3. Génération de la réponse avec les documents reranked
             if docs:  # Vérifier que nous avons des documents avant de continuer