exprt import

chat_with_pps.py

@@ -12,6 +12,44 @@ from high_chart import test_chart
 from export_doc import export_conversation,convert_pp_to_csv,get_conversation
 import random
 
+def parse_conversation(file_content):
+    conversation = []
+    current_speaker = None
+    current_message = []
+
+    for line in file_content.decode('utf-8').splitlines():
+        line = line.strip()
+        if line.startswith('AI:'):
+            if current_message:
+                conversation.append((current_speaker, "\n".join(current_message)))
+                current_message = []
+            current_speaker = 'AI'
+            current_message.append(line[3:].strip())
+        elif line.startswith('Moi:'):
+            if current_message:
+                conversation.append((current_speaker, "\n".join(current_message)))
+                current_message = []
+            current_speaker = 'Moi'
+            current_message.append(line[4:].strip())
+        else:
+            current_message.append(line)
+    
+    if current_message:
+        conversation.append((current_speaker, "\n".join(current_message)))
+    
+    return conversation
+
+def convert_to_message_objects(conversation):
+    message_objects = []
+    for speaker, message in conversation:
+        if speaker == 'AI':
+            message_objects.append(AIMessage(content=message))
+        elif speaker == 'Moi':
+            message_objects.append(HumanMessage(content=message))
+    
+    message_objects.pop(0)
+    return message_objects
+
 
 load_dotenv()
 def generate_random_color():
@@ -120,8 +158,8 @@ def disp_carto_in_chat():
 @st.experimental_dialog("Telecharger",width="small")
 def dowmload_history():
     brand_name = st.session_state['Nom de la marque']
-    format = st.radio("Choisissez le document à telecharger",["",f"Rapport des parties prenantes {brand_name} (PDF)",f"DataFrame des parties prenantes {brand_name} (CSV)",f"historique de conversation {brand_name} (Fichier Texte)"])
-    if format == f"Rapport des parties prenantes {st.session_state['Nom de la marque']} (PDF)":
+    format = st.radio("Choisissez le document à telecharger",["",f"Rapport des parties prenantes (PDF)",f"DataFrame des parties prenantes (CSV)",f"historique de conversation (Fichier Texte)"])
+    if format == f"Rapport des parties prenantes (PDF)":
         with st.spinner("Generation en cours..."):
             summary = get_response("Donne moi un RESUME de la Conversation", st.session_state.chat_history,format_context(st.session_state['pp_grouped'],st.session_state['Nom de la marque']),st.session_state.model)
             summary = ''.join(summary)
@@ -129,12 +167,12 @@ def dowmload_history():
             if pdf:
                 st.download_button("Télécharger le PDF", data=pdf, file_name=f"Cartographie {brand_name}.pdf", mime="application/pdf")
         
-    if format == f"DataFrame des parties prenantes {brand_name} (CSV)":
+    if format == f"DataFrame des parties prenantes (CSV)":
         csv = convert_pp_to_csv(st.session_state['pp_grouped'])
         if csv:
             st.download_button("Télécharger le CSV", data=csv, file_name=f"parties_prenantes {brand_name}.csv", mime="text/csv")
     
-    if format == f"historique de conversation {brand_name} (Fichier Texte)":
+    if format == f"historique de conversation (Fichier Texte)":
         conv = get_conversation()
         if not conv:
             st.error("Une erreur s'est produite lors de la récupération de l'historique de conversation")
@@ -145,6 +183,16 @@ def dowmload_history():
     
     if st.button("Fermer"):
         st.rerun()
+
+@st.experimental_dialog("Importer",width="small")
+def import_conversation():
+    uploaded_file = st.file_uploader("Choisissez un fichier texte", type=["txt"])
+    if uploaded_file is not None:
+        file_content = uploaded_file.read()
+        conversation = parse_conversation(file_content)
+        message_object = convert_to_message_objects(conversation)
+        st.session_state.chat_history.extend(message_object)
+        st.experimental_rerun()
     
 
 
@@ -170,7 +218,7 @@ def display_chat():
     
     #sticky bar at the top
     header = st.container()
-    col1,col2,col3, col4 = header.columns([2,3,2,4])
+    col1,col2,col3, col4,col5,col6 = header.columns([2,3,2,3,2,1])
 
     if col1.button("Prompts"):
         show_prompts()
@@ -182,7 +230,8 @@ def display_chat():
     if col4.button("Télécharger"):
         dowmload_history()
     header.write("""<div class='fixed-header'/>""", unsafe_allow_html=True)
-
+    if col5.button("Importer"):
+        import_conversation()
 
     # Custom CSS for the sticky header
     st.markdown(

partie_prenante_carte.py

@@ -327,7 +327,7 @@ def display_pp():
     brand_name = st.text_input("Nom de la marque", st.session_state["Nom de la marque"])
     st.session_state["Nom de la marque"] = brand_name
 
-    option = st.radio("Source", ("A partir de votre site web", "A partir de vos documents entreprise","A partir de votre Cartographie des parties prenantes telechargée en CSV"))
+    option = st.radio("Source", ("A partir de votre site web", "A partir de vos documents entreprise","A partir de fichier interieur (CSV)"))
 
     #if the user chooses to extract from website
     if option == "A partir de votre site web":
@@ -400,7 +400,7 @@ def display_pp():
 
                     # disp_vertical_slider(partie_prenante)
                     # st.altair_chart(c, use_container_width=True)
-    if option == "A partir de votre Cartographie des parties prenantes telechargée en CSV":
+    if option == "A partir de fichier interieur (CSV)":
         uploaded_file = st.file_uploader("Télécharger le fichier CSV", type="csv")
         if uploaded_file is not None:
             load_csv(uploaded_file)