Update app.py

app.py

@@ -2426,6 +2426,7 @@ def load_nps_and_r2(file_path):
 
     return nps_img, r2_img
 
+
 with gr.Blocks(css=css, js=js_func) as demo:
    gr.HTML("""
    <style>
@@ -2438,11 +2439,11 @@ with gr.Blocks(css=css, js=js_func) as demo:
    </style>
    """)
 
-   # 🔁 States (moved to top)
+   # 🔁 States
    vw_customers_state11 = gr.State(value=["example_files/VW Owners.xlsx"])
    vw_prospects_state12 = gr.State(value=["example_files/Volkswagen Non Customers.xlsx"])
-   vw_customers_state = gr.State(value=["VW Owners.xlsx"])
-   vw_prospects_state = gr.State(value=["Volkswagen Non Customers.xlsx"])
+   vw_customers_state = gr.State(value=["VW Owners.xlsx"])  # Just the filename
+   vw_prospects_state = gr.State(value=["Volkswagen Non Customers.xlsx"])  # Just the filename
    selected_dataset_ai = gr.State(value="VW Owners.xlsx")
    chat_history = gr.State(value=[])
    data_visible_state = gr.State(value=False)
@@ -2463,7 +2464,7 @@ with gr.Blocks(css=css, js=js_func) as demo:
            vw_customers_btn = gr.Button("VW Owners", elem_id="vw_customers_btn")
            vw_prospects_btn = gr.Button("VW Prospects", elem_id="vw_prospects_btn")
 
-   # 🧠 Chat section (moved to top)
+   # 🧠 Chat section
    with gr.Column(elem_id="chat_container"):
        prompt_textbox = gr.Textbox(value=predefined_prompt, scale=4, label="Insert your prompt", visible=True)
        submit_button = gr.Button("Submit")
@@ -2493,9 +2494,37 @@ with gr.Blocks(css=css, js=js_func) as demo:
                gr.Markdown("""<div style='text-align: center;'><h3>How much of your NPS is determined by TrustLogic®</h3></div>""")
                trust_r2_img = gr.HTML()
    
-       # Create placeholder components for dynamic outputs
-       with gr.Column() as dynamic_outputs_container:
-           pass
+       # Move outputs components inside data_section
+       markdown_3 = gr.Markdown(visible=True)
+       markdown_4 = gr.Markdown(visible=True)
+       trust_plot = gr.Image(value=None, label="Trust Drivers", visible=True)
+       nps_plot = gr.Image(value=None, label="NPS Drivers", visible=True)
+       loyalty_plot = gr.Image(value=None, label="Loyalty Drivers", visible=True)
+       consideration_plot = gr.Image(value=None, label="Consideration Drivers", visible=True)
+       satisfaction_plot = gr.Image(value=None, label="Satisfaction Drivers", visible=True)
+       summary_text = gr.Textbox(value=None, label="Analysis Summary", visible=True)
+       markdown_5 = gr.Markdown(visible=True)
+       markdown_6 = gr.Markdown(visible=True)
+       df_builder_pivot = gr.Dataframe(value=None, label="", visible=True)
+
+   # Create outputs list for functions
+   outputs = [
+       markdown_3, markdown_4, trust_plot, nps_plot, loyalty_plot, 
+       consideration_plot, satisfaction_plot, summary_text, 
+       markdown_5, markdown_6, df_builder_pivot
+   ]
+
+   # Modified process_examples to return values
+   def process_examples_values(file_name):
+       output_components = process_examples(file_name)
+       # Extract values from components
+       values = []
+       for comp in output_components[:11]:  # Only first 11 components
+           if hasattr(comp, 'value'):
+               values.append(comp.value)
+           else:
+               values.append(None)
+       return values
 
    # Show/hide data section functionality
    def toggle_data_section(visible):
@@ -2529,32 +2558,32 @@ with gr.Blocks(css=css, js=js_func) as demo:
        outputs=trust_score_output,
    )
    
-   # Modified to handle component rendering
-   @gr.render(inputs=[vw_customers_state, vw_prospects_state, selected_dataset_ai], triggers=[vw_customers_btn.click, vw_prospects_btn.click, demo.load])
-   def render_dynamic_outputs(vw_cust_state, vw_pros_state, selected_dataset):
-       # Determine which dataset to use
-       if selected_dataset == "VW Owners.xlsx":
-           file_inputs = vw_cust_state
-       else:
-           file_inputs = vw_pros_state
-           
-       # Get the components from variable_outputs
-       output_components = variable_outputs(file_inputs)
-       
-       # Render the components
-       with gr.Column():
-           for component in output_components:
-               if component is not None and hasattr(component, 'render'):
-                   component.render()
+   vw_customers_btn.click(
+       fn=process_examples_values,
+       inputs=[vw_customers_state],
+       outputs=outputs,
+   )
+   
+   vw_prospects_btn.click(
+       fn=process_examples_values,
+       inputs=[vw_prospects_state],
+       outputs=outputs,
+   )
+   
+   def set_vw_owners():
+       return "VW Owners.xlsx"
    
+   def set_vw_prospects():
+       return "Volkswagen Non Customers.xlsx"
+
    vw_customers_btn.click(
-       fn=lambda: "VW Owners.xlsx",
+       fn=set_vw_owners,
        inputs=[],
        outputs=selected_dataset_ai
    )
    
    vw_prospects_btn.click(
-       fn=lambda: "Volkswagen Non Customers.xlsx",
+       fn=set_vw_prospects,
        inputs=[],
        outputs=selected_dataset_ai
    )
@@ -2582,9 +2611,16 @@ with gr.Blocks(css=css, js=js_func) as demo:
        inputs=[], 
        outputs=trust_score_output
    )
+   
+   demo.load(
+       fn=process_examples_values, 
+       inputs=[vw_customers_state], 
+       outputs=outputs
+   )
 
 try:
    demo.launch(server_name="0.0.0.0")
 except Exception as e:
    logger.error(f"Error launching Gradio app: {e}")
-   raise e
+   raise e
+