Update app.py

app.py

@@ -2408,23 +2408,25 @@ import base64
 
 
 
+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 = """"""
 
@@ -2447,85 +2449,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;
 }
 """
 
@@ -2533,199 +2535,196 @@ 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("### 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
-
-       chat_history = gr.State(value=[])
+    gr.HTML("""
+    <style>
+    #trust_driver_img, #nps_driver_img {
+        display: inline-block;
+        width: 49%;
+        margin-right: 1%;
+        vertical-align: top;
+    }
+    </style>
+    """)
+
+    # 🔁 States (moved to top)
+    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=[])
+    data_visible_state = gr.State(value=False)
+
+    # 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/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]
-       )
+                
+    
+            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()
+
+    # Show/hide data section functionality
+    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
+    
+    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