Update app.py

app.py

@@ -109,6 +109,23 @@ df_cate = pd.DataFrame(categories)
 # def update_label(label1):
 #     return gr.update(choices=list(df.columns))
 
+# Functions needed for Split Files
+
+def functionCall(fi_input, dropdown, choice):
+    if choice == "Intelligent split":
+        return split_in_df(fi_input)
+    else:
+        return split_in_keywords(fi_input,dropdown)
+
+def change_textbox(dropdown,radio):
+    print(dropdown)
+    if radio == "Intelligent split by keywords":
+        return gr.Dropdown(dropdown, multiselect=True, visible=True, allow_custom_value=True)
+    else:
+        return gr.Dropdown(dropdown, multiselect=True, visible=False, allow_custom_value=True)
+
+### Split files end
+
 
 def reset_cate(df_categories):
     if df_categories.equals(df_cate):
@@ -175,6 +192,11 @@ 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"
+        )
+        dropdown_split = gr.Dropdown(["introduction", "objective", "conclusion", "summary"], multiselect=True, visible=False, allow_custom_value=True)
+        
         fi_input = gr.File(file_count='multiple')
 
 
@@ -263,7 +285,9 @@ with gr.Blocks() as demo:
 
 
     # Split files
-    fi_input.upload(split_in_df, inputs=fi_input, outputs=fi_excel)
+    #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)
 
     #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])