Update app.py

app.py

@@ -6,6 +6,7 @@ from chart_generation import *
 from charts_advanced import *
 from users_management import *
 from code_df_custom import *
+from split_files_to_excel import *
 
 
 # Categories
@@ -131,6 +132,9 @@ with gr.Blocks() as demo:
                 btn_search_status = gr.Button("Search for status", scale=2)
         btn_extract = gr.Button("Extract excel from URL")
 
+    with gr.Tab("Split Files"):
+        fi_input = gr.File(file_count='multiple')
+
 
     with gr.Tab("Ask LLM"):
         gr.Markdown("### This section utilizes Large Language Models (LLMs) to query rows in an Excel file")
@@ -208,6 +212,10 @@ with gr.Blocks() as demo:
     fi_excel.change(get_columns, inputs=[fi_excel], outputs=[dd_source_ask, dd_source_class, dd_label1, dd_label2, dd_searchcol, df_output,st_filename, df_input])
     
     btn_extract.click(extractionPrincipale, inputs=[dd_url, fi_excel, dd_status], outputs=[fi_excel])
+
+
+    # Split files
+    fi_input.upload(split_in_df, inputs=fi_input, outputs=fi_excel)
     
     mist_button.click(chat_with_mistral, inputs=[dd_source_ask, tb_destcol, dd_prompt, fi_excel, dd_url, dd_searchcol, dd_keywords, dd_llm, st_user], outputs=[fi_excel, df_output, dd_prompt, dd_keywords, st_user])