Update src/app.py

src/app.py

@@ -222,10 +222,11 @@ if query:
             # 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, raw_context, lang = st.session_state.generator.retrieve_qa(
+                    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
@@ -267,12 +268,11 @@ 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):
-        final = st.session_state.generator.generate_answer(
-        messages=st.session_state.messages,
-        solution=solution,
-        lang=lang,
-        raw_context=raw_context
+       qa_pairs, solution, lang, sources = st.session_state.generator.retrieve_qa(
+        query=query,
+        docs_scores=docs
     )
+
         st.session_state.messages.append({"role": "assistant", "content": final})
        # save_conversation_to_mongo(st.session_state.messages)
         for key in ["qa_pairs", "solution", "lang", "clarif_idx", "clarif_shown"]: