Update app.py

app.py

@@ -2553,6 +2553,7 @@ def load_nps_and_r2(file_path):
 
 
 
+
 with gr.Blocks(css=css, js=js_func) as demo:
     gr.HTML("""
     <style>
@@ -2572,14 +2573,31 @@ with gr.Blocks(css=css, js=js_func) as demo:
                 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’.
+            To trigger the analysis for <strong>Prospects</strong>, toggle to 'VW Prospects'.
         """)
 
-    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")
@@ -2587,8 +2605,6 @@ with gr.Blocks(css=css, js=js_func) as demo:
         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()
@@ -2597,113 +2613,118 @@ 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()
     
-                
         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=[])
-
-    # 🧠 Chat section
-    with gr.Column(elem_id="chat_container"):
-        gr.Markdown("###  Test-drive the results in the TrustLogicAI")
-        gr.Markdown("Our AI uses the 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)
-
-        
-        
-        submit_button.click(
-            fn=chatbot_response,
-            inputs=[prompt_textbox, chat_history, selected_dataset_ai],
-            outputs=[bot, chat_history]
-        )
-
+    # 🔁 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]
+    )
 
+    show_data_btn.click(
+        fn=show_analysis,
+        inputs=[],
+        outputs=[analysis_section]
+    )
 
-        ## All widget functions here ##
+    ## 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=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
+    )
 
-        
-        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 
+    raise e
+
+