Update app.py

app.py

@@ -4,7 +4,7 @@ from excel_chat import *
 from classification import *
 from chart_generation import *
 from charts_advanced import *
-#from users_management import *
+from users_management import *
 
 
 # Categories
@@ -84,19 +84,19 @@ df_cate = pd.DataFrame(categories)
 
 with gr.Blocks() as demo:
     
-    # with gr.Row():
-    #     with gr.Column():
-    #         gr.Markdown("## Extaction, Classification and AI tool")
-    #     with gr.Column():
-    #         md_username = gr.Markdown()
-    #         btn_logout = gr.Button("Logout")
-    # with gr.Accordion(label="**Login** to keep user preferences", open=False):
-    #     st_user = gr.State()
-    #     with gr.Column():
-    #         tb_user = gr.Textbox(label='Username')
-    #         tb_pwd = gr.Textbox(label='Password', type='password')
-    #     with gr.Row():
-    #         btn_login = gr.Button('Login')
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("## Extaction, Classification and AI tool")
+        with gr.Column():
+            md_username = gr.Markdown()
+            btn_logout = gr.Button("Logout")
+    with gr.Accordion(label="**Login** to keep user preferences", open=False):
+        st_user = gr.State()
+        with gr.Column():
+            tb_user = gr.Textbox(label='Username')
+            tb_pwd = gr.Textbox(label='Password', type='password')
+        with gr.Row():
+            btn_login = gr.Button('Login')
 
     with gr.Tab("File extraction"):
         gr.Markdown(" Put either just a link, or a link and an excel file with an 'Actions' column")
@@ -154,9 +154,9 @@ with gr.Blocks() as demo:
     fi_excel = gr.File(label="Excel File")
 
     # authentication
-    # btn_login.click(auth_user, inputs=[tb_user, tb_pwd], outputs=[st_user, md_username, dd_prompt, dd_keywords])
-    # tb_pwd.submit(auth_user, inputs=[tb_user, tb_pwd], outputs=[st_user, md_username, dd_prompt, dd_keywords])
-    # btn_logout.click(logout, inputs=None, outputs=[st_user, md_username, dd_prompt, dd_keywords])
+    btn_login.click(auth_user, inputs=[tb_user, tb_pwd], outputs=[st_user, md_username, dd_prompt, dd_keywords])
+    tb_pwd.submit(auth_user, inputs=[tb_user, tb_pwd], outputs=[st_user, md_username, dd_prompt, dd_keywords])
+    btn_logout.click(logout, inputs=None, outputs=[st_user, md_username, dd_prompt, dd_keywords])
     
 
     btn_search_status.click(extract_statuses, inputs=dd_url, outputs=dd_status)
@@ -167,7 +167,7 @@ with gr.Blocks() as demo:
     
     btn_extract.click(extractionPrincipale, inputs=[dd_url, fi_excel, dd_status], outputs=[fi_excel, tb_message])
     
-    mist_button.click(chat_with_mistral, inputs=[dd_source_ask, tb_destcol, dd_prompt, fi_excel, dd_url, dd_searchcol, dd_keywords, rd_llm], outputs=[fi_excel, df_output]) #todo ", st_user"
+    mist_button.click(chat_with_mistral, inputs=[dd_source_ask, tb_destcol, dd_prompt, fi_excel, dd_url, dd_searchcol, dd_keywords, rd_llm, st_user], outputs=[fi_excel, df_output]) #todo "
     
     btn_classif.click(classification, inputs=[dd_source_class, fi_excel, df_category], outputs=[fi_excel, df_output])