Update app.py

app.py

@@ -14,9 +14,11 @@ with gr.Blocks() as demo:
     gr.Markdown("## Extaction, Classification and AI tool")
     with gr.Tab("File extraction"):
         gr.Markdown(" Put either just a link, or a link and an excel file with an 'Actions' column")
-        tb_url = gr.Textbox(label="URL (e.g. https://www.3gpp.org/ftp/TSG_SA/WG1_Serv/TSGS1_105_Athens/Docs)")
+        with gr.Row():
+            dd_url = gr.Dropdown(label="(e.g. https://www.3gpp.org/ftp/TSG_SA/WG1_Serv/TSGS1_105_Athens/Docs)", multiselect=False, value="https://www.3gpp.org/ftp/", allow_custom_value=True, scale=9)
+            btn_search = gr.Button("Search")
         btn_extract = gr.Button("Extract excel from URL")
-        tb_status_message = gr.Textbox(label="Status")
+        tb_message = gr.Textbox(label="Status")
 
     with gr.Tab("Query on columns with mistral"):
         dd_source_ask = gr.Dropdown(label="Source Column(s)", multiselect=True)
@@ -27,6 +29,8 @@ with gr.Blocks() as demo:
             
     with gr.Tab("Classification by topic"):
         dd_source_class = gr.Dropdown(label="Source Column", multiselect=False)
+        gr.Markdown("### The predefined categories can be modified at any time")
+        df_category = gr.DataFrame(label='categories', value=df_cate, interactive=True)
         btn_classif = gr.Button("Categorize")
 
     with gr.Tab("Charts Generation"):
@@ -37,31 +41,30 @@ with gr.Blocks() as demo:
         plt_figure = gr.Plot()
 
     with gr.Tab("Chart Generation"):
+        gr.Markdown("## 🚧 Actuellement en maintenance 🚧")
         with gr.Tab("Overall"):
-            btn_overall = gr.Button("Overall Review")              
+            btn_overall = gr.Button("Overall Review")
         with gr.Tab("By Expert"):
             rd_exp=gr.Radio(["Satellite Networks / Nicolas", "Emergency Communication / Dorin", "Trend Analysis / Ly-Thanh", "Security Trust / Mireille", "Distributed Networks / Guillaume", "Network Security / Khawla", "USIM Management / Vincent", "Eco-Design / Pierre"], label="Expert Name")
             btn_expert = gr.Button("Top 10 by expert")
         with gr.Tab("By Company"):
             tb_com=gr.Textbox(label="Company Name",info="You can write 1, 2 or 3 company names at the same time")
-            btn_type = gr.Button("Document type by company")
+            btn_type = gr.Button("Company info")
         with gr.Row():
             plt_chart = gr.Plot(label="Graphique")
             plt_chart2 = gr.Plot(label="Graphique")
             plt_chart3 = gr.Plot(label="Graphique")
-        
     with gr.Accordion("Excel Preview", open=False):
-        df_output = gr.DataFrame()
-        
+          df_output = gr.DataFrame()
     fi_excel = gr.File(label="Excel File")
 
     fi_excel.change(get_columns, inputs=[fi_excel], outputs=[dd_source_ask, dd_source_class, dd_label1, dd_label2, df_output])
     
-    btn_extract.click(extractionPrincipale, inputs=[tb_url, fi_excel], outputs=[fi_excel, tb_status_message])
+    btn_extract.click(extractionPrincipale, inputs=[dd_url, fi_excel], outputs=[fi_excel, tb_message])
     
-    mist_button.click(chat_with_mistral, inputs=[dd_source_ask, tb_destcol, tb_prompt, tb_filename, fi_excel, tb_url], outputs=[fi_excel, df_output])
+    mist_button.click(chat_with_mistral, inputs=[dd_source_ask, tb_destcol, tb_prompt, tb_filename, fi_excel], outputs=[fi_excel, df_output])
     
-    btn_classif.click(classification, inputs=[dd_source_class, fi_excel], outputs=[fi_excel, df_output])
+    btn_classif.click(classification, inputs=[dd_source_class, fi_excel, df_category], outputs=[fi_excel, df_output])
     
     btn_chart.click(create_bar_plot, inputs=[fi_excel, dd_label1, dd_label2], outputs=[plt_figure])