From 9857d6fb21ab9f53a3987337fb3dd0f9515e401c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Pelletier?= <francois@jevalide.ca>
Date: Sat, 24 Aug 2024 00:26:59 -0400
Subject: [PATCH] Ajout de Oauth2

---
 main.py          | 36 +++++++++++++++++++++++++-----------
 requirements.txt |  4 ++--
 2 files changed, 27 insertions(+), 13 deletions(-)

diff --git a/main.py b/main.py
index c7cb723..aaa7dbf 100644
--- a/main.py
+++ b/main.py
@@ -64,7 +64,20 @@ fabriquedoc_endpoint = os.environ.get("FABRIQUEDOC_ENDPOINT", "http://127.0.0.1:
 
 st.write(f"Endpoint : {fabriquedoc_endpoint}")
 
-tab1, tab2, tab3, tab4 = st.tabs(["Markdown", "Aperçu", "Paramètres", "Images"])
+tab0, tab1, tab2, tab3, tab4 = st.tabs(["Login", "Markdown", "Aperçu", "Paramètres", "Images"])
+
+with tab0:
+    with st.form(key='authentication'):
+        username = st.text_input("Username")
+        password = st.text_input("Password", type="password")
+        submit_button = st.form_submit_button(label='Login')
+        if submit_button:
+            # Send a POST request to your authentication endpoint
+            response = requests.post(f'{fabriquedoc_endpoint}/token/', data={"username": username, "password": password})
+            if response.status_code == 200:
+                bearer_token = response.json()["access_token"]
+                # Store the bearer token in the session state
+                st.session_state['bearer_token'] = bearer_token
 
 with tab1:
     def button1_callback():
@@ -83,18 +96,19 @@ with tab2:
 
 with tab3:
     st.header("Paramètres")
+    http_headers = {"Authorization": f"Bearer {st.session_state['bearer_token']}"}
     # Styles
-    response_styles = requests.get(f"{fabriquedoc_endpoint}/styles/").json()
+    response_styles = requests.get(f"{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"{fabriquedoc_endpoint}/formats/{selected_style}/").json()
+    response_formats = requests.get(f"{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"{fabriquedoc_endpoint}/format_parameters/{selected_style}/{selected_format}/").json()
+        f"{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:",
@@ -154,11 +168,11 @@ with tab3:
             stilltime=stilltime
         )
 
-        headers = {"Content-Type": "application/json"}
+        post_headers = http_headers | {"Content-Type": "application/json"}
         # Send the POST request with the JSON data in the request body
-        response = requests.get(f"{fabriquedoc_endpoint}/generer/",
-                                json=document_specs.model_dump(),
-                                headers=headers)
+        response = requests.post(f"{fabriquedoc_endpoint}/generer/",
+                                 json=document_specs.model_dump(),
+                                 headers=http_headers)
         # Check the response status code
         if 200 <= response.status_code <= 299:
             # If the request is successful, get the file data from the response
@@ -182,10 +196,10 @@ with tab4:
 
     st.write("Images disponibles")
     # list uploaded files with a request to the GET /images endpoint
-    response = requests.get(f"{fabriquedoc_endpoint}/images/")
+    response = requests.get(f"{fabriquedoc_endpoint}/images/", headers=http_headers)
     images = response.json()["images"]
     selected_image = st.selectbox("Choisis une image:", images)
-    image_response = requests.get(f"{fabriquedoc_endpoint}/images/{selected_image}")
+    image_response = requests.get(f"{fabriquedoc_endpoint}/images/{selected_image}", headers=http_headers)
     image_data = image_response.content
     st.image(image_data)
 
@@ -200,7 +214,7 @@ with tab4:
             files = {"file": uploaded_file}
 
             # Submit the file to the endpoint
-            response = requests.post(url, files=files)
+            response = requests.post(url, files=files, headers=http_headers)
 
             # Check the response status
             if response.status_code < 300:
diff --git a/requirements.txt b/requirements.txt
index 944ac5d..5dd8e8c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 streamlit~=1.30.0
 requests~=2.31.0
-pydantic~=2.0.1
-mdformat~=0.7.16
+pydantic~=2.0.3
+mdformat~=0.7.17
 python-dotenv~=1.0.1
\ No newline at end of file