Update src/app.py

src/app.py

@@ -216,26 +216,33 @@ if query:
                 # Si le reranker a échoué
                 else:
                     pass
+                
+
 
 
                     
             # 3. Génération de la réponse avec les documents reranked
             if docs:  # Vérifier que nous avons des documents avant de continuer
-                #with st.spinner():
-                    qa_pairs, solution, lang = st.session_state.generator.retrieve_qa(
-                        query=query,
-                        docs_scores=docs
-                    )
-                    st.session_state.qa_pairs = qa_pairs
-                    st.session_state.solution = solution
-                    st.session_state.lang = lang
+                qa_pairs, solution, lang = st.session_state.generator.retrieve_qa(
+                    query=query,
+                    docs_scores=docs
+                )
+                st.session_state.qa_pairs = qa_pairs
+                st.session_state.solution = solution
+                st.session_state.lang = lang
+            
+                if not solution.strip():
+                    st.session_state.messages.append({
+                        "role": "assistant",
+                        "content": "I'm sorry, I cannot answer this request. It is beyond my capabilities or not related to car insurance."
+                    })
             else:
                 # Aucun document disponible
                 st.session_state.messages.append({
                     "role": "assistant",
                     "content": "Aucun document pertinent trouvé pour votre requête. Pouvez-vous la reformuler ?"
                 })
-                #save_conversation_to_mongo(st.session_state.messages)
+                  
                 
         except ValueError as e:
             st.session_state.messages.append({
@@ -266,14 +273,6 @@ if "qa_pairs" in st.session_state:
 
     # Lorsque toutes les clarifications sont remplies, générer la réponse finale
     if st.session_state.clarif_idx >= len(st.session_state.qa_pairs):
-        st.write("Solution récupérée:", repr(st.session_state.solution))
-    
-        if not st.session_state.solution.strip():
-                st.session_state.messages.append({
-                    "role": "assistant",
-                    "content": "I'm sorry, I cannot answer this request. It is beyond my capabilities or not related to car insurance."
-                })
-        else:
             final = st.session_state.generator.generate_answer(
                 st.session_state.messages,
                 st.session_state.solution,