Update app.py

app.py

@@ -2406,23 +2406,25 @@ from sklearn.metrics import r2_score
 
 import base64
 
+
+
 def reset_variables():
-    """
-    Reset global variables to handle dataset changes.
-    """
-    global selected_dataset_ai
-    selected_dataset_ai = None
-        
+   """
+   Reset global variables to handle dataset changes.
+   """
+   global selected_dataset_ai
+   selected_dataset_ai = None
+       
 def update_ai_dataset_selection(selection):
-    global selected_dataset_ai  # Ensure the global variable is updated
-    reset_variables()
-    chat_history.clear()
-
-    if selection == "VW Owners":
-        selected_dataset_ai = vw_customers_state.value[0]  # Use the customer dataset
-    elif selection == "VW Prospects":
-        selected_dataset_ai = vw_prospects_state.value[0]  # Use the prospects dataset
-    return selected_dataset_ai
+   global selected_dataset_ai  # Ensure the global variable is updated
+   reset_variables()
+   chat_history.clear()
+
+   if selection == "VW Owners":
+       selected_dataset_ai = vw_customers_state.value[0]  # Use the customer dataset
+   elif selection == "VW Prospects":
+       selected_dataset_ai = vw_prospects_state.value[0]  # Use the prospects dataset
+   return selected_dataset_ai
 
 placeholder_text = """"""
 
@@ -2445,85 +2447,85 @@ Style:
 """
 # Text input box for the user to enter their prompt
 prompt_textbox = gr.Textbox(
-    value=predefined_prompt,
-    scale=4,
-    label="Insert your prompt",
-    visible=True,
+   value=predefined_prompt,
+   scale=4,
+   label="Insert your prompt",
+   visible=True,
 )
 submit_button = gr.Button("Submit")
 bot = gr.Chatbot(placeholder=placeholder_text)
 
 js_func = """
 function refresh() {
-    // Force light theme if not already
-    const url = new URL(window.location);
-    if (url.searchParams.get('__theme') !== 'light') {
-        url.searchParams.set('__theme', 'light');
-        window.location.href = url.href;
-        return;
-    }
-
-    // Highlight toggle
-    const btn1 = document.querySelector("#vw_customers_btn");
-    const btn2 = document.querySelector("#vw_prospects_btn");
-
-    if (btn1 && btn2) {
-        btn1.classList.add("active-btn"); // Default on load
-
-        btn1.addEventListener("click", () => {
-            btn1.classList.add("active-btn");
-            btn2.classList.remove("active-btn");
-        });
-
-        btn2.addEventListener("click", () => {
-            btn2.classList.add("active-btn");
-            btn1.classList.remove("active-btn");
-        });
-    }
+   // Force light theme if not already
+   const url = new URL(window.location);
+   if (url.searchParams.get('__theme') !== 'light') {
+       url.searchParams.set('__theme', 'light');
+       window.location.href = url.href;
+       return;
+   }
+
+   // Highlight toggle
+   const btn1 = document.querySelector("#vw_customers_btn");
+   const btn2 = document.querySelector("#vw_prospects_btn");
+
+   if (btn1 && btn2) {
+       btn1.classList.add("active-btn"); // Default on load
+
+       btn1.addEventListener("click", () => {
+           btn1.classList.add("active-btn");
+           btn2.classList.remove("active-btn");
+       });
+
+       btn2.addEventListener("click", () => {
+           btn2.classList.add("active-btn");
+           btn1.classList.remove("active-btn");
+       });
+   }
 }
 """
 
 
 css = """
 #vw_customers_btn, #vw_prospects_btn {
-    border: 2px solid #ccc;
-    padding: 10px 20px;
-    font-size: 16px;
-    font-weight: 500;
-    border-radius: 6px;
-    color: #333;
-    background-color: #fff;
-    margin: 0 5px;
-    transition: all 0.25s ease;
-    min-width: 140px;
-    text-align: center;
+   border: 2px solid #ccc;
+   padding: 10px 20px;
+   font-size: 16px;
+   font-weight: 500;
+   border-radius: 6px;
+   color: #333;
+   background-color: #fff;
+   margin: 0 5px;
+   transition: all 0.25s ease;
+   min-width: 140px;
+   text-align: center;
 }
 
 /* Highlighted (clicked) button — with teal text */
 .active-btn {
-    background-color: #e0f7f5 !important;  /* Light teal background */
-    color: teal !important;               /* Teal text */
-    border-color: teal !important;
-    box-shadow: 0 0 6px rgba(0, 128, 128, 0.3);
+   background-color: #e0f7f5 !important;  /* Light teal background */
+   color: teal !important;               /* Teal text */
+   border-color: teal !important;
+   box-shadow: 0 0 6px rgba(0, 128, 128, 0.3);
 }
 
 /* Optional hover effect */
 #vw_customers_btn:hover, #vw_prospects_btn:hover {
-    background-color: #f9f9f9;
+   background-color: #f9f9f9;
 }
 
 .gr-button:has(svg) {
-    display: none !important;
+   display: none !important;
 }
 .gr-image-label {
-    display: none !important;
+   display: none !important;
 }
 #chat_container {
-    max-height: 900px;
-    overflow-y: auto;
-    margin-top: 0 !important;
-    margin-bottom: 0 !important;
-    padding-bottom: 0 !important;
+   max-height: 900px;
+   overflow-y: auto;
+   margin-top: 0 !important;
+   margin-bottom: 0 !important;
+   padding-bottom: 0 !important;
 }
 """
 
@@ -2531,179 +2533,198 @@ css = """
 
 
 def highlight_button(button_name):
-    return gr.Button.update(variant="primary")
+   return gr.Button.update(variant="primary")
 
 # 🧠 Chatbot backend
 
 
 
 def vwload_nps_and_r2(file_path):
-    nps_img = calculate_nps_image_from_excel(file_path)
-    r2_img = vwcalculate_r2_image_from_excel(file_path)
-    return nps_img,r2_img
+   nps_img = calculate_nps_image_from_excel(file_path)
+   r2_img = vwcalculate_r2_image_from_excel(file_path)
+   return nps_img,r2_img
 
 
 
 
 def load_nps_and_r2(file_path):
-    nps_img = calculate_nps_image_from_excel(file_path)
-    r2_img = calculate_r2_image_from_excel(file_path)
+   nps_img = calculate_nps_image_from_excel(file_path)
+   r2_img = calculate_r2_image_from_excel(file_path)
 
-    return nps_img, r2_img
+   return nps_img, r2_img
 
 
 
 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("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.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")
-    
-        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()
 
-        chat_history = gr.State(value=[])
+               
+   
+           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()
 
-    # 🧠 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.")
+       # 🔁 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
 
-        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)
+       chat_history = gr.State(value=[])
 
-        
-        
-        submit_button.click(
-            fn=chatbot_response,
-            inputs=[prompt_textbox, chat_history, selected_dataset_ai],
-            outputs=[bot, chat_history]
-        )
+       # 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 ##
+       ## 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,
+       )
+       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
-        )
-        
+       
+       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