fabriquedoc-frontend/app_tab3.py

105 lines
4.6 KiB
Python
Raw Permalink Normal View History

import datetime
import requests
import streamlit as st
from models import DocumentSpecs
def app_tab3():
st.header("Paramètres")
http_headers = {"Authorization": f"Bearer {st.session_state['bearer_token']}"}
# Styles
response_styles = requests.get(f"{st.session_state['fabriquedoc_endpoint']}/styles/", headers=http_headers).json()
styles = response_styles.get("styles")
selected_style = st.selectbox("Select a style:", styles)
# Formats
response_formats = requests.get(f"{st.session_state['fabriquedoc_endpoint']}/formats/{selected_style}/", headers=http_headers).json()
formats = response_formats.get("formats")
selected_format = st.selectbox("Select a format:", formats)
# Autres paramètres
response_format_parameters = requests.get(
f"{st.session_state['fabriquedoc_endpoint']}/format_parameters/{selected_style}/{selected_format}/", headers=http_headers).json()
linkcolor = st.text_input("Link color:",
value=response_format_parameters.get("linkcolor"))
tocdepth = st.number_input("Table of Contents depth:",
value=int(response_format_parameters.get("tocdepth")),
min_value=1,
max_value=5,
step=1)
pdfengine = st.text_input("PDF engine:",
value=response_format_parameters.get("pdfengine"))
fontsize = st.number_input("Font size:",
value=int(response_format_parameters.get("fontsize")),
step=1)
paperwidth = st.number_input("Paper width:",
value=int(response_format_parameters.get("paperwidth")),
step=30)
paperheight = st.number_input("Paper height:",
value=int(response_format_parameters.get("paperheight")),
step=30)
ratio = st.number_input("Ratio:",
value=int(response_format_parameters.get("ratio")),
step=10)
margin = st.number_input("Margin:",
value=int(response_format_parameters.get("margin")),
step=10)
vmargin = st.number_input("Vertical margin:",
value=int(response_format_parameters.get("vmargin")),
step=10)
extension = st.selectbox("Extension:", ["jpg", "pdf", "mp4"])
if extension == "mp4":
fps = st.number_input("FPS:",
value=int(response_format_parameters.get("fps")),
step=1)
stilltime = st.number_input("Still time:",
value=int(response_format_parameters.get("stilltime")),
step=1)
else:
fps = 0
stilltime = 0
# Envoi
if st.button("Generate post"):
document_specs = DocumentSpecs(
format=selected_format,
style=selected_style,
linkcolor=linkcolor,
tocdepth=tocdepth,
pdfengine=pdfengine,
content=st.session_state['content'],
fontsize=fontsize,
paperwidth=paperwidth,
paperheight=paperheight,
ratio=ratio,
margin=margin,
vmargin=vmargin,
extension=extension,
fps=fps,
stilltime=stilltime
)
post_headers = http_headers | {"Content-Type": "application/json"}
# Send the POST request with the JSON data in the request body
response = requests.post(f"{st.session_state['fabriquedoc_endpoint']}/generer/",
json=document_specs.model_dump(),
headers=post_headers)
# Check the response status code
if 200 <= response.status_code <= 299:
# If the request is successful, get the file data from the response
file_data = response.content
datef = datetime.datetime.now().strftime("%m-%d-%Y")
if document_specs.extension in ["jpg"]:
extn = "zip"
else:
extn = document_specs.extension
file_name = f"{document_specs.style}-{document_specs.format}-{datef}-output.{extn}"
st.download_button('Télécharger', file_data, file_name=file_name)
else:
# If the request is unsuccessful, display the error status code and message
st.error(f"Request failed with status code {response.status_code}: {response.text}")