Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 
-# Predefined team lead data for option 1
+# Predefined team lead data
 team_leads = [
     "Neeraj (Healthcare)",
     "Ramya (Finance)",
@@ -8,6 +8,16 @@ team_leads = [
     "Arjun (Logistics)"
 ]
 
+# Predefined team members for Neeraj
+neeraj_team = [
+    "Abhi",
+    "Bharat",
+    "Keerthi",
+    "Ramya",
+    "Pranay",
+    "Gopal"
+]
+
 # Chatbot logic to handle user input
 def chatbot(user_input, conversation_state=None):
     # Initialize conversation state if None
@@ -31,9 +41,11 @@ def chatbot(user_input, conversation_state=None):
     # Step 2: Handle option selection
     elif conversation_state["step"] == "options":
         if user_input == "1":
-            conversation_state["step"] = "initial"  # Reset for next interaction
+            conversation_state["step"] = "team_leads"
+            # Add numbering to team leads
+            numbered_leads = [f"{i+1}. {lead}" for i, lead in enumerate(team_leads)]
             return (
-                "Available Team Leads:\n" + "\n".join(team_leads),
+                "Available Team Leads:\n" + "\n".join(numbered_leads) + "\n\nType the name of a team lead (e.g., 'Neeraj') to see their team members.",
                 conversation_state
             )
         elif user_input in ["2", "3", "4"]:
@@ -48,6 +60,23 @@ def chatbot(user_input, conversation_state=None):
                 conversation_state
             )
 
+    # Step 3: Handle team lead selection
+    elif conversation_state["step"] == "team_leads":
+        if user_input == "neeraj":
+            conversation_state["step"] = "initial"  # Reset for next interaction
+            # Add numbering to team members
+            numbered_members = [f"{i+1}. {member}" for i, member in enumerate(neeraj_team)]
+            return (
+                "Neeraj's Team Members:\n" + "\n".join(numbered_members),
+                conversation_state
+            )
+        else:
+            conversation_state["step"] = "initial"  # Reset if invalid
+            return (
+                "Please select a valid team lead name (e.g., 'Neeraj'). Type 'hi' to start over.",
+                conversation_state
+            )
+
     # Default response for invalid input
     return (
         "Please type 'hi' to start the conversation.",