Update app.py

app.py

@@ -5,6 +5,7 @@ from classification import *
 from chart_generation import *
 from charts_advanced import *
 from users_management import *
+from code_df_custom import *
 
 
 # Categories
@@ -148,6 +149,19 @@ with gr.Blocks() as demo:
             plt_chart = gr.Plot(label="Graphique")
             plt_chart2 = gr.Plot(label="Graphique")
             plt_chart3 = gr.Plot(label="Graphique")
+
+    with gr.Tab("Code on your file"):
+        with gr.Accordion("Input DataFrame Preview", open=False):
+            df_input = gr.DataFrame(interactive=False)
+        gr.Markdown("```python\ndf = pd.read_excel(YOUR_FILE)\n```")
+        cd_code = gr.Code(value="# Create a copy of the original DataFrame\nnew_df = df.copy()\n\n# Add a new column to the copy\nnew_df['NewColumn'] = 'New Value'", language='python')
+        gr.Markdown("```python\new_df.to_excel(YOUR_NEW_FILE)\nreturn YOUR_NEW_FILE\n```")
+
+        btn_run_code = gr.Button()
+        df_output = gr.DataFrame(interactive=False)
+        btn_export_df = gr.Button('Export df as excel')
+    st_filename = gr.State()
+    
     with gr.Accordion("Excel Preview", open=False):
           df_output = gr.DataFrame()
     fi_excel = gr.File(label="Excel File")
@@ -157,12 +171,12 @@ with gr.Blocks() as demo:
     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])
     
-
+    # 3GPP scraping
     btn_search_status.click(extract_statuses, inputs=dd_url, outputs=dd_status)
     btn_search.click(browse_folder, inputs=dd_url, outputs=dd_url)
     dd_url.change(browse_folder, inputs=dd_url, outputs=dd_url)
     #fi_excel.change(get_expert,inputs=fi_excel, outputs=dd_exp)
-    fi_excel.change(get_columns, inputs=[fi_excel], outputs=[dd_source_ask, dd_source_class, dd_label1, dd_label2, dd_searchcol, df_output])
+    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])
     
@@ -171,6 +185,10 @@ with gr.Blocks() as demo:
     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])
+
+    btn_run_code.click(run_code, inputs=[fi_excel, cd_code], outputs=[df_output])
+    btn_export_df.click(export_df, inputs=[df_output, st_filename], outputs=fi_excel)
+    
     
     btn_overall.click(generate_company_chart,inputs=[fi_excel], outputs=[plt_chart])
     btn_overall.click(status_chart,inputs=[fi_excel], outputs=[plt_chart2])