Update app.py

app.py

@@ -1,59 +1,62 @@
 import gradio as gr
 from swarm import Swarm, Agent
 
-# Initialize Swarm client
 client = Swarm()
 
-# Define the agent functions
-def transfer_to_agent_b():
-    return agent_b
-
-def transfer_to_agent_c():
-    return agent_c
-
+# Agent functions for task delegation
 def transfer_to_spanish_agent():
-    """Transfer Spanish speaking users immediately."""
+    """Transfer Spanish-speaking users to the Spanish agent."""
     return spanish_agent
 
-# Define the agents
-agent_a = Agent(
-    name="Agent A",
-    instructions="You are a helpful agent.",
-    functions=[transfer_to_agent_b],
-)
+def transfer_to_english_agent():
+    """Transfer English-speaking users to the English agent."""
+    return english_agent
 
-agent_b = Agent(
-    name="Agent B",
-    instructions="Only speak in Haikus.",
-    functions=[transfer_to_spanish_agent],
-)
+def verify_order_task():
+    """Handle order verification task."""
+    return order_verifier_agent
 
-agent_c = Agent(
-    name="Agent C",
-    instructions="Welcome to the matrix.",
+# General agent that identifies the user's language
+general_agent = Agent(
+    name="General Agent",
+    instructions="Assist the user based on language and delegate tasks if needed.",
+    functions=[transfer_to_spanish_agent, transfer_to_english_agent],
 )
 
+# Spanish-specific agent
 spanish_agent = Agent(
     name="Spanish Agent",
-    instructions="You only speak Spanish.",
+    instructions="You only speak Spanish. Help with customer support in Spanish.",
 )
 
+# English-specific agent
 english_agent = Agent(
     name="English Agent",
-    instructions="You only speak English.",
+    instructions="You only speak English. Help with customer support in English.",
 )
 
-# Append function to transfer to Spanish agent for English agent
-english_agent.functions.append(transfer_to_spanish_agent)
+# Task-specific agent for verifying orders
+order_verifier_agent = Agent(
+    name="Order Verifier",
+    instructions="Verify customer orders and check their status.",
+)
 
-# Create the Gradio interface function
-def chat_with_agent(user_input):
+# Creating the Gradio interface function
+def customer_support_demo(user_input):
     messages = [{"role": "user", "content": user_input}]
-    response = client.run(agent=english_agent, messages=messages)
+    
+    # Check for language and delegate to the appropriate agent
+    if "hola" in user_input.lower():
+        response = client.run(agent=spanish_agent, messages=messages)
+    elif "hello" in user_input.lower():
+        response = client.run(agent=english_agent, messages=messages)
+    else:
+        response = client.run(agent=general_agent, messages=messages)
+    
     return response.messages[-1]["content"]
 
-# Create Gradio interface
-iface = gr.Interface(fn=chat_with_agent, inputs="text", outputs="text", live=True)
+# Create the Gradio interface
+iface = gr.Interface(fn=customer_support_demo, inputs="text", outputs="text", live=True)
 
 # Launch the interface
 iface.launch()