Update PanelInterface.py

PanelInterface.py

@@ -2,43 +2,35 @@
 import gradio as gr
 import logging
 
-def chatbot_interface(message, history=None):
-    if history is None:
-        history = []
-    response = f"You asked: {message}"
-    history.append({"role": "user", "content": message})
-    history.append({"role": "assistant", "content": response})
-    return history, "", history
-
 def build_interface():
     logging.info("Building Gradio interface...")
 
     panelists = [
-        {"ID": "P1", "Name": "Sourav", "Description": "Student from Kolkata", "Bio": "Sourav is a 25-year-old sociology student from Kolkata."},
-        {"ID": "P2", "Name": "Anjali", "Description": "College student from Chennai", "Bio": "Anjali is a 17-year-old girl from Chennai, aspiring to be a psychologist."},
-        {"ID": "P3", "Name": "Ravi", "Description": "Teen from Hyderabad", "Bio": "Ravi is a high schooler passionate about cricket and coding."},
-        {"ID": "P4", "Name": "Divya", "Description": "Engineering student from Trichy", "Bio": "Divya is 21, studying electronics in Trichy, with interests in AI and robotics."}
+        {"ID": "Panellist1", "Name": "Sourav", "Description": "25 year old male from Kolkata", "Bio": "Sourav's bio."},
+        {"ID": "Panellist2", "Name": "Divya", "Description": "17 year old female from Chennai", "Bio": "Divya's bio."},
+        {"ID": "Panellist3", "Name": "Rahul", "Description": "16 year old male from Hyderabad", "Bio": "Rahul's bio."},
+        {"ID": "Panellist4", "Name": "Meena", "Description": "21 year old female from Trichy", "Bio": "Meena's bio."}
     ]
 
-    panel_html = ""
-    bios_dict = {}
-
-    for p in panelists:
-        panel_html += f'<p><a href="#" onclick="showModal(event, \"{p["ID"]}\")"><b>{p["Name"]}</b></a> – {p["Description"]}</p>'
-        bios_dict[p["ID"]] = p["Bio"]
-
-    bios_js_object = str(bios_dict).replace("'", "\"")
+    bios_js_object = {p["ID"]: p["Bio"] for p in panelists}
+    panel_html = "".join(
+        f'<p><a href="#" onclick="showModal(event, \'{p["ID"]}\')"><b>{p["Name"]}</b></a> – {p["Description"]}</p>'
+        for p in panelists
+    )
 
     with gr.Blocks(css="assets/custom.css") as demo:
+        logging.info("Adding logo and welcome message...")
         with gr.Row(elem_classes="logo-row"):
-            gr.Image("predata-logo.png", height=200, show_label=False, elem_id="logo")
+            gr.Image("static/predata-logo.png", height=200, show_label=False, elem_id="logo")
 
         with gr.Row(elem_classes="welcome-section"):
             gr.Markdown("""
             # Welcome to PreData.AI's Market Research Panel [Demo]
             ## Introducing our AI-powered panel:
-            - Ask a question by typing into the box below.
-            - You can address a question to a **specific panellist** or the entire group.
+            - This panel consists of some **AI-powered agents** who will answer your questions.
+            - Ask them a question by **typing into the input box below**.
+            - Feel free to address a question to a **specific panellist**.
+            - If no panellist is specified, all of the panellists will respond.
             """)
 
         with gr.Row():
