Spaces:
Running
Running
| import gradio as gr | |
| from gradio_consilium_roundtable import consilium_roundtable | |
| import json | |
| def simulate_travel_planning(): | |
| """Simulate the travel planning conversation from Turin to Philadelphia""" | |
| # Define avatar images for each agent | |
| avatar_images = { | |
| "User Proxy Agent": "https://upload.wikimedia.org/wikipedia/commons/9/99/Sample_User_Icon.png", | |
| "Floor Manager": "https://upload.wikimedia.org/wikipedia/commons/thumb/0/05/Robot_icon.svg/250px-Robot_icon.svg.png", | |
| "Convener": "https://upload.wikimedia.org/wikipedia/commons/thumb/0/05/Robot_icon.svg/250px-Robot_icon.svg.png", | |
| "Flight Assistant": "https://upload.wikimedia.org/wikipedia/commons/thumb/e/e4/Plane-solid.svg/330px-Plane-solid.svg.png", | |
| "Hotel Assistant": "https://upload.wikimedia.org/wikipedia/commons/thumb/5/59/Noun_Project_house_icon_475319_cc.svg/251px-Noun_Project_house_icon_475319_cc.svg.png", | |
| "Rental Car Assistant": "https://upload.wikimedia.org/wikipedia/commons/7/7e/Car_icon_transparent.png", | |
| "Philadelphia Activities Assistant": "https://upload.wikimedia.org/wikipedia/commons/thumb/3/3b/Masks-theater-solid.svg/330px-Masks-theater-solid.svg.png" | |
| } | |
| # Initial state - empty, ready for user message | |
| initial_state = { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [], | |
| "currentSpeaker": None, | |
| "thinking": [], | |
| "showBubbles": [], | |
| "avatarImages": avatar_images | |
| } | |
| states = [ | |
| # 1. User Proxy Agent presents the user's request | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "User Proxy Agent", "text": "I need to make plans for my trip from Turin to Philadelphia from December 5 to December 10. It'll just be me."} | |
| ], | |
| "currentSpeaker": None, | |
| "thinking": [], | |
| "showBubbles": ["User Proxy Agent"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 2. Floor Manager coordinates and Convener starts thinking | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "User Proxy Agent", "text": "I need to make plans for my trip from Turin to Philadelphia from December 5 to December 10. It'll just be me."}, | |
| {"speaker": "Floor Manager", "text": "🎯 **FLOOR ACTIVATED** - Travel Planning Session\n\nInitiating specialist coordination for Turin → Philadelphia trip..."} | |
| ], | |
| "currentSpeaker": "Floor Manager", | |
| "thinking": ["Convener"], | |
| "showBubbles": ["User Proxy Agent"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 2. Convener invites all travel specialists | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "User Proxy Agent", "text": "I need to make plans for my trip from Turin to Philadelphia from December 5 to December 10. It'll just be me."}, | |
| {"speaker": "Floor Manager", "text": "🎯 FLOOR ACTIVATED - Travel Planning Session"}, | |
| {"speaker": "Convener", "text": "📋 **CONVENING TRAVEL SPECIALISTS**\n\nInviting all specialists to join the floor:\n✈️ Flight Assistant\n🏨 Hotel Assistant\n🚗 Rental Car Assistant\n🎭 Philadelphia Activities Assistant"} | |
| ], | |
| "currentSpeaker": "Convener", | |
| "thinking": ["Flight Assistant", "Hotel Assistant"], | |
| "showBubbles": ["User Proxy Agent", "Floor Manager"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 4. Flight Assistant provides flight options | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "User Proxy Agent", "text": "I need to make plans for my trip from Turin to Philadelphia from December 5 to December 10. It'll just be me."}, | |
| {"speaker": "Floor Manager", "text": "🎯 FLOOR coordinating travel specialists"}, | |
| {"speaker": "Convener", "text": "📋 All travel specialists invited to the floor"}, | |
| {"speaker": "Flight Assistant", "text": "✈️ **FLIGHT OPTIONS**\n\nTurin → Philadelphia, December 5:\n• Via Rome: €850 (12h 15m)\n• Via Paris: €920 (13h 10m)\n\nMultiple options available!"} | |
| ], | |
| "currentSpeaker": "Flight Assistant", | |
| "thinking": ["Hotel Assistant", "Rental Car Assistant"], | |
| "showBubbles": ["User Proxy Agent", "Floor Manager", "Convener"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 5. Hotel Assistant provides accommodation options | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "User Proxy Agent", "text": "I need to make plans for my trip from Turin to Philadelphia from December 5 to December 10. It'll just be me."}, | |
| {"speaker": "Flight Assistant", "text": "✈️ Flight options: €850-920, multiple routes"}, | |
| {"speaker": "Hotel Assistant", "text": "🏨 **PHILADELPHIA HOTELS**\n\nDecember 5-10 (5 nights):\n• Center City: $195-280/night\n• Airport area: $140-165/night\n\nGreat options in both areas!"} | |
| ], | |
| "currentSpeaker": "Hotel Assistant", | |
| "thinking": ["Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "showBubbles": ["User Proxy Agent", "Flight Assistant"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 6. Rental Car Assistant provides transportation options | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "User Proxy Agent", "text": "I need to make plans for my trip from Turin to Philadelphia from December 5 to December 10. It'll just be me."}, | |
| {"speaker": "Flight Assistant", "text": "✈️ Flights: €850-920, multiple routing options"}, | |
| {"speaker": "Hotel Assistant", "text": "🏨 Hotels: $140-280/night, city center and airport"}, | |
| {"speaker": "Rental Car Assistant", "text": "🚗 **RENTAL CARS**\n\nDecember 5-10 pickup at airport:\n• Economy: $45-52/day\n• Mid-size: $65-68/day\n• Premium: $95/day\n\nAlso excellent public transit available!"} | |
| ], | |
| "currentSpeaker": "Rental Car Assistant", | |
| "thinking": ["Philadelphia Activities Assistant"], | |
| "showBubbles": ["Flight Assistant", "Hotel Assistant"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 7. Philadelphia Activities Assistant completes initial planning | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "User Proxy Agent", "text": "Sounds good to me!"}, | |
| {"speaker": "Flight Assistant", "text": "✈️ Flight options available"}, | |
| {"speaker": "Hotel Assistant", "text": "🏨 Hotel accommodations ready"}, | |
| {"speaker": "Rental Car Assistant", "text": "🚗 Ground transportation sorted"}, | |
| {"speaker": "Philadelphia Activities Assistant", "text": "🎭 **PHILADELPHIA ACTIVITIES**\n\nGreat options for December 5-10:\n• Christmas Village at Love Park\n• Philadelphia Museum of Art\n• Independence Hall & Liberty Bell\n• Reading Terminal Market\n• Amazing cheesesteaks!\n\nPhiladelphia will be wonderful!"} | |
| ], | |
| "currentSpeaker": "Philadelphia Activities Assistant", | |
| "thinking": [], | |
| "showBubbles": ["Flight Assistant", "Hotel Assistant", "Rental Car Assistant"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 8. User decides to book the original plan | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "Flight Assistant", "text": "✈️ Flight options available (€850-920)"}, | |
| {"speaker": "Hotel Assistant", "text": "🏨 Hotel accommodations ready ($140-280/night)"}, | |
| {"speaker": "Rental Car Assistant", "text": "🚗 Ground transportation sorted ($45-95/day)"}, | |
| {"speaker": "Philadelphia Activities Assistant", "text": "🎭 Philadelphia activities planned for December 5-10"}, | |
| {"speaker": "User Proxy Agent", "text": "Perfect! This all looks great. Please go ahead and book the flights, hotel, and rental car for December 5-10. I'm ready to finalize everything!"} | |
| ], | |
| "currentSpeaker": "User Proxy Agent", | |
| "thinking": ["Philadelphia Activities Assistant"], | |
| "showBubbles": ["Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Floor Manager"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 9. Activities Assistant suggests Bocelli concert AFTER user wants to book | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "User Proxy Agent", "text": "Perfect! This all looks great. Please go ahead and book the flights, hotel, and rental car for December 5-10. I'm ready to finalize everything!"}, | |
| {"speaker": "Philadelphia Activities Assistant", "text": "🎼 **Wait! Before we book...** I just noticed something incredible!\n\nAndrea Bocelli is performing December 11th at Kimmel Center. Since you're from Italy, this could be a once-in-a-lifetime opportunity! Consider staying one extra day?"} | |
| ], | |
| "currentSpeaker": "Philadelphia Activities Assistant", | |
| "thinking": ["Flight Assistant", "Hotel Assistant", "Rental Car Assistant"], | |
| "showBubbles": ["User Proxy Agent", "Floor Manager"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 10. Other agents hear Bocelli suggestion and adapt their recommendations | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "User Proxy Agent", "text": "Perfect! This all looks great. Please go ahead and book the flights, hotel, and rental car for December 5-10. I'm ready to finalize everything!"}, | |
| {"speaker": "Philadelphia Activities Assistant", "text": "🎼 **Bocelli concert December 11th!** Consider extending one day?"}, | |
| {"speaker": "Hotel Assistant", "text": "🏨 **Hearing about Bocelli!** 👂\nKimmel area hotel available Dec 11th - same $195 rate. Worth extending!"}, | |
| {"speaker": "Flight Assistant", "text": "✈️ **Concert suggestion noted!** 🎵\nMonday Dec 12 flights are €100+ cheaper than Sunday! Extend and save money."}, | |
| {"speaker": "Rental Car Assistant", "text": "🚗 **Adapting to concert plan!**\nWeekly rate is better anyway - only +$55 for 2 extra days."} | |
| ], | |
| "currentSpeaker": None, | |
| "thinking": ["User Proxy Agent"], | |
| "showBubbles": ["Floor Manager", "Convener", "Philadelphia Activities Assistant"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 11. User says YES to the enhanced plan | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "Philadelphia Activities Assistant", "text": "🎼 Bocelli concert December 11th! Consider extending?"}, | |
| {"speaker": "Hotel Assistant", "text": "🏨 Hotel extension available, same rate!"}, | |
| {"speaker": "Flight Assistant", "text": "✈️ Monday flights cheaper - extend and save!"}, | |
| {"speaker": "Rental Car Assistant", "text": "🚗 Weekly rental rate is better anyway!"}, | |
| {"speaker": "User Proxy Agent", "text": "YES! Absolutely! Bocelli from my home country performing in Philadelphia? And I save money on flights? This is perfect - let's extend the trip!"} | |
| ], | |
| "currentSpeaker": "User Proxy Agent", | |
| "thinking": ["Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "showBubbles": ["Floor Manager"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 12. All agents process the enhanced bookings | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "User Proxy Agent", "text": "YES! Absolutely! Bocelli from my home country performing in Philadelphia? And I save money on flights? This is perfect - let's extend the trip!"}, | |
| {"speaker": "Flight Assistant", "text": "✈️ **Extended flights booked!**\nTRN→PHL Dec 5, return Dec 12\nTotal: €780 (saved €70!)"}, | |
| {"speaker": "Hotel Assistant", "text": "🏨 **Extended stay confirmed!**\nKimmel Center area, Dec 5-12\n7 nights, $1,365 total"}, | |
| {"speaker": "Rental Car Assistant", "text": "🚗 **Extended rental confirmed!**\nHonda Accord, Dec 5-12\n7 days, $315 total"}, | |
| {"speaker": "Philadelphia Activities Assistant", "text": "🎼 **Bocelli concert booked!**\nDec 11, 8pm, Orchestra section\nConfirmation: KIMMEL-456"} | |
| ], | |
| "currentSpeaker": None, | |
| "thinking": [], | |
| "showBubbles": ["User Proxy Agent"], | |
| "avatarImages": avatar_images | |
| }, | |
| # 13. Final celebration and completion | |
| { | |
| "participants": ["User Proxy Agent", "Floor Manager", "Convener", "Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant"], | |
| "messages": [ | |
| {"speaker": "Flight Assistant", "text": "✈️ Enhanced flights with savings (€780)"}, | |
| {"speaker": "Hotel Assistant", "text": "🏨 Extended stay confirmed ($1,365)"}, | |
| {"speaker": "Rental Car Assistant", "text": "🚗 Extended rental confirmed ($315)"}, | |
| {"speaker": "Philadelphia Activities Assistant", "text": "🎼 Bocelli concert secured!"}, | |
| {"speaker": "Floor Manager", "text": "🎯 **Trip enhanced through collaboration!**\nOriginal: 5 days, ~$2,100\nNew: 7 days + Bocelli, ~$2,650\nAgent teamwork created amazing value! 🎼"}, | |
| {"speaker": "User Proxy Agent", "text": "This collaborative planning is incredible! You saved me money on flights and gave me Bocelli - thank you all!"} | |
| ], | |
| "currentSpeaker": None, | |
| "thinking": [], | |
| "showBubbles": ["Flight Assistant", "Hotel Assistant", "Rental Car Assistant", "Philadelphia Activities Assistant", "Floor Manager", "User Proxy Agent"], | |
| "avatarImages": avatar_images | |
| } | |
| ] | |
| return initial_state, states | |
| def update_discussion_state(state_index, states): | |
| """Get the next state in the travel planning discussion""" | |
| if state_index >= len(states): | |
| state_index = 0 | |
| return states[state_index], state_index + 1 | |
| # Initialize the travel planning discussion | |
| initial_state, discussion_states = simulate_travel_planning() | |
| with gr.Blocks(title="🗣️ Open Floor Protocol - Roundtable Demo") as demo: | |
| gr.Markdown(""" | |
| # 🗣️ Open Floor Protocol - Roundtable Demo | |
| **Watch AI travel specialists collaborate and adapt!** This simulates a conversation flow using the Open Floor Protocol: a user requests trip planning from Turin to Philadelphia, and specialist agents not only provide recommendations but **listen to each other** and **dynamically improve** the plan together. | |
| 🎯 **Key Feature:** Notice how the Activities Assistant discovers a special concert opportunity, and ALL other agents immediately adapt their recommendations based on what they heard - creating a better trip that actually saves money! | |
| **Read more about [Open Floor](https://github.com/open-voice-interoperability/openfloor-docs) and the [official Python library](https://github.com/open-voice-interoperability/openfloor-python)** | |
| """) | |
| # State management | |
| state_counter = gr.State(0) | |
| # The roundtable component | |
| roundtable = consilium_roundtable( | |
| label="Shared Conversational Space", | |
| show_label=True, | |
| label_icon="https://avatars.githubusercontent.com/u/46052400?s=48&v=4", | |
| value=initial_state | |
| ) | |
| with gr.Row(): | |
| next_btn = gr.Button("▶️ Next Planning Step", variant="primary") | |
| reset_btn = gr.Button("🔄 Reset Planning Session", variant="secondary") | |
| # Status display | |
| with gr.Row(): | |
| status_display = gr.Markdown("**Status:** 🎯 Travel planning floor ready - awaiting user request") | |
| # Trip details display | |
| with gr.Row(): | |
| trip_info = gr.Markdown(""" | |
| ### 🧳 Trip Overview | |
| **Route:** Turin, Italy → Philadelphia, USA | |
| **Dates:** December 5-10, 2024 | |
| **Traveler:** Solo trip | |
| **Specialists:** Flight, Hotel, Car Rental, Activities | |
| """) | |
| gr.Markdown(""" | |
| Avatar Images from Wikimedia: | |
| * User Proxy Agent: https://upload.wikimedia.org/wikipedia/commons/9/99/Sample_User_Icon.png | |
| * Floor Manager: https://upload.wikimedia.org/wikipedia/commons/thumb/0/05/Robot_icon.svg/250px-Robot_icon.svg.png | |
| * Convener: https://upload.wikimedia.org/wikipedia/commons/thumb/0/05/Robot_icon.svg/250px-Robot_icon.svg.png | |
| * Flight Assistant: https://upload.wikimedia.org/wikipedia/commons/thumb/e/e4/Plane-solid.svg/330px-Plane-solid.svg.png | |
| * Hotel Assistant: https://upload.wikimedia.org/wikipedia/commons/thumb/5/59/Noun_Project_house_icon_475319_cc.svg/251px-Noun_Project_house_icon_475319_cc.svg.png | |
| * Rental Car Assistant: https://upload.wikimedia.org/wikipedia/commons/7/7e/Car_icon_transparent.png | |
| * Philadelphia Activities Assistant: https://upload.wikimedia.org/wikipedia/commons/thumb/3/3b/Masks-theater-solid.svg/330px-Masks-theater-solid.svg.png | |
| """) | |
| def next_state(current_counter): | |
| new_state, new_counter = update_discussion_state(current_counter, discussion_states) | |
| # Convert to proper JSON string | |
| json_state = json.dumps(new_state) | |
| # Create status message based on current state | |
| thinking_list = new_state.get("thinking", []) | |
| current_speaker = new_state.get("currentSpeaker") | |
| # Custom status messages for travel planning context | |
| if current_counter == 0: | |
| status = "**Status:** 🎯 Floor Manager activating coordination protocols" | |
| elif current_counter == 1: | |
| status = "**Status:** 📋 Convener inviting specialist agents to the floor" | |
| elif current_counter == 6: | |
| status = "**Status:** 📝 User decides to book original 5-day plan..." | |
| elif current_counter == 7: | |
| status = "**Status:** 🎼 Activities Assistant makes last-minute discovery!" | |
| elif current_counter == 8: | |
| status = "**Status:** 🤝 All agents adapting after hearing concert suggestion" | |
| elif current_counter == 9: | |
| status = "**Status:** ✅ User says YES to the enhanced plan!" | |
| elif current_counter == 10: | |
| status = "**Status:** 📋 All agents processing enhanced bookings..." | |
| elif current_counter == 11: | |
| status = "**Status:** 🎼 User celebrating collaborative success - conversation complete!" | |
| elif thinking_list: | |
| agents = ', '.join(thinking_list) | |
| status = f"**Status:** 💭 {agents} {'is' if len(thinking_list) == 1 else 'are'} preparing recommendations..." | |
| elif current_speaker: | |
| status = f"**Status:** 🎤 {current_speaker} has the floor..." | |
| else: | |
| status = "**Status:** 🎯 Travel coordination in progress on the shared floor" | |
| return json_state, new_counter, status | |
| def reset_discussion(): | |
| json_state = json.dumps(initial_state) | |
| return json_state, 0, "**Status:** 🎯 Travel planning floor reset - ready for new requests" | |
| next_btn.click( | |
| next_state, | |
| inputs=[state_counter], | |
| outputs=[roundtable, state_counter, status_display] | |
| ) | |
| reset_btn.click( | |
| reset_discussion, | |
| outputs=[roundtable, state_counter, status_display] | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() |