gestion source

src/streamlit_app.py

@@ -1,9 +1,7 @@
 import streamlit as st
 import pandas as pd
 from datetime import datetime
-import json
 from RAG import answer_question, retrieve_chunks, embed_question, build_context
-import base64
 
 # Configuration de la page
 st.set_page_config(
@@ -89,61 +87,6 @@ if "messages" not in st.session_state:
 if "sources" not in st.session_state:
     st.session_state["sources"] = []
 
-# Fonction pour exporter les conversations
-def export_conversation(format_type="txt"):
-    if not st.session_state["messages"]:
-        return None
-    
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    
-    if format_type == "txt":
-        content = f"Conversation Tindle RAG - {datetime.now().strftime('%d/%m/%Y %H:%M')}\n"
-        content += "=" * 50 + "\n\n"
-        
-        for i, msg in enumerate(st.session_state["messages"]):
-            content += f"{msg['role'].upper()}: {msg['content']}\n\n"
-            
-            # Ajouter les sources si disponibles
-            if msg["role"] == "assistant" and i < len(st.session_state["sources"]):
-                content += "Sources utilisées:\n"
-                for j, source in enumerate(st.session_state["sources"][i], 1):
-                    content += f"{j}. {source['id']}@{source['offset']} (score: {source['score']:.3f})\n"
-                    content += f"   {source['text'][:200]}...\n\n"
-        
-        return content, f"conversation_tindle_{timestamp}.txt"
-    
-    elif format_type == "json":
-        data = {
-            "timestamp": datetime.now().isoformat(),
-            "messages": st.session_state["messages"],
-            "sources": st.session_state["sources"]
-        }
-        return json.dumps(data, indent=2, ensure_ascii=False), f"conversation_tindle_{timestamp}.json"
-
-# Barre d'export
-col1, col2, col3 = st.columns([1, 1, 2])
-with col1:
-    if st.button("📄 Exporter TXT"):
-        if st.session_state["messages"]:
-            content, filename = export_conversation("txt")
-            st.download_button(
-                label="💾 Télécharger TXT",
-                data=content,
-                file_name=filename,
-                mime="text/plain"
-            )
-
-with col2:
-    if st.button("📊 Exporter JSON"):
-        if st.session_state["messages"]:
-            content, filename = export_conversation("json")
-            st.download_button(
-                label="💾 Télécharger JSON",
-                data=content,
-                file_name=filename,
-                mime="application/json"
-            )
-
 # Affichage de l'historique du chat avec sources
 for i, msg in enumerate(st.session_state["messages"]):
     if msg["role"] == "user":
@@ -154,9 +97,11 @@ for i, msg in enumerate(st.session_state["messages"]):
             st.markdown(msg["content"])
             
             # Afficher les sources si disponibles et demandées
-            if show_sources and i < len(st.session_state["sources"]) and st.session_state["sources"][i]:
-                with st.expander(f"📚 Sources utilisées ({len(st.session_state['sources'][i])} passages)"):
-                    for j, source in enumerate(st.session_state["sources"][i], 1):
+            # Correction de l'index : on cherche les sources pour cette réponse spécifique
+            source_index = len([m for m in st.session_state["messages"][:i] if m["role"] == "assistant"])
+            if show_sources and source_index < len(st.session_state["sources"]) and st.session_state["sources"][source_index]:
+                with st.expander(f"📚 Sources utilisées ({len(st.session_state['sources'][source_index])} passages)"):
+                    for j, source in enumerate(st.session_state["sources"][source_index], 1):
                         st.markdown(f"""
                         <div class="source-preview">
                         <strong>Source {j}:</strong> {source['id']}@{source['offset']}