Spaces:

bziiit
/

Carto-RSE

Build error

App Files Files Community

Ilyas KHIAT commited on Jun 18, 2024

Commit

439e01f

1 Parent(s): 81f75af

pdf export

Browse files

Files changed (8) hide show

app.py +1 -0
chat_with_pps.py +20 -3
export_doc.py +67 -0
high_chart.py +2 -5
packages.txt +1 -0
partie_prenante_carte.py +1 -0
prompt.py +0 -1
requirements.txt +5 -1

app.py CHANGED Viewed

@@ -26,6 +26,7 @@ from chat_te import display_chat_te
 def main():
     st.markdown(":point_left: Cliquez pour vous inspirer", unsafe_allow_html=True)
     st.sidebar.title("OPEN DATA & IA au service de la RSE")
     section_principale = st.sidebar.radio(

 def main():
     st.markdown(":point_left: Cliquez pour vous inspirer", unsafe_allow_html=True)
     st.sidebar.title("OPEN DATA & IA au service de la RSE")
     section_principale = st.sidebar.radio(

chat_with_pps.py CHANGED Viewed

@@ -8,6 +8,8 @@ from langchain_mistralai.chat_models import ChatMistralAI
 from download_chart import construct_plot
 from prompt import get_prompts_list
 from st_copy_to_clipboard import st_copy_to_clipboard
 load_dotenv()
@@ -85,11 +87,19 @@ def choose_model(index):
     if st.button("Valider"):
         st.rerun()
 def display_chat():
     # app config
     st.title("Chatbot")
     models_name = {
         "Mistral (FR)":1,
         "GPT-4o":0
@@ -106,15 +116,20 @@ def display_chat():
     #sticky bar at the top
     header = st.container()
-    col1,col2,col3 = header.columns([1,2,4])
     if col1.button("Prompts"):
         show_prompts()
     if col2.button(f"Modèle: {st.session_state.model}"):
         index = models_name[st.session_state.model]
         choose_model(index)
     header.write("""<div class='fixed-header'/>""", unsafe_allow_html=True)
     # Custom CSS for the sticky header
     st.markdown(
         """
@@ -171,5 +186,7 @@ def display_chat():
             st_copy_to_clipboard(response)
             if "cartographie" in response:
                     display_chart()
         st.session_state.chat_history.append(AIMessage(content=response))

 from download_chart import construct_plot
 from prompt import get_prompts_list
 from st_copy_to_clipboard import st_copy_to_clipboard
+from high_chart import test_chart
+from export_doc import export_conversation
 load_dotenv()
     if st.button("Valider"):
         st.rerun()
+@st.experimental_dialog("Ma cartographie",width="large")
+def disp_carto_in_chat():
+    test_chart()
+@st.experimental_dialog("Telecharger",width="large")
+def dowmload_history():
+    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)
+    export_conversation(AIMessage(content=summary).content)
 def display_chat():
     # app config
     st.title("Chatbot")
     models_name = {
         "Mistral (FR)":1,
         "GPT-4o":0
     #sticky bar at the top
     header = st.container()
+    col1,col2,col3, col4 = header.columns([2,3,2,4])
     if col1.button("Prompts"):
         show_prompts()
     if col2.button(f"Modèle: {st.session_state.model}"):
         index = models_name[st.session_state.model]
         choose_model(index)
+    if col3.button("Ma Carto"):
+        disp_carto_in_chat()
+    if col4.button("Télécharger"):
+        dowmload_history()
     header.write("""<div class='fixed-header'/>""", unsafe_allow_html=True)
     # Custom CSS for the sticky header
     st.markdown(
         """
             st_copy_to_clipboard(response)
             if "cartographie" in response:
                     display_chart()
+            if "rajoute" in response:
+                st.session_state['pp_grouped'].append({"name":"partie prenante","x":0,"y":0})
         st.session_state.chat_history.append(AIMessage(content=response))

export_doc.py ADDED Viewed

	@@ -0,0 +1,67 @@

+import streamlit as st
+import markdown2
+import pdfkit
+from io import BytesIO
+from IPython.display import display, FileLink
+import base64
+from langchain_core.messages import AIMessage, HumanMessage
+def create_pdf_from_markdown(logo_path, image_path, conversation,summary):
+    # Convertir la conversation en markdown
+    markdown_text = "\n".join([f"### {entry['speaker']}:\n {entry['text']}\n ---" for entry in conversation])
+    markdown_summary = f"{summary}\n --- \n ---"
+    st.write(markdown_summary)
+    # Convertir le markdown en HTML
+    html_content = markdown2.markdown(markdown_text)
+    html_summary = markdown2.markdown(markdown_summary)
+    # image_base64 = base64.b64encode(image_path).decode('utf-8')
+    # Créer le HTML complet avec les images et le texte
+    html_template = f"""
+    <!DOCTYPE html>
+    <html lang="en">
+    <head>
+        <meta charset="UTF-8">
+    </head>
+    <body>
+        <div style="text-align: center;">
+            <h1>Rapport de Conversation {st.session_state["Nom de la marque"]}</h1>
+            <img src="{logo_path}" alt="Logo" style="width: 150px;"/>
+        </div>
+        <div style="text-align: center; margin-top: 20px;">
+            <img src="data:image/png;base64" alt="Cartographie" style="width: 100%;"/>
+        </div>
+        <h2>RESUME</h2>
+        {html_summary}
+        <h2>Historique de la Conversation</h2>
+        {html_content}
+    </body>
+    </html>
+    """
+    # Convertir le HTML en PDF
+    pdf = pdfkit.from_string(html_template, False)
+    return pdf
+def get_conversation():
+    conversation = []
+    for message in st.session_state.chat_history:
+        if isinstance(message, AIMessage):
+            conversation.append({"speaker": "AI", "text": message.content})
+        elif isinstance(message, HumanMessage):
+            conversation.append({"speaker": "Moi", "text": message.content})
+    return conversation
+def export_conversation(summary):
+    logo_path = "https://static.wixstatic.com/media/d7d3da_b69e03ae99224f7d8b6e358918e60071~mv2.png/v1/crop/x_173,y_0,w_1906,h_938/fill/w_242,h_119,al_c,q_85,usm_0.66_1.00_0.01,enc_auto/BZIIIT_LOGO-HORIZ-COULEUR.png"  # Replace with your image path
+    conversation = get_conversation()
+    image_path = "newplot.png"
+    pdf = create_pdf_from_markdown(logo_path, image_path, conversation,summary)
+    st.success("PDF généré avec succès!")
+    if st.download_button("Télécharger le PDF", data=pdf, file_name=f"Cartographie {st.session_state["Nom de la marque"]}.pdf", mime="application/pdf"):
+        st.rerun()

high_chart.py CHANGED Viewed

@@ -183,22 +183,19 @@ def construct_data():
 def test_chart():
    #The component can render any highchart definition
-   if len(st.session_state['pp_grouped']) == 0:
       return None
    points = st.session_state['pp_grouped']
    cd2["series"][0]["data"] = points
    chart = hct.streamlit_highcharts(cd2,640) #640 is the chart height
    # if chart:
    #    st.session_state['pp_grouped'] = chart
-   emp = st.empty()
    col0, col1, col2, col3 = st.columns([1,1,1,1])
    if col1.button("Sauvegarder",key="save"):
       st.session_state['pp_grouped'] = chart.copy()
-      emp.success("Saved")
       st.experimental_rerun()
    if col2.button("Télécharger",key="download"):

 def test_chart():
    #The component can render any highchart definition
+   if "pp_grouped" not in st.session_state or len(st.session_state['pp_grouped']) == 0:
       return None
    points = st.session_state['pp_grouped']
    cd2["series"][0]["data"] = points
    chart = hct.streamlit_highcharts(cd2,640) #640 is the chart height
    # if chart:
    #    st.session_state['pp_grouped'] = chart
    col0, col1, col2, col3 = st.columns([1,1,1,1])
    if col1.button("Sauvegarder",key="save"):
       st.session_state['pp_grouped'] = chart.copy()
       st.experimental_rerun()
    if col2.button("Télécharger",key="download"):

packages.txt ADDED Viewed

	@@ -0,0 +1 @@


1	+ wkhtmltopdf

partie_prenante_carte.py CHANGED Viewed

@@ -290,6 +290,7 @@ def display_pp():
     load_dotenv()
     fire_crawl_api_key = os.getenv("FIRECRAWL_API_KEY")
     #check if brand name and description are already set
     if "Nom de la marque" not in st.session_state:
         st.session_state["Nom de la marque"] = ""

     load_dotenv()
     fire_crawl_api_key = os.getenv("FIRECRAWL_API_KEY")
     #check if brand name and description are already set
     if "Nom de la marque" not in st.session_state:
         st.session_state["Nom de la marque"] = ""

prompt.py CHANGED Viewed

@@ -4,7 +4,6 @@ import pandas as pd
 from data_manager_bziiit import get_prompts
 from langchain_core.messages import AIMessage, HumanMessage
 from session import get_rag
-import pyperclip
 prompts = []
 def get_prompts_list():

 from data_manager_bziiit import get_prompts
 from langchain_core.messages import AIMessage, HumanMessage
 from session import get_rag
 prompts = []
 def get_prompts_list():

requirements.txt CHANGED Viewed

@@ -38,4 +38,8 @@ langchain-core
 langchain-mistralai
 firecrawl-py
 st_copy_to_clipboard
-pyperclip

 langchain-mistralai
 firecrawl-py
 st_copy_to_clipboard
+pyperclip
+fpdf2
+markdown2
+base64
+io