Update app.py

app.py

@@ -3420,6 +3420,7 @@ def crear_app():
                                     return gr.update(choices=listado)
                             
                                 categorias_mass_state_pmt = gr.State(value=[]) # Estado para almacenar categorías disponibles
+                                iniciativas_pmt = gr.State(value=[])
                                 categorias_mass_pmt = gr.Dropdown(
                                                         choices=categorias_mass_state_pmt.value, multiselect=False, label="Categorías"#, info="Selecciona una categoria para aplicar el filtro."
                                                     )
@@ -3434,11 +3435,25 @@ def crear_app():
                                     elementos = df[categoria].dropna().unique().tolist()
                                     # return ", ".join(elementos)
                                     return gr.update(choices=elementos)
+
+                                def listar_iniciativas(file): 
+                                    if file is None: # or categoria not in file.columns:
+                                        return "No hay datos disponibles"
+                                    df = pd.read_excel(file)
+                                    if 'iniciativa' not in df.columns:
+                                        return f"No se encontró 'iniciativa' en el archivo"
+                                    elementos = df['iniciativa'].dropna().unique().tolist()
+                                    # return ", ".join(elementos)
+                                    return gr.update(choices=elementos)
                                 
                                 # filtro_categoria_state = gr.State(value=[]) # Estado para almacenar categorías seleccionadas para filtro
                                 filtro_categoria_pmt= gr.Dropdown(
                                                         choices=[], multiselect=True, interactive=True, label="Elementos de categoría", info="Selecciona una o varias opciones para aplicar el análisis."
                                                     )
+
+                                file_output.change(listar_iniciativas,
+                                                     file_output,
+                                                     iniciativas_pmt)
                                 
                                 bttn_filtro_categoria_pmt = gr.Button("Aplicar filtro por categoría", variant="secondary")
                                 
@@ -3478,8 +3493,57 @@ def crear_app():
                                                 plot_mass_metas_pmt = gr.Plot(label="Ranking Global METAS")
                                             with gr.Column(scale=1):
                                                 exp_mass_metas_pmt = gr.Markdown()
-                
 
+
+                            ## AJUSTE POR FILTROS DE CATEGORIA
+                                with gr.Column():
+                                    bttn_filtro_categoria.click(
+                                        fn=lambda df, categoria, elementos: df[df[categoria].isin(elementos)] if df is not None else None,
+                                        inputs=[base_ods_mass_mix, categorias_mass, filtro_categoria],
+                                        outputs=[base_ods_mass_mix_cat]
+                                    )
+
+                                    bttn_filtro_categoria.click(
+                                        fn=lambda df, categoria, elementos: df[df[categoria].isin(elementos)] if df is not None else None,
+                                        inputs=[base_meta_mass_mix, categorias_mass, filtro_categoria],
+                                        outputs=[base_meta_mass_mix_cat]
+                                    )
+
+                                    bttn_filtro_categoria.click(
+                                        fn=lambda df, categoria, elementos: df[df[categoria].isin(elementos)] if df is not None else None,
+                                        inputs=[base_indicadores_mass_mix, categorias_mass, filtro_categoria],
+                                        outputs=[base_indicadores_mass_mix_cat]
+                                    )        
+
+                                    base_ods_mass_mix_cat.change(
+                                        fn=tab_viz19,
+                                        inputs=[base_ods_mass_mix_cat, base_meta_mass_mix_cat, base_indicadores_mass_mix_cat],
+                                        outputs=[html19_1, html19_2, html19_3, exp19]
+                                    )
+
+                                    base_meta_mass_mix_cat.change(
+                                        fn=tab_viz19,
+                                        inputs=[base_ods_mass_mix_cat, base_meta_mass_mix_cat, base_indicadores_mass_mix_cat],
+                                        outputs=[html19_1, html19_2, html19_3, exp19]
+                                    )
+
+                                    base_indicadores_mass_mix_cat.change(
+                                        fn=tab_viz19,
+                                        inputs=[base_ods_mass_mix_cat, base_meta_mass_mix_cat, base_indicadores_mass_mix_cat],
+                                        outputs=[html19_1, html19_2, html19_3, exp19]
+                                    )
+
+
+
+                            # btn_mass_emb.click(
+                            #     fn=procesar_archivo,
+                            #     inputs=[file_output],
+                            #     outputs=[output_ods_mass, output_meta_mass, output_indicadores_mass,categorias_mass_state, msj_procesamiento_mass],
+                            #     show_progress=True
+                            # )
+
+                            btn_mass_emb_pmt.click(procesar_lote_llm, iniciativas_pmt, [base_ods_mass_mix_pmt,base_meta_mass_mix_pmt,base_indicadores_mass_mix_pmt], show_progress=True)
+                            
                 with gr.Tab("MASIVA CON PROMPTING", visible=False):
                     gr.Markdown("Aquí puedes processar múltiples iniciativas con enfoque de prompting. (Usa la plantilla proporcionada)")