import streamlit as st
import plotly.graph_objects as go
import plotly.express as px
from PIL import Image
import pdfkit
import pandas as pd
import plotly.io as pio
import tempfile
def construct_plot():
# Define data
stakeholders = st.session_state['pp_grouped']
if stakeholders is None or len(stakeholders) == 0:
st.error("Aucune partie prenante n'a été définie")
return None
# Create plot
fig = go.Figure()
# Add category areas
fig.add_shape(type="rect", x0=0, y0=50, x1=50, y1=100, fillcolor="lightblue", opacity=0.2, line_width=0)
fig.add_shape(type="rect", x0=50, y0=50, x1=100, y1=100, fillcolor="lightyellow", opacity=0.2, line_width=0)
fig.add_shape(type="rect", x0=0, y0=0, x1=50, y1=50, fillcolor="lightcoral", opacity=0.2, line_width=0)
fig.add_shape(type="rect", x0=50, y0=0, x1=100, y1=50, fillcolor="lightcyan", opacity=0.2, line_width=0)
# Add category titles
fig.add_annotation(x=10, y=90, text="Rendre satisfait", showarrow=False)
fig.add_annotation(x=60, y=90, text="Gérer étroitement", showarrow=False)
fig.add_annotation(x=10, y=40, text="Suivre de près", showarrow=False)
fig.add_annotation(x=60, y=40, text="Tenir informé", showarrow=False)
x_array = [stakeholder['x'] for stakeholder in stakeholders]
y_array = [stakeholder['y'] for stakeholder in stakeholders]
name_array = [stakeholder['name'] for stakeholder in stakeholders]
color_sequence = px.colors.qualitative.Plotly
# Add stakeholders to plot
for i,stakeholder in enumerate(stakeholders):
fig.add_trace(go.Scatter(
x=[stakeholder['x']],
y=[stakeholder['y']],
mode='markers+text',
marker=dict(color=color_sequence[i % len(color_sequence)],size=20),
textposition="top center",
name=stakeholder['name']
))
# Update layout
fig.update_layout(
legend=dict( orientation="h", yanchor="bottom",y=1.02,title="Parties prenantes"),
height=600,
title=dict(text="Cartographie des parties prenantes", x=0.5, y=1, xanchor="center", yanchor="top"),
xaxis=dict(title="Influence", range=[0, 100]),
yaxis=dict(title="Pouvoir", range=[0, 100]),
showlegend=True
)
# Display plot in Streamlit
return fig
def save_plot_as_pdf(fig, logo_path, title):
st.write("saving plot as pdf")
with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmpfile:
st.write("tmpfile created")
image_bytes = pio.to_image(fig, format='png')
st.write("image_bytes")
tmpfile.write(image_bytes)
tmpfile.close()
plot_image_path = tmpfile.name
st.write(plot_image_path)
html_content = f"""
{title}
{"![](" + logo_path + ")
" if logo_path else ""}
{title}
"""
with tempfile.NamedTemporaryFile(delete=False, suffix=".html") as tmpfile:
tmpfile.write(html_content.encode('utf-8'))
tmpfile.close()
html_path = tmpfile.name
pdf_path = html_path.replace('.html', '.pdf')
st.write(pdf_path)
pdfkit.from_file(html_path, pdf_path)
return pdf_path
def download_pdf():
# Construct plot
fig = construct_plot()
if fig is None:
return None
st.write("fig constructed")
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"
pdf_title = "Cartographie des parties prenantes"
# Button to download PDF
pdf_path = save_plot_as_pdf(fig, logo_path, pdf_title)
st.write("pdf saved")
with open(pdf_path, "rb") as pdf_file:
st.download_button(label="Download PDF", data=pdf_file, file_name="stakeholder_analysis.pdf")