@@ -46,25 +38,26 @@ def build_interface():
                 gr.HTML(f"""
                 <div style='text-align: left;'>
                     <h2>Our Panellists</h2>
+                    <i>Click on our panellist's name so they can introduce themselves</i><br>
                     {panel_html}
                 </div>
-                <div id='bioModal' style='display:none;position:fixed;top:50%;left:50%;transform:translate(-50%, -50%);background-color:white;border:1px solid #ccc;box-shadow:0px 4px 6px rgba(0,0,0,0.2);width:400px;max-height:300px;padding:20px;text-align:left;z-index:1000;overflow-y:auto;'>
+                <div id='bioModal' style='display: none; position: fixed; top: 50%; left: 50%; transform: translate(-50%, -50%); background-color: white; border: 1px solid #ccc; box-shadow: 0px 4px 6px rgba(0, 0, 0, 0.2); width: 400px; max-height: 300px; padding: 20px; text-align: left; z-index: 1000; overflow-y: auto;'>
                     <p id='bioContent'></p>
-                    <button onclick='closeModal()' style='margin-top:10px;padding:5px 10px;background-color:#007bff;color:white;border:none;border-radius:5px;cursor:pointer;'>Close</button>
+                    <button onclick='closeModal()' style='margin-top: 10px; padding: 5px 10px; background-color: #007bff; color: white; border: none; border-radius: 5px; cursor: pointer;'>Close</button>
                 </div>
-                <div id='modalBackdrop' style='display:none;position:fixed;top:0;left:0;width:100%;height:100%;background-color:rgba(0,0,0,0.5);z-index:999;' onclick='closeModal()'></div>
+                <div id='modalBackdrop' style='display: none; position: fixed; top: 0; left: 0; width: 100%; height: 100%; background-color: rgba(0, 0, 0, 0.5); z-index: 999;' onclick='closeModal()'></div>
                 <script>
-                const bios = {bios_js_object};
-                function showModal(event, name) {{
-                    event.preventDefault();
-                    document.getElementById('bioContent').innerText = bios[name] || 'Bio not found.';
-                    document.getElementById('bioModal').style.display = 'block';
-                    document.getElementById('modalBackdrop').style.display = 'block';
-                }}
-                function closeModal() {{
-                    document.getElementById('bioModal').style.display = 'none';
-                    document.getElementById('modalBackdrop').style.display = 'none';
-                }}
+                    const bios = {str(bios_js_object).replace("'", '"')};
+                    function showModal(event, name) {{
+                        event.preventDefault();
+                        document.getElementById('bioContent').innerText = bios[name] || 'Bio not found.';
+                        document.getElementById('bioModal').style.display = 'block';
+                        document.getElementById('modalBackdrop').style.display = 'block';
+                    }}
+                    function closeModal() {{
+                        document.getElementById('bioModal').style.display = 'none';
+                        document.getElementById('modalBackdrop').style.display = 'none';
+                    }}
                 </script>
                 """)
 
@@ -73,12 +66,45 @@ def build_interface():
                 msg = gr.Textbox(placeholder="Ask your question to the panel here...")
                 history = gr.State([])
                 msg.submit(chatbot_interface, [msg, history], [chatbot, msg, history])
+                logging.info("Connected message box to chatbot_interface.")
 
         with gr.Row(elem_classes="footer-row"):
             gr.Markdown("""
-            <div style="text-align:center;margin-top:20px;font-size:14px;">
-                © 2025 PreData.AI - All rights reserved.
+            <div style="text-align: center; margin-top: 20px; font-size: 14px;">
+                © 2025 PreData.AI - All rights reserved. | Contact us at
+                <a href="mailto:hello@predata.ai" onclick="event.stopPropagation();"
+                   style="text-decoration: none; color: #007bff;">hello@predata.ai</a>
             </div>
             """)
 
+    logging.info("Interface build complete.")
     return demo
+
+
+def chatbot_interface(message, history=None):
+    """
+    Handles chatbot interaction. Can be used both in Gradio and from MAIN.
+    """
+    if history is None:
+        history = []
+
+    logging.info(f"User message received: {message}")
+    try:
+        responses = ask_interview_question(respondent_agents_dict, message, processor_llm)
+        logging.info(f"Interview response is {responses}")
+    except Exception as e:
+        logging.error(f"Error during interview processing: {e}")
+        responses = ["Sorry, something went wrong."]
+
+    if isinstance(responses, str):
+        responses = [responses]
+    elif not isinstance(responses, list):
+        logging.warning(f"Unexpected response type ({type(responses)}). Wrapping in list.")
+        responses = [str(responses)]
+
+    formatted_responses = [{"role": "assistant", "content": str(r)} for r in responses]
+    history.append({"role": "user", "content": message})
+    history.extend(formatted_responses)
+
+    logging.info(f"Returning updated history with {len(formatted_responses)} response(s).")
+    return history, "", history