Update app.py

app.py

@@ -0,0 +1,88 @@
+
+
+import streamlit as st
+from config import index, client, langsmith_project
+from pdf_processing import get_existing_pdf, load_and_preprocess_pdf, split_text
+from pinecone_utils import index_pdf, retrieve_documents
+from graph_agent import agent
+
+# Load and index PDF (only once)
+@st.cache_resource
+def initialize_pdf_indexing():
+    pdf_path = get_existing_pdf()
+    if pdf_path:
+        text = load_and_preprocess_pdf(pdf_path)
+        texts = split_text(text)
+        index_pdf(texts)
+
+# Initialize PDF indexing
+initialize_pdf_indexing()
+
+# 🔹 Initialisation de l'historique des interactions
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
+
+# 🔹 Fonction pour traiter la requête de l'utilisateur
+def process_query(query):
+    """Process the user query and update the chat history."""
+    # Ajouter la question de l'utilisateur à l'historique
+    st.session_state.chat_history.append({"role": "user", "content": query})
+
+    # Exécuter l'agent avec un indicateur de chargement
+    with st.spinner("Recherche en cours..."):
+        initial_state = {"query": query, "messages": [], "relevant_docs": [], "response": ""}
+        result = agent.invoke(initial_state)
+        response = result["response"]
+
+        # Ajouter la réponse de l'assistant à l'historique
+        st.session_state.chat_history.append({"role": "assistant", "content": response})
+
+        # Log the query and response in LangSmith
+        client.create_run(
+            name="RAG_Query",
+            run_type="chain",
+            project_name=langsmith_project,
+            inputs={"query": query},
+            outputs={"response": response},
+        )
+
+# 🔹 Barre latérale améliorée
+def display_sidebar():
+    """Display the sidebar with additional information and options."""
+    with st.sidebar:
+        st.title("📄 La confession muette")
+        st.write("Posez vos questions sur le document.")
+        st.image(agent.get_graph().draw_mermaid_png(), caption="Workflow Graph")
+
+# 🔹 Afficher l'historique de chat
+def display_chat_history():
+    """Display the chat history in a conversational format."""
+    for message in st.session_state.chat_history:
+        if message["role"] == "user":
+            with st.chat_message("user"):
+                st.markdown(message["content"])
+        elif message["role"] == "assistant":
+            with st.chat_message("assistant"):
+                st.markdown(message["content"])
+
+# 🔹 Point d'entrée de l'application
+def main():
+    """Main function to run the Streamlit app."""
+    st.title("Architecture A")
+
+    # Afficher la barre latérale
+    display_sidebar()
+
+    # Afficher l'historique de chat
+    display_chat_history()
+
+    # Case d'entrée des questions en bas
+    query = st.chat_input("Posez votre question ici:")
+    if query:
+        process_query(query)
+        st.rerun()  # Rafraîchir la page pour afficher immédiatement les nouvelles réponses
+
+if __name__ == "__main__":
+    main()
+
+