Update app.py

app.py

@@ -2554,180 +2554,179 @@ def load_nps_and_r2(file_path):
 
 
 
-
-
 with gr.Blocks(css=css, js=js_func) as demo:
-   gr.HTML("""
-   <style>
-   #trust_driver_img, #nps_driver_img {
-       display: inline-block;
-       width: 49%;
-       margin-right: 1%;
-       vertical-align: top;
-   }
-   </style>
-   """)
-
-   # Title and intro
-   with gr.Column():
-       gr.Markdown("""
-           <h2 style="text-align: center; font-size: 2.25rem; font-weight: 600;">
-               TrustLogic AI Demo
-           </h2>
-       """)
-       gr.Markdown("### Experience TrustLogic AI")
-       gr.Markdown("Experience how quick and easy it is to build the right trust based on insight with our TrustLogic AI")
-
-   # Dataset selection buttons
-   with gr.Column():
-       with gr.Row():
-           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)
-   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")
-       bot = gr.Chatbot(placeholder=placeholder_text)
-
-   # Button to show data
-   show_data_btn = gr.Button("Show the data", variant="primary", size="lg")
-   
-   # Data analysis section (initially hidden)
-   with gr.Column(visible=False) as data_section:
-       gr.Markdown("### Uncover the underlying data used for the prompt")
-       gr.Markdown("Quickly identify what drives your NPS and trust across different segments using the automated analyser.")
-       gr.Markdown("""
-           <span style="font-size:15px;">Volkswagen Example</span><br>
-           As a default, the analysis displays <strong>Volkswagen Owner</strong> results. 
-           To trigger the analysis for <strong>Prospects</strong>, toggle to 'VW Prospects'.
-       """)
-   
-       with gr.Row(equal_height=True):
-           with gr.Column(scale=1):
-               nps_img_output = gr.HTML()
-
-               
-   
-           with gr.Column(scale=1):
-               trust_score_output = gr.HTML()
-   
-           with gr.Column(scale=1):
-               gr.Markdown("""<div style='text-align: center;'><h3>How much of your NPS is determined by TrustLogic®</h3></div>""")
-               trust_r2_img = gr.HTML()
-   
-               
-       with gr.Column():
-           outputs = reset_outputs()
-
-       # 🔁 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"])
-       selected_dataset_ai = gr.State(value="VW Owners.xlsx")  # ✅ Matches dictionary key
+    gr.HTML("""
+    <style>
+    #trust_driver_img, #nps_driver_img {
+        display: inline-block;
+        width: 49%;
+        margin-right: 1%;
+        vertical-align: top;
+    }
+    </style>
+    """)
+
+    # Title and intro
+    with gr.Column():
+        gr.Markdown("""
+            <h2 style="text-align: center; font-size: 2.25rem; font-weight: 600;">
+                TrustLogic AI Demo
+            </h2>
+        """)
+        gr.Markdown("### Experience TrustLogic AI")
+        gr.Markdown("Experience how quick and easy it is to build the right trust based on insight with our TrustLogic AI")
+
+    # Dataset selection buttons
+    with gr.Column():
+        with gr.Row():
+            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)
+    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")
+        bot = gr.Chatbot(placeholder=placeholder_text)
+
+    # Button to show data
+    show_data_btn = gr.Button("Show the data", variant="primary", size="lg")
+    
+    # Data analysis section (initially hidden)
+    with gr.Column(visible=False) as data_section:
+        gr.Markdown("### Uncover the underlying data used for the prompt")
+        gr.Markdown("Quickly identify what drives your NPS and trust across different segments using the automated analyser.")
+        gr.Markdown("""
+            <span style="font-size:15px;">Volkswagen Example</span><br>
+            As a default, the analysis displays <strong>Volkswagen Owner</strong> results. 
+            To trigger the analysis for <strong>Prospects</strong>, toggle to 'VW Prospects'.
+        """)
+
+        with gr.Row(equal_height=True):
+            with gr.Column(scale=1):
+                nps_img_output = gr.HTML()
+
+            with gr.Column(scale=1):
+                trust_score_output = gr.HTML()
+
+            with gr.Column(scale=1):
+                gr.Markdown("""<div style='text-align: center;'><h3>How much of your NPS is determined by TrustLogic®</h3></div>""")
+                trust_r2_img = gr.HTML()
+
+        with gr.Column():
+            outputs = reset_outputs()
+
+    # 🔁 States - MOVED OUTSIDE OF DATA SECTION
+    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"])
+    selected_dataset_ai = gr.State(value="VW Owners.xlsx")  # ✅ Matches dictionary key
+    chat_history = gr.State(value=[])
+
+    # Show/hide data section functionality
+    data_visible_state = gr.State(value=False)
+    
+    def toggle_data_section(visible):
+        new_visible = not visible
+        button_text = "Hide the data" if new_visible else "Show the data"
+        return gr.update(visible=new_visible), button_text, new_visible
+    
+    def set_vw_owners():
+        return "VW Owners.xlsx"
+    
+    def set_vw_prospects():
+        return "Volkswagen Non Customers.xlsx"
+
+    # Event handlers - MOVED OUTSIDE OF DATA SECTION
+    show_data_btn.click(
+        fn=toggle_data_section,
+        inputs=data_visible_state,
+        outputs=[data_section, show_data_btn, data_visible_state]
+    )
+    
+    submit_button.click(
+        fn=chatbot_response,
+        inputs=[prompt_textbox, chat_history, selected_dataset_ai],
+        outputs=[bot, chat_history]
+    )
 
-       chat_history = gr.State(value=[])
+    ## All widget functions here ##
 
-       # Show/hide data section functionality
-       data_visible_state = gr.State(value=False)
-       
-       def toggle_data_section(visible):
-           new_visible = not visible
-           button_text = "Hide the data" if new_visible else "Show the data"
-           return gr.Column.update(visible=new_visible), button_text, new_visible
-       
-       show_data_btn.click(
-           fn=toggle_data_section,
-           inputs=data_visible_state,
-           outputs=[data_section, show_data_btn, data_visible_state]
-       )
-       
-       submit_button.click(
-           fn=chatbot_response,
-           inputs=[prompt_textbox, chat_history, selected_dataset_ai],
-           outputs=[bot, chat_history]
-       )
+    vw_customers_btn.click(
+        fn=display_trust_score_1,  
+        inputs=[],
+        outputs=trust_score_output,
+    )
+    
+    vw_prospects_btn.click(
+        fn=display_trust_score_2,  
+        inputs=[],
+        outputs=trust_score_output,
+    )
+    
+    vw_customers_btn.click(
+        fn=process_examples,
+        inputs=[vw_customers_state],
+        outputs=outputs,
+    )
+    
+    vw_prospects_btn.click(
+        fn=process_examples,
+        inputs=[vw_prospects_state],
+        outputs=outputs,
+    )
+    
+    vw_customers_btn.click(
+        fn=set_vw_owners,
+        inputs=[],
+        outputs=selected_dataset_ai
+    )
+    
+    vw_prospects_btn.click(
+        fn=set_vw_prospects,
+        inputs=[],
+        outputs=selected_dataset_ai
+    )
+    
+    vw_customers_btn.click(
+        fn=lambda f: vwload_nps_and_r2(f[0]),
+        inputs=[vw_customers_state11],
+        outputs=[nps_img_output, trust_r2_img],
+    )
+    
+    vw_prospects_btn.click(
+        fn=lambda f: load_nps_and_r2(f[0]),
+        inputs=[vw_prospects_state12],
+        outputs=[nps_img_output, trust_r2_img],
+    )
+            
+    # Load functions for initial state
+    demo.load(
+        fn=lambda f: vwload_nps_and_r2(f[0]),
+        inputs=[vw_customers_state11],
+        outputs=[nps_img_output, trust_r2_img],
+    )
+    
+    demo.load(
+        fn=display_trust_score_1, 
+        inputs=[], 
+        outputs=trust_score_output
+    )
+    
+    demo.load(
+        fn=process_examples, 
+        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
 
 
-       ## All widget functions here ##
 
-       vw_customers_btn.click(
-           fn=display_trust_score_1,  
-           inputs=[],
-           outputs=trust_score_output,
-       )
-       
-       vw_prospects_btn.click(
-           fn=display_trust_score_2,  
-           inputs=[],
-           outputs=trust_score_output,
-       )
-       
-       vw_customers_btn.click(
-           fn=process_examples,
-           inputs=[vw_customers_state],
-           outputs= outputs,
-       )
-       
-       vw_prospects_btn.click(
-           fn=process_examples,
-           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=set_vw_owners,
-           inputs=[],
-           outputs=selected_dataset_ai
-       )
-       
-       vw_prospects_btn.click(
-           fn=set_vw_prospects,
-           inputs=[],
-           outputs=selected_dataset_ai
-       )
-       vw_customers_btn.click(
-           fn=lambda f: vwload_nps_and_r2(f[0]),
-           inputs=[vw_customers_state11],
-           outputs=[nps_img_output, trust_r2_img],
-       )
-       
-       vw_prospects_btn.click(
-           fn=lambda f: load_nps_and_r2(f[0]),
-           inputs=[vw_prospects_state12],
-           outputs=[nps_img_output, trust_r2_img],
-       )
                
-       
-       demo.load(
-           fn=lambda f: vwload_nps_and_r2(f[0]),
-           inputs=[vw_customers_state11],
-           outputs=[nps_img_output, trust_r2_img],
-       )
-       demo.load(
-       fn=display_trust_score_1, 
-       inputs=[], 
-       outputs=trust_score_output
-       )
-       
-       demo.load(
-           fn=process_examples, 
-           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
-
-
+