Update app.py

app.py

@@ -112,9 +112,11 @@ df_cate = pd.DataFrame(categories)
 
 ### Functions needed for Split Files
 
-def functionCall(fi_input, dropdown, choice):
+def functionCall(fi_input, dropdown, choice, chunk_size):
     if choice == "Intelligent split":
         return split_in_df(fi_input)
+    elif choice == "Non intelligent split":
+        return non_intelligent_split(fi_input, chunk_size)
     else:
         return split_by_keywords(fi_input,dropdown)
 
@@ -122,9 +124,11 @@ def change_textbox(dropdown,radio):
     if len(dropdown) == 0 :
         dropdown = ["introduction", "objective", "summary", "conclusion"]
     if radio == "Intelligent split by keywords":
-        return gr.Dropdown(dropdown, multiselect=True, visible=True, allow_custom_value=True)
+        return gr.Dropdown(dropdown, multiselect=True, visible=True, allow_custom_value=True), gr.Number(visible=False)
+    elif radio == "Non intelligent split":
+        return gr.Dropdown(dropdown, visible=False),gr.Number(label="Chunk size", value=1000, interactive=True, visible=True)
     else:
-        return gr.Dropdown(dropdown, multiselect=True, visible=False, allow_custom_value=True)
+        return gr.Dropdown(dropdown, visible=False),gr.Number(visible=False)
 
 ### Split files end
 
@@ -222,11 +226,13 @@ with gr.Blocks() as demo:
     with gr.Tab("Split Files"):
         gr.Markdown("### Upload your standard documentation (pdf, doc, docx) to split it into paragraphs in an Excel file")
         radio = gr.Radio(
-            ["Intelligent split", "Intelligent split by keywords"], label="Choose your selection", value = "Intelligent split"
+            ["Intelligent split", "Intelligent split by keywords", "Non intelligent split"], label="Choose your selection", value = "Intelligent split"
         )
         dropdown_split = gr.Dropdown(["introduction", "objective", "conclusion", "summary"], multiselect=True, visible=False, allow_custom_value=True, label="Select or add keywords")
+        nb_split = gr.Number(label="Chunk size", value=1000, interactive=True, visible=False)
         
         fi_input = gr.File(file_count='multiple')
+        btn_split = gr.Button("Split")
 
     with gr.Tab("Ask LLM"):
         gr.Markdown("### This section utilizes Large Language Models (LLMs) to query rows in an Excel file")
@@ -323,8 +329,9 @@ with gr.Blocks() as demo:
 
     # Split files
     #fi_input.upload(split_in_df, inputs=fi_input, outputs=fi_excel)
-    fi_input.upload(functionCall, inputs=[fi_input, dropdown_split, radio], outputs=fi_excel)
-    radio.change(fn=change_textbox, inputs=[dropdown_split,radio], outputs=dropdown_split)
+    fi_input.upload(functionCall, inputs=[fi_input, dropdown_split, radio, nb_split], outputs=fi_excel)
+    btn_split.click(functionCall, inputs=[fi_input, dropdown_split, radio, nb_split], outputs=fi_excel)
+    radio.change(fn=change_textbox, inputs=[dropdown_split,radio], outputs=[dropdown_split, nb_split])
 
     #llm
     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, fi_config])