Update app.py

app.py

@@ -2554,177 +2554,180 @@ 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;">
-                What drives your NPS and trust?
-            </h2>
-        """)
-        gr.Markdown("Experience how quick and easy it is to build the right trust based on insight with our TrustLogic AI")
-
-    # 🧠 AI Demo Section - MOVED TO TOP
-    with gr.Column(elem_id="chat_container"):
-        gr.Markdown("### TrustLogicAI Demo")
-        gr.Markdown("Our AI uses analysis results to generate trust-optimised content.")
-
-        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 reveal data analysis
-    with gr.Column():
-        gr.Markdown("After results: Uncover the underlying data used for the prompt.")
-        show_data_btn = gr.Button("Show the data <VW Customer Data>", visible=False)  # Initially hidden
-        
-    # Analysis Results Section - MOVED BELOW CHAT
-    with gr.Column(visible=False, elem_id="analysis_section") as analysis_section:
-        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():
-            vw_customers_btn = gr.Button("VW Owners", elem_id="vw_customers_btn")
-            vw_prospects_btn = gr.Button("VW Prospects", elem_id="vw_prospects_btn")
-    
-        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
-    chat_history = gr.State(value=[])
-
-    # Modified chatbot response to show data button after interaction
-    def chatbot_response_with_reveal(prompt, history, dataset):
-        response, updated_history = chatbot_response(prompt, history, dataset)
-        return response, updated_history, gr.update(visible=True)  # Show the data button
-
-    # Function to reveal analysis section
-    def show_analysis():
-        return gr.update(visible=True)
-
-    # Function definitions for button actions
-    def set_vw_owners():
-        return "VW Owners.xlsx"
-    
-    def set_vw_prospects():
-        return "Volkswagen Non Customers.xlsx"
-
-    # Event handlers
-    submit_button.click(
-        fn=chatbot_response_with_reveal,
-        inputs=[prompt_textbox, chat_history, selected_dataset_ai],
-        outputs=[bot, chat_history, show_data_btn]
-    )
+   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()
 
-    show_data_btn.click(
-        fn=show_analysis,
-        inputs=[],
-        outputs=[analysis_section]
-    )
+               
+   
+           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
+
+       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.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]
+       )
 
-    ## 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,
-    )
-    
-    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 (only when analysis section is visible)
-    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
-    )
 
+       ## 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")
+   demo.launch(server_name="0.0.0.0")
 except Exception as e:
-    logger.error(f"Error launching Gradio app: {e}")
-    raise e
+   logger.error(f"Error launching Gradio app: {e}")
+   raise e