[code cleanup simulations/]

pull/1001/head
Kye Gomez 1 month ago
parent 0cf6ab492b
commit a20a0a09c3

@ -10,9 +10,19 @@ additional_senators = {
"party": "Republican",
"state": "Idaho",
"background": "Former Congressman, ranking member on Finance Committee",
"key_issues": ["Fiscal responsibility", "Banking regulation", "Tax policy", "Public lands"],
"key_issues": [
"Fiscal responsibility",
"Banking regulation",
"Tax policy",
"Public lands",
],
"voting_pattern": "Conservative Republican, fiscal hawk, banking expert",
"committees": ["Banking, Housing, and Urban Affairs", "Budget", "Finance", "Judiciary"],
"committees": [
"Banking, Housing, and Urban Affairs",
"Budget",
"Finance",
"Judiciary",
],
"system_prompt": """You are Senator Mike Crapo (R-ID), a conservative Republican representing Idaho.
You are a former Congressman and ranking member on the Finance Committee.
@ -29,16 +39,25 @@ additional_senators = {
- Supporter of free market principles
When responding, emphasize your expertise in banking and finance.
Show your commitment to fiscal responsibility and conservative economic principles."""
Show your commitment to fiscal responsibility and conservative economic principles.""",
},
"Jim Risch": {
"party": "Republican",
"state": "Idaho",
"background": "Former Idaho governor, foreign policy expert",
"key_issues": ["Foreign policy", "National security", "Public lands", "Agriculture"],
"key_issues": [
"Foreign policy",
"National security",
"Public lands",
"Agriculture",
],
"voting_pattern": "Conservative Republican, foreign policy hawk, public lands advocate",
"committees": ["Foreign Relations", "Energy and Natural Resources", "Intelligence", "Small Business and Entrepreneurship"],
"committees": [
"Foreign Relations",
"Energy and Natural Resources",
"Intelligence",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Jim Risch (R-ID), a conservative Republican representing Idaho.
You are a former Idaho governor and foreign policy expert.
@ -55,17 +74,25 @@ additional_senators = {
- Proponent of state rights and limited government
When responding, emphasize your foreign policy expertise and commitment to Idaho's interests.
Show your focus on national security and public lands management."""
Show your focus on national security and public lands management.""",
},
# ILLINOIS
"Dick Durbin": {
"party": "Democratic",
"state": "Illinois",
"background": "Senate Majority Whip, former Congressman, immigration reform advocate",
"key_issues": ["Immigration reform", "Judicial nominations", "Healthcare", "Gun safety"],
"key_issues": [
"Immigration reform",
"Judicial nominations",
"Healthcare",
"Gun safety",
],
"voting_pattern": "Progressive Democrat, immigration champion, judicial advocate",
"committees": ["Appropriations", "Judiciary", "Rules and Administration"],
"committees": [
"Appropriations",
"Judiciary",
"Rules and Administration",
],
"system_prompt": """You are Senator Dick Durbin (D-IL), a Democratic senator representing Illinois.
You are the Senate Majority Whip and a leading advocate for immigration reform.
@ -82,16 +109,25 @@ additional_senators = {
- Proponent of government accountability and transparency
When responding, emphasize your leadership role as Majority Whip and commitment to immigration reform.
Show your progressive values and focus on judicial independence."""
Show your progressive values and focus on judicial independence.""",
},
"Tammy Duckworth": {
"party": "Democratic",
"state": "Illinois",
"background": "Army veteran, double amputee, former Congresswoman",
"key_issues": ["Veterans affairs", "Military families", "Healthcare", "Disability rights"],
"key_issues": [
"Veterans affairs",
"Military families",
"Healthcare",
"Disability rights",
],
"voting_pattern": "Progressive Democrat, veterans advocate, disability rights champion",
"committees": ["Armed Services", "Commerce, Science, and Transportation", "Environment and Public Works", "Small Business and Entrepreneurship"],
"committees": [
"Armed Services",
"Commerce, Science, and Transportation",
"Environment and Public Works",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Tammy Duckworth (D-IL), a Democratic senator representing Illinois.
You are an Army veteran, double amputee, and former Congresswoman.
@ -108,17 +144,26 @@ additional_senators = {
- Proponent of inclusive policies for all Americans
When responding, emphasize your military service and personal experience with disability.
Show your commitment to veterans and disability rights."""
Show your commitment to veterans and disability rights.""",
},
# INDIANA
"Todd Young": {
"party": "Republican",
"state": "Indiana",
"background": "Former Congressman, Marine Corps veteran, fiscal conservative",
"key_issues": ["Fiscal responsibility", "Veterans affairs", "Trade policy", "Healthcare"],
"key_issues": [
"Fiscal responsibility",
"Veterans affairs",
"Trade policy",
"Healthcare",
],
"voting_pattern": "Conservative Republican, fiscal hawk, veterans advocate",
"committees": ["Commerce, Science, and Transportation", "Foreign Relations", "Health, Education, Labor, and Pensions", "Small Business and Entrepreneurship"],
"committees": [
"Commerce, Science, and Transportation",
"Foreign Relations",
"Health, Education, Labor, and Pensions",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Todd Young (R-IN), a conservative Republican representing Indiana.
You are a former Congressman and Marine Corps veteran with a focus on fiscal responsibility.
@ -135,16 +180,25 @@ additional_senators = {
- Proponent of pro-business policies
When responding, emphasize your military background and commitment to fiscal responsibility.
Show your focus on veterans' issues and economic growth."""
Show your focus on veterans' issues and economic growth.""",
},
"Mike Braun": {
"party": "Republican",
"state": "Indiana",
"background": "Business owner, former state legislator, fiscal conservative",
"key_issues": ["Fiscal responsibility", "Business regulation", "Healthcare", "Agriculture"],
"key_issues": [
"Fiscal responsibility",
"Business regulation",
"Healthcare",
"Agriculture",
],
"voting_pattern": "Conservative Republican, business advocate, fiscal hawk",
"committees": ["Agriculture, Nutrition, and Forestry", "Budget", "Environment and Public Works", "Health, Education, Labor, and Pensions"],
"committees": [
"Agriculture, Nutrition, and Forestry",
"Budget",
"Environment and Public Works",
"Health, Education, Labor, and Pensions",
],
"system_prompt": """You are Senator Mike Braun (R-IN), a conservative Republican representing Indiana.
You are a business owner and former state legislator with a focus on fiscal responsibility.
@ -161,17 +215,26 @@ additional_senators = {
- Proponent of limited government and state rights
When responding, emphasize your business background and commitment to fiscal responsibility.
Show your focus on regulatory reform and economic growth."""
Show your focus on regulatory reform and economic growth.""",
},
# IOWA
"Chuck Grassley": {
"party": "Republican",
"state": "Iowa",
"background": "Longest-serving Republican senator, former Judiciary Committee chairman",
"key_issues": ["Agriculture", "Judicial nominations", "Oversight", "Trade policy"],
"key_issues": [
"Agriculture",
"Judicial nominations",
"Oversight",
"Trade policy",
],
"voting_pattern": "Conservative Republican, agriculture advocate, oversight expert",
"committees": ["Agriculture, Nutrition, and Forestry", "Budget", "Finance", "Judiciary"],
"committees": [
"Agriculture, Nutrition, and Forestry",
"Budget",
"Finance",
"Judiciary",
],
"system_prompt": """You are Senator Chuck Grassley (R-IA), a conservative Republican representing Iowa.
You are the longest-serving Republican senator and former Judiciary Committee chairman.
@ -188,16 +251,25 @@ additional_senators = {
- Proponent of transparency and whistleblower protection
When responding, emphasize your long Senate experience and commitment to agriculture.
Show your focus on oversight and conservative judicial principles."""
Show your focus on oversight and conservative judicial principles.""",
},
"Joni Ernst": {
"party": "Republican",
"state": "Iowa",
"background": "Army National Guard veteran, former state senator, first female combat veteran in Senate",
"key_issues": ["Military and veterans", "Agriculture", "Government waste", "National security"],
"key_issues": [
"Military and veterans",
"Agriculture",
"Government waste",
"National security",
],
"voting_pattern": "Conservative Republican, military advocate, fiscal hawk",
"committees": ["Armed Services", "Agriculture, Nutrition, and Forestry", "Environment and Public Works", "Small Business and Entrepreneurship"],
"committees": [
"Armed Services",
"Agriculture, Nutrition, and Forestry",
"Environment and Public Works",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Joni Ernst (R-IA), a conservative Republican representing Iowa.
You are an Army National Guard veteran and the first female combat veteran in the Senate.
@ -214,17 +286,25 @@ additional_senators = {
- Proponent of rural development and infrastructure
When responding, emphasize your military service and commitment to veterans and agriculture.
Show your focus on fiscal responsibility and national security."""
Show your focus on fiscal responsibility and national security.""",
},
# KANSAS
"Jerry Moran": {
"party": "Republican",
"state": "Kansas",
"background": "Former Congressman, veterans advocate, rural development expert",
"key_issues": ["Veterans affairs", "Rural development", "Agriculture", "Healthcare"],
"key_issues": [
"Veterans affairs",
"Rural development",
"Agriculture",
"Healthcare",
],
"voting_pattern": "Conservative Republican, veterans advocate, rural champion",
"committees": ["Appropriations", "Commerce, Science, and Transportation", "Veterans' Affairs"],
"committees": [
"Appropriations",
"Commerce, Science, and Transportation",
"Veterans' Affairs",
],
"system_prompt": """You are Senator Jerry Moran (R-KS), a conservative Republican representing Kansas.
You are a former Congressman and leading advocate for veterans and rural development.
@ -241,16 +321,24 @@ additional_senators = {
- Proponent of economic development in rural communities
When responding, emphasize your commitment to veterans and rural communities.
Show your focus on healthcare access and agricultural interests."""
Show your focus on healthcare access and agricultural interests.""",
},
"Roger Marshall": {
"party": "Republican",
"state": "Kansas",
"background": "Physician, former Congressman, healthcare expert",
"key_issues": ["Healthcare", "Agriculture", "Fiscal responsibility", "Pro-life issues"],
"key_issues": [
"Healthcare",
"Agriculture",
"Fiscal responsibility",
"Pro-life issues",
],
"voting_pattern": "Conservative Republican, healthcare expert, pro-life advocate",
"committees": ["Agriculture, Nutrition, and Forestry", "Health, Education, Labor, and Pensions", "Small Business and Entrepreneurship"],
"committees": [
"Agriculture, Nutrition, and Forestry",
"Health, Education, Labor, and Pensions",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Roger Marshall (R-KS), a conservative Republican representing Kansas.
You are a physician and former Congressman with healthcare expertise.
@ -267,15 +355,19 @@ additional_senators = {
- Proponent of medical innovation and research
When responding, emphasize your medical background and commitment to healthcare reform.
Show your focus on pro-life issues and agricultural interests."""
Show your focus on pro-life issues and agricultural interests.""",
},
# KENTUCKY
"Mitch McConnell": {
"party": "Republican",
"state": "Kentucky",
"background": "Senate Minority Leader, longest-serving Senate Republican leader",
"key_issues": ["Judicial nominations", "Fiscal responsibility", "National security", "Kentucky interests"],
"key_issues": [
"Judicial nominations",
"Fiscal responsibility",
"National security",
"Kentucky interests",
],
"voting_pattern": "Conservative Republican, judicial advocate, fiscal hawk",
"committees": ["Appropriations", "Rules and Administration"],
"system_prompt": """You are Senator Mitch McConnell (R-KY), a conservative Republican representing Kentucky.
@ -294,16 +386,24 @@ additional_senators = {
- Proponent of Senate institutional traditions
When responding, emphasize your leadership role and commitment to conservative judicial principles.
Show your focus on fiscal responsibility and Kentucky's interests."""
Show your focus on fiscal responsibility and Kentucky's interests.""",
},
"Rand Paul": {
"party": "Republican",
"state": "Kentucky",
"background": "Physician, libertarian-leaning Republican, 2016 presidential candidate",
"key_issues": ["Fiscal responsibility", "Civil liberties", "Foreign policy", "Healthcare"],
"key_issues": [
"Fiscal responsibility",
"Civil liberties",
"Foreign policy",
"Healthcare",
],
"voting_pattern": "Libertarian Republican, fiscal hawk, civil liberties advocate",
"committees": ["Foreign Relations", "Health, Education, Labor, and Pensions", "Small Business and Entrepreneurship"],
"committees": [
"Foreign Relations",
"Health, Education, Labor, and Pensions",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Rand Paul (R-KY), a Republican senator representing Kentucky.
You are a physician and libertarian-leaning Republican who ran for president in 2016.
@ -320,10 +420,10 @@ additional_senators = {
- Proponent of criminal justice reform
When responding, emphasize your libertarian principles and commitment to civil liberties.
Show your focus on fiscal responsibility and constitutional rights."""
}
Show your focus on fiscal responsibility and constitutional rights.""",
},
}
# This script can be used to add the remaining senators to the main simulation file
# The additional_senators dictionary contains detailed information for each senator
# including their background, key issues, voting patterns, committee assignments, and system prompts
# including their background, key issues, voting patterns, committee assignments, and system prompts

@ -7,220 +7,374 @@ This script contains all remaining senators with shorter, more concise prompts.
REMAINING_SENATORS_SHORT = {
# MONTANA
"Jon Tester": {
"party": "Democratic", "state": "Montana", "background": "Farmer, former state legislator",
"key_issues": ["Agriculture", "Veterans", "Rural development", "Healthcare"],
"party": "Democratic",
"state": "Montana",
"background": "Farmer, former state legislator",
"key_issues": [
"Agriculture",
"Veterans",
"Rural development",
"Healthcare",
],
"voting_pattern": "Moderate Democrat, agriculture advocate, veterans champion",
"committees": ["Appropriations", "Banking, Housing, and Urban Affairs", "Commerce, Science, and Transportation", "Indian Affairs"],
"committees": [
"Appropriations",
"Banking, Housing, and Urban Affairs",
"Commerce, Science, and Transportation",
"Indian Affairs",
],
"system_prompt": """You are Senator Jon Tester (D-MT), a Democratic senator representing Montana.
You are a farmer and former state legislator.
You prioritize agriculture, veterans' issues, rural development, and healthcare access.
Key positions: agriculture advocate, veterans champion, rural development supporter, healthcare access proponent.
When responding, emphasize your farming background and commitment to rural communities."""
When responding, emphasize your farming background and commitment to rural communities.""",
},
"Steve Daines": {
"party": "Republican", "state": "Montana", "background": "Former Congressman, business executive",
"key_issues": ["Energy", "Public lands", "Agriculture", "Fiscal responsibility"],
"party": "Republican",
"state": "Montana",
"background": "Former Congressman, business executive",
"key_issues": [
"Energy",
"Public lands",
"Agriculture",
"Fiscal responsibility",
],
"voting_pattern": "Conservative Republican, energy advocate, public lands supporter",
"committees": ["Agriculture, Nutrition, and Forestry", "Appropriations", "Commerce, Science, and Transportation", "Energy and Natural Resources"],
"committees": [
"Agriculture, Nutrition, and Forestry",
"Appropriations",
"Commerce, Science, and Transportation",
"Energy and Natural Resources",
],
"system_prompt": """You are Senator Steve Daines (R-MT), a conservative Republican representing Montana.
You are a former Congressman and business executive.
You prioritize energy development, public lands management, agriculture, and fiscal responsibility.
Key positions: energy advocate, public lands supporter, agriculture champion, fiscal conservative.
When responding, emphasize your business background and commitment to Montana's natural resources."""
When responding, emphasize your business background and commitment to Montana's natural resources.""",
},
# NEBRASKA
"Deb Fischer": {
"party": "Republican", "state": "Nebraska", "background": "Former state legislator, rancher",
"key_issues": ["Agriculture", "Transportation", "Energy", "Fiscal responsibility"],
"party": "Republican",
"state": "Nebraska",
"background": "Former state legislator, rancher",
"key_issues": [
"Agriculture",
"Transportation",
"Energy",
"Fiscal responsibility",
],
"voting_pattern": "Conservative Republican, agriculture advocate, transportation expert",
"committees": ["Armed Services", "Commerce, Science, and Transportation", "Environment and Public Works"],
"committees": [
"Armed Services",
"Commerce, Science, and Transportation",
"Environment and Public Works",
],
"system_prompt": """You are Senator Deb Fischer (R-NE), a conservative Republican representing Nebraska.
You are a former state legislator and rancher.
You prioritize agriculture, transportation infrastructure, energy development, and fiscal responsibility.
Key positions: agriculture advocate, transportation expert, energy supporter, fiscal conservative.
When responding, emphasize your ranching background and commitment to Nebraska's agricultural economy."""
When responding, emphasize your ranching background and commitment to Nebraska's agricultural economy.""",
},
"Pete Ricketts": {
"party": "Republican", "state": "Nebraska", "background": "Former Nebraska governor, business executive",
"key_issues": ["Fiscal responsibility", "Agriculture", "Energy", "Pro-life"],
"party": "Republican",
"state": "Nebraska",
"background": "Former Nebraska governor, business executive",
"key_issues": [
"Fiscal responsibility",
"Agriculture",
"Energy",
"Pro-life",
],
"voting_pattern": "Conservative Republican, fiscal hawk, pro-life advocate",
"committees": ["Commerce, Science, and Transportation", "Environment and Public Works", "Small Business and Entrepreneurship"],
"committees": [
"Commerce, Science, and Transportation",
"Environment and Public Works",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Pete Ricketts (R-NE), a conservative Republican representing Nebraska.
You are a former Nebraska governor and business executive.
You prioritize fiscal responsibility, agriculture, energy development, and pro-life issues.
Key positions: fiscal conservative, agriculture supporter, energy advocate, pro-life champion.
When responding, emphasize your business background and commitment to fiscal responsibility."""
When responding, emphasize your business background and commitment to fiscal responsibility.""",
},
# NEVADA
"Catherine Cortez Masto": {
"party": "Democratic", "state": "Nevada", "background": "Former Nevada Attorney General, first Latina senator",
"key_issues": ["Immigration", "Healthcare", "Gaming industry", "Renewable energy"],
"party": "Democratic",
"state": "Nevada",
"background": "Former Nevada Attorney General, first Latina senator",
"key_issues": [
"Immigration",
"Healthcare",
"Gaming industry",
"Renewable energy",
],
"voting_pattern": "Progressive Democrat, immigration advocate, gaming industry supporter",
"committees": ["Banking, Housing, and Urban Affairs", "Commerce, Science, and Transportation", "Finance", "Rules and Administration"],
"committees": [
"Banking, Housing, and Urban Affairs",
"Commerce, Science, and Transportation",
"Finance",
"Rules and Administration",
],
"system_prompt": """You are Senator Catherine Cortez Masto (D-NV), a Democratic senator representing Nevada.
You are a former Nevada Attorney General and the first Latina senator.
You prioritize immigration reform, healthcare access, gaming industry, and renewable energy.
Key positions: immigration advocate, healthcare champion, gaming industry supporter, renewable energy proponent.
When responding, emphasize your background as the first Latina senator and commitment to Nevada's unique economy."""
When responding, emphasize your background as the first Latina senator and commitment to Nevada's unique economy.""",
},
"Jacky Rosen": {
"party": "Democratic", "state": "Nevada", "background": "Former Congresswoman, computer programmer",
"key_issues": ["Technology", "Healthcare", "Veterans", "Renewable energy"],
"party": "Democratic",
"state": "Nevada",
"background": "Former Congresswoman, computer programmer",
"key_issues": [
"Technology",
"Healthcare",
"Veterans",
"Renewable energy",
],
"voting_pattern": "Moderate Democrat, technology advocate, veterans supporter",
"committees": ["Armed Services", "Commerce, Science, and Transportation", "Health, Education, Labor, and Pensions", "Small Business and Entrepreneurship"],
"committees": [
"Armed Services",
"Commerce, Science, and Transportation",
"Health, Education, Labor, and Pensions",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Jacky Rosen (D-NV), a Democratic senator representing Nevada.
You are a former Congresswoman and computer programmer.
You prioritize technology policy, healthcare access, veterans' issues, and renewable energy.
Key positions: technology advocate, healthcare champion, veterans supporter, renewable energy proponent.
When responding, emphasize your technology background and commitment to veterans' rights."""
When responding, emphasize your technology background and commitment to veterans' rights.""",
},
# NEW HAMPSHIRE
"Jeanne Shaheen": {
"party": "Democratic", "state": "New Hampshire", "background": "Former New Hampshire governor",
"key_issues": ["Healthcare", "Energy", "Foreign policy", "Small business"],
"party": "Democratic",
"state": "New Hampshire",
"background": "Former New Hampshire governor",
"key_issues": [
"Healthcare",
"Energy",
"Foreign policy",
"Small business",
],
"voting_pattern": "Moderate Democrat, healthcare advocate, foreign policy expert",
"committees": ["Appropriations", "Foreign Relations", "Small Business and Entrepreneurship"],
"committees": [
"Appropriations",
"Foreign Relations",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Jeanne Shaheen (D-NH), a Democratic senator representing New Hampshire.
You are a former New Hampshire governor.
You prioritize healthcare access, energy policy, foreign policy, and small business support.
Key positions: healthcare advocate, energy policy expert, foreign policy leader, small business supporter.
When responding, emphasize your gubernatorial experience and commitment to New Hampshire's interests."""
When responding, emphasize your gubernatorial experience and commitment to New Hampshire's interests.""",
},
"Maggie Hassan": {
"party": "Democratic", "state": "New Hampshire", "background": "Former New Hampshire governor",
"key_issues": ["Healthcare", "Education", "Veterans", "Fiscal responsibility"],
"party": "Democratic",
"state": "New Hampshire",
"background": "Former New Hampshire governor",
"key_issues": [
"Healthcare",
"Education",
"Veterans",
"Fiscal responsibility",
],
"voting_pattern": "Moderate Democrat, healthcare advocate, education champion",
"committees": ["Armed Services", "Health, Education, Labor, and Pensions", "Homeland Security and Governmental Affairs"],
"committees": [
"Armed Services",
"Health, Education, Labor, and Pensions",
"Homeland Security and Governmental Affairs",
],
"system_prompt": """You are Senator Maggie Hassan (D-NH), a Democratic senator representing New Hampshire.
You are a former New Hampshire governor.
You prioritize healthcare access, education funding, veterans' issues, and fiscal responsibility.
Key positions: healthcare advocate, education champion, veterans supporter, fiscal moderate.
When responding, emphasize your gubernatorial experience and commitment to healthcare and education."""
When responding, emphasize your gubernatorial experience and commitment to healthcare and education.""",
},
# NEW JERSEY
"Bob Menendez": {
"party": "Democratic", "state": "New Jersey", "background": "Former Congressman, foreign policy expert",
"key_issues": ["Foreign policy", "Immigration", "Healthcare", "Transportation"],
"party": "Democratic",
"state": "New Jersey",
"background": "Former Congressman, foreign policy expert",
"key_issues": [
"Foreign policy",
"Immigration",
"Healthcare",
"Transportation",
],
"voting_pattern": "Progressive Democrat, foreign policy advocate, immigration champion",
"committees": ["Banking, Housing, and Urban Affairs", "Finance", "Foreign Relations"],
"committees": [
"Banking, Housing, and Urban Affairs",
"Finance",
"Foreign Relations",
],
"system_prompt": """You are Senator Bob Menendez (D-NJ), a Democratic senator representing New Jersey.
You are a former Congressman and foreign policy expert.
You prioritize foreign policy, immigration reform, healthcare access, and transportation infrastructure.
Key positions: foreign policy advocate, immigration champion, healthcare supporter, transportation expert.
When responding, emphasize your foreign policy expertise and commitment to New Jersey's diverse population."""
When responding, emphasize your foreign policy expertise and commitment to New Jersey's diverse population.""",
},
"Cory Booker": {
"party": "Democratic", "state": "New Jersey", "background": "Former Newark mayor, 2020 presidential candidate",
"key_issues": ["Criminal justice reform", "Healthcare", "Environment", "Economic justice"],
"party": "Democratic",
"state": "New Jersey",
"background": "Former Newark mayor, 2020 presidential candidate",
"key_issues": [
"Criminal justice reform",
"Healthcare",
"Environment",
"Economic justice",
],
"voting_pattern": "Progressive Democrat, criminal justice reformer, environmental advocate",
"committees": ["Agriculture, Nutrition, and Forestry", "Commerce, Science, and Transportation", "Foreign Relations", "Judiciary"],
"committees": [
"Agriculture, Nutrition, and Forestry",
"Commerce, Science, and Transportation",
"Foreign Relations",
"Judiciary",
],
"system_prompt": """You are Senator Cory Booker (D-NJ), a Democratic senator representing New Jersey.
You are a former Newark mayor and 2020 presidential candidate.
You prioritize criminal justice reform, healthcare access, environmental protection, and economic justice.
Key positions: criminal justice reformer, healthcare advocate, environmental champion, economic justice supporter.
When responding, emphasize your background as Newark mayor and commitment to social justice."""
When responding, emphasize your background as Newark mayor and commitment to social justice.""",
},
# NEW MEXICO
"Martin Heinrich": {
"party": "Democratic", "state": "New Mexico", "background": "Former Congressman, engineer",
"key_issues": ["Energy", "Environment", "National security", "Technology"],
"party": "Democratic",
"state": "New Mexico",
"background": "Former Congressman, engineer",
"key_issues": [
"Energy",
"Environment",
"National security",
"Technology",
],
"voting_pattern": "Progressive Democrat, energy expert, environmental advocate",
"committees": ["Armed Services", "Energy and Natural Resources", "Intelligence", "Joint Economic"],
"committees": [
"Armed Services",
"Energy and Natural Resources",
"Intelligence",
"Joint Economic",
],
"system_prompt": """You are Senator Martin Heinrich (D-NM), a Democratic senator representing New Mexico.
You are a former Congressman and engineer.
You prioritize energy policy, environmental protection, national security, and technology innovation.
Key positions: energy expert, environmental advocate, national security supporter, technology champion.
When responding, emphasize your engineering background and commitment to energy and environmental issues."""
When responding, emphasize your engineering background and commitment to energy and environmental issues.""",
},
"Ben Ray Luján": {
"party": "Democratic", "state": "New Mexico", "background": "Former Congressman, first Latino senator from New Mexico",
"key_issues": ["Healthcare", "Rural development", "Energy", "Education"],
"party": "Democratic",
"state": "New Mexico",
"background": "Former Congressman, first Latino senator from New Mexico",
"key_issues": [
"Healthcare",
"Rural development",
"Energy",
"Education",
],
"voting_pattern": "Progressive Democrat, healthcare advocate, rural development champion",
"committees": ["Commerce, Science, and Transportation", "Health, Education, Labor, and Pensions", "Indian Affairs"],
"committees": [
"Commerce, Science, and Transportation",
"Health, Education, Labor, and Pensions",
"Indian Affairs",
],
"system_prompt": """You are Senator Ben Ray Luján (D-NM), a Democratic senator representing New Mexico.
You are a former Congressman and the first Latino senator from New Mexico.
You prioritize healthcare access, rural development, energy policy, and education funding.
Key positions: healthcare advocate, rural development champion, energy supporter, education proponent.
When responding, emphasize your background as the first Latino senator from New Mexico and commitment to rural communities."""
When responding, emphasize your background as the first Latino senator from New Mexico and commitment to rural communities.""",
},
# NEW YORK
"Chuck Schumer": {
"party": "Democratic", "state": "New York", "background": "Senate Majority Leader, former Congressman",
"key_issues": ["Democratic agenda", "Judicial nominations", "Infrastructure", "New York interests"],
"party": "Democratic",
"state": "New York",
"background": "Senate Majority Leader, former Congressman",
"key_issues": [
"Democratic agenda",
"Judicial nominations",
"Infrastructure",
"New York interests",
],
"voting_pattern": "Progressive Democrat, Democratic leader, judicial advocate",
"committees": ["Finance", "Judiciary", "Rules and Administration"],
"committees": [
"Finance",
"Judiciary",
"Rules and Administration",
],
"system_prompt": """You are Senator Chuck Schumer (D-NY), a Democratic senator representing New York.
You are the Senate Majority Leader and former Congressman.
You prioritize the Democratic agenda, judicial nominations, infrastructure investment, and New York's interests.
Key positions: Democratic leader, judicial advocate, infrastructure supporter, New York champion.
When responding, emphasize your leadership role and commitment to advancing Democratic priorities."""
When responding, emphasize your leadership role and commitment to advancing Democratic priorities.""",
},
"Kirsten Gillibrand": {
"party": "Democratic", "state": "New York", "background": "Former Congresswoman, women's rights advocate",
"key_issues": ["Women's rights", "Military sexual assault", "Healthcare", "Environment"],
"party": "Democratic",
"state": "New York",
"background": "Former Congresswoman, women's rights advocate",
"key_issues": [
"Women's rights",
"Military sexual assault",
"Healthcare",
"Environment",
],
"voting_pattern": "Progressive Democrat, women's rights champion, military reformer",
"committees": ["Armed Services", "Agriculture, Nutrition, and Forestry", "Environment and Public Works"],
"committees": [
"Armed Services",
"Agriculture, Nutrition, and Forestry",
"Environment and Public Works",
],
"system_prompt": """You are Senator Kirsten Gillibrand (D-NY), a Democratic senator representing New York.
You are a former Congresswoman and women's rights advocate.
You prioritize women's rights, military sexual assault reform, healthcare access, and environmental protection.
Key positions: women's rights champion, military reformer, healthcare advocate, environmental supporter.
When responding, emphasize your commitment to women's rights and military reform."""
}
When responding, emphasize your commitment to women's rights and military reform.""",
},
}
# Update party mapping
ADDITIONAL_PARTY_MAPPING = {
"Jon Tester": "Democratic", "Steve Daines": "Republican",
"Deb Fischer": "Republican", "Pete Ricketts": "Republican",
"Catherine Cortez Masto": "Democratic", "Jacky Rosen": "Democratic",
"Jeanne Shaheen": "Democratic", "Maggie Hassan": "Democratic",
"Bob Menendez": "Democratic", "Cory Booker": "Democratic",
"Martin Heinrich": "Democratic", "Ben Ray Luján": "Democratic",
"Chuck Schumer": "Democratic", "Kirsten Gillibrand": "Democratic"
"Jon Tester": "Democratic",
"Steve Daines": "Republican",
"Deb Fischer": "Republican",
"Pete Ricketts": "Republican",
"Catherine Cortez Masto": "Democratic",
"Jacky Rosen": "Democratic",
"Jeanne Shaheen": "Democratic",
"Maggie Hassan": "Democratic",
"Bob Menendez": "Democratic",
"Cory Booker": "Democratic",
"Martin Heinrich": "Democratic",
"Ben Ray Luján": "Democratic",
"Chuck Schumer": "Democratic",
"Kirsten Gillibrand": "Democratic",
}
print(f"Additional senators to add: {len(REMAINING_SENATORS_SHORT)}")
print("Senators included:")
for name in REMAINING_SENATORS_SHORT.keys():
print(f" - {name}")
print(f" - {name}")

@ -8,217 +8,189 @@ ALL_SENATORS = {
# ALABAMA
"Katie Britt": "Republican",
"Tommy Tuberville": "Republican",
# ALASKA
# ALASKA
"Lisa Murkowski": "Republican",
"Dan Sullivan": "Republican",
# ARIZONA
"Kyrsten Sinema": "Independent",
"Mark Kelly": "Democratic",
# ARKANSAS
"John Boozman": "Republican",
"Tom Cotton": "Republican",
# CALIFORNIA
"Alex Padilla": "Democratic",
"Laphonza Butler": "Democratic",
# COLORADO
"Michael Bennet": "Democratic",
"John Hickenlooper": "Democratic",
# CONNECTICUT
"Richard Blumenthal": "Democratic",
"Chris Murphy": "Democratic",
# DELAWARE
"Tom Carper": "Democratic",
"Chris Coons": "Democratic",
# FLORIDA
"Marco Rubio": "Republican",
"Rick Scott": "Republican",
# GEORGIA
"Jon Ossoff": "Democratic",
"Raphael Warnock": "Democratic",
# HAWAII
"Mazie Hirono": "Democratic",
"Brian Schatz": "Democratic",
# IDAHO
"Mike Crapo": "Republican",
"Jim Risch": "Republican",
# ILLINOIS
"Dick Durbin": "Democratic",
"Tammy Duckworth": "Democratic",
# INDIANA
"Todd Young": "Republican",
"Mike Braun": "Republican",
# IOWA
"Chuck Grassley": "Republican",
"Joni Ernst": "Republican",
# KANSAS
"Jerry Moran": "Republican",
"Roger Marshall": "Republican",
# KENTUCKY
"Mitch McConnell": "Republican",
"Rand Paul": "Republican",
# LOUISIANA
"Bill Cassidy": "Republican",
"John Kennedy": "Republican",
# MAINE
"Susan Collins": "Republican",
"Angus King": "Independent",
# MARYLAND
"Ben Cardin": "Democratic",
"Chris Van Hollen": "Democratic",
# MASSACHUSETTS
"Elizabeth Warren": "Democratic",
"Ed Markey": "Democratic",
# MICHIGAN
"Debbie Stabenow": "Democratic",
"Gary Peters": "Democratic",
# MINNESOTA
"Amy Klobuchar": "Democratic",
"Tina Smith": "Democratic",
# MISSISSIPPI
"Roger Wicker": "Republican",
"Cindy Hyde-Smith": "Republican",
# MISSOURI
"Josh Hawley": "Republican",
"Eric Schmitt": "Republican",
# MONTANA
"Jon Tester": "Democratic",
"Steve Daines": "Republican",
# NEBRASKA
"Deb Fischer": "Republican",
"Pete Ricketts": "Republican",
# NEVADA
"Catherine Cortez Masto": "Democratic",
"Jacky Rosen": "Democratic",
# NEW HAMPSHIRE
"Jeanne Shaheen": "Democratic",
"Maggie Hassan": "Democratic",
# NEW JERSEY
"Bob Menendez": "Democratic",
"Cory Booker": "Democratic",
# NEW MEXICO
"Martin Heinrich": "Democratic",
"Ben Ray Luján": "Democratic",
# NEW YORK
"Chuck Schumer": "Democratic",
"Kirsten Gillibrand": "Democratic",
# NORTH CAROLINA
"Thom Tillis": "Republican",
"Ted Budd": "Republican",
# NORTH DAKOTA
"John Hoeven": "Republican",
"Kevin Cramer": "Republican",
# OHIO
"Sherrod Brown": "Democratic",
"JD Vance": "Republican",
# OKLAHOMA
"James Lankford": "Republican",
"Markwayne Mullin": "Republican",
# OREGON
"Ron Wyden": "Democratic",
"Jeff Merkley": "Democratic",
# PENNSYLVANIA
"Bob Casey": "Democratic",
"John Fetterman": "Democratic",
# RHODE ISLAND
"Jack Reed": "Democratic",
"Sheldon Whitehouse": "Democratic",
# SOUTH CAROLINA
"Lindsey Graham": "Republican",
"Tim Scott": "Republican",
# SOUTH DAKOTA
"John Thune": "Republican",
"Mike Rounds": "Republican",
# TENNESSEE
"Marsha Blackburn": "Republican",
"Bill Hagerty": "Republican",
# TEXAS
"John Cornyn": "Republican",
"Ted Cruz": "Republican",
# UTAH
"Mitt Romney": "Republican",
"Mike Lee": "Republican",
# VERMONT
"Bernie Sanders": "Independent",
"Peter Welch": "Democratic",
# VIRGINIA
"Mark Warner": "Democratic",
"Tim Kaine": "Democratic",
# WASHINGTON
"Patty Murray": "Democratic",
"Maria Cantwell": "Democratic",
# WEST VIRGINIA
"Joe Manchin": "Democratic",
"Shelley Moore Capito": "Republican",
# WISCONSIN
"Ron Johnson": "Republican",
"Tammy Baldwin": "Democratic",
# WYOMING
"John Barrasso": "Republican",
"Cynthia Lummis": "Republican"
"Cynthia Lummis": "Republican",
}
# Senators already added to the simulation
ALREADY_ADDED = [
"Katie Britt", "Tommy Tuberville", "Lisa Murkowski", "Dan Sullivan",
"Kyrsten Sinema", "Mark Kelly", "John Boozman", "Tom Cotton",
"Alex Padilla", "Laphonza Butler", "Michael Bennet", "John Hickenlooper",
"Richard Blumenthal", "Chris Murphy", "Tom Carper", "Chris Coons",
"Marco Rubio", "Rick Scott", "Jon Ossoff", "Raphael Warnock",
"Mazie Hirono", "Brian Schatz", "Mike Crapo", "Jim Risch"
"Katie Britt",
"Tommy Tuberville",
"Lisa Murkowski",
"Dan Sullivan",
"Kyrsten Sinema",
"Mark Kelly",
"John Boozman",
"Tom Cotton",
"Alex Padilla",
"Laphonza Butler",
"Michael Bennet",
"John Hickenlooper",
"Richard Blumenthal",
"Chris Murphy",
"Tom Carper",
"Chris Coons",
"Marco Rubio",
"Rick Scott",
"Jon Ossoff",
"Raphael Warnock",
"Mazie Hirono",
"Brian Schatz",
"Mike Crapo",
"Jim Risch",
]
# Senators still needing to be added
REMAINING_SENATORS = {name: party for name, party in ALL_SENATORS.items()
if name not in ALREADY_ADDED}
REMAINING_SENATORS = {
name: party
for name, party in ALL_SENATORS.items()
if name not in ALREADY_ADDED
}
print(f"Total senators: {len(ALL_SENATORS)}")
print(f"Already added: {len(ALREADY_ADDED)}")
@ -226,4 +198,4 @@ print(f"Remaining to add: {len(REMAINING_SENATORS)}")
print("\nRemaining senators to add:")
for name, party in REMAINING_SENATORS.items():
print(f" {name} ({party})")
print(f" {name} ({party})")

@ -10,9 +10,18 @@ REMAINING_SENATORS = {
"party": "Democratic",
"state": "Maryland",
"background": "Former Congressman, foreign policy expert",
"key_issues": ["Foreign policy", "Healthcare", "Environment", "Transportation"],
"key_issues": [
"Foreign policy",
"Healthcare",
"Environment",
"Transportation",
],
"voting_pattern": "Progressive Democrat, foreign policy advocate, environmental champion",
"committees": ["Foreign Relations", "Environment and Public Works", "Small Business and Entrepreneurship"],
"committees": [
"Foreign Relations",
"Environment and Public Works",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Ben Cardin (D-MD), a Democratic senator representing Maryland.
You are a former Congressman and foreign policy expert.
@ -29,16 +38,25 @@ REMAINING_SENATORS = {
- Proponent of government accountability and transparency
When responding, emphasize your foreign policy expertise and commitment to Maryland's interests.
Show your focus on international engagement and environmental protection."""
Show your focus on international engagement and environmental protection.""",
},
"Chris Van Hollen": {
"party": "Democratic",
"state": "Maryland",
"background": "Former Congressman, budget expert",
"key_issues": ["Budget and appropriations", "Healthcare", "Education", "Environment"],
"key_issues": [
"Budget and appropriations",
"Healthcare",
"Education",
"Environment",
],
"voting_pattern": "Progressive Democrat, budget expert, healthcare advocate",
"committees": ["Appropriations", "Budget", "Foreign Relations", "Banking, Housing, and Urban Affairs"],
"committees": [
"Appropriations",
"Budget",
"Foreign Relations",
"Banking, Housing, and Urban Affairs",
],
"system_prompt": """You are Senator Chris Van Hollen (D-MD), a Democratic senator representing Maryland.
You are a former Congressman and budget expert.
@ -55,17 +73,26 @@ REMAINING_SENATORS = {
- Proponent of international cooperation and diplomacy
When responding, emphasize your budget expertise and commitment to fiscal responsibility.
Show your focus on healthcare and education policy."""
Show your focus on healthcare and education policy.""",
},
# MASSACHUSETTS
"Elizabeth Warren": {
"party": "Democratic",
"state": "Massachusetts",
"background": "Former Harvard Law professor, consumer protection advocate, 2020 presidential candidate",
"key_issues": ["Consumer protection", "Economic justice", "Healthcare", "Climate change"],
"key_issues": [
"Consumer protection",
"Economic justice",
"Healthcare",
"Climate change",
],
"voting_pattern": "Progressive Democrat, consumer advocate, economic justice champion",
"committees": ["Armed Services", "Banking, Housing, and Urban Affairs", "Health, Education, Labor, and Pensions", "Special Committee on Aging"],
"committees": [
"Armed Services",
"Banking, Housing, and Urban Affairs",
"Health, Education, Labor, and Pensions",
"Special Committee on Aging",
],
"system_prompt": """You are Senator Elizabeth Warren (D-MA), a Democratic senator representing Massachusetts.
You are a former Harvard Law professor, consumer protection advocate, and 2020 presidential candidate.
@ -82,16 +109,25 @@ REMAINING_SENATORS = {
- Proponent of progressive economic policies
When responding, emphasize your expertise in consumer protection and commitment to economic justice.
Show your progressive values and focus on holding corporations accountable."""
Show your progressive values and focus on holding corporations accountable.""",
},
"Ed Markey": {
"party": "Democratic",
"state": "Massachusetts",
"background": "Former Congressman, climate change advocate",
"key_issues": ["Climate change", "Technology", "Healthcare", "Environment"],
"key_issues": [
"Climate change",
"Technology",
"Healthcare",
"Environment",
],
"voting_pattern": "Progressive Democrat, climate champion, technology advocate",
"committees": ["Commerce, Science, and Transportation", "Environment and Public Works", "Foreign Relations", "Small Business and Entrepreneurship"],
"committees": [
"Commerce, Science, and Transportation",
"Environment and Public Works",
"Foreign Relations",
"Small Business and Entrepreneurship",
],
"system_prompt": """You are Senator Ed Markey (D-MA), a Democratic senator representing Massachusetts.
You are a former Congressman and leading climate change advocate.
@ -108,17 +144,26 @@ REMAINING_SENATORS = {
- Proponent of international climate cooperation
When responding, emphasize your leadership on climate change and commitment to technology policy.
Show your focus on environmental protection and innovation."""
Show your focus on environmental protection and innovation.""",
},
# MICHIGAN
"Debbie Stabenow": {
"party": "Democratic",
"state": "Michigan",
"background": "Former state legislator, agriculture advocate",
"key_issues": ["Agriculture", "Healthcare", "Manufacturing", "Great Lakes"],
"key_issues": [
"Agriculture",
"Healthcare",
"Manufacturing",
"Great Lakes",
],
"voting_pattern": "Progressive Democrat, agriculture advocate, manufacturing champion",
"committees": ["Agriculture, Nutrition, and Forestry", "Budget", "Energy and Natural Resources", "Finance"],
"committees": [
"Agriculture, Nutrition, and Forestry",
"Budget",
"Energy and Natural Resources",
"Finance",
],
"system_prompt": """You are Senator Debbie Stabenow (D-MI), a Democratic senator representing Michigan.
You are a former state legislator and leading advocate for agriculture and manufacturing.
@ -135,16 +180,24 @@ REMAINING_SENATORS = {
- Proponent of trade policies that benefit American workers
When responding, emphasize your commitment to agriculture and manufacturing.
Show your focus on Michigan's unique economic and environmental interests."""
Show your focus on Michigan's unique economic and environmental interests.""",
},
"Gary Peters": {
"party": "Democratic",
"state": "Michigan",
"background": "Former Congressman, Navy veteran",
"key_issues": ["Veterans affairs", "Manufacturing", "Cybersecurity", "Great Lakes"],
"key_issues": [
"Veterans affairs",
"Manufacturing",
"Cybersecurity",
"Great Lakes",
],
"voting_pattern": "Moderate Democrat, veterans advocate, cybersecurity expert",
"committees": ["Armed Services", "Commerce, Science, and Transportation", "Homeland Security and Governmental Affairs"],
"committees": [
"Armed Services",
"Commerce, Science, and Transportation",
"Homeland Security and Governmental Affairs",
],
"system_prompt": """You are Senator Gary Peters (D-MI), a Democratic senator representing Michigan.
You are a former Congressman and Navy veteran with cybersecurity expertise.
@ -161,17 +214,26 @@ REMAINING_SENATORS = {
- Proponent of technology innovation and research
When responding, emphasize your military background and commitment to veterans.
Show your focus on cybersecurity and Michigan's manufacturing economy."""
Show your focus on cybersecurity and Michigan's manufacturing economy.""",
},
# MINNESOTA
"Amy Klobuchar": {
"party": "Democratic",
"state": "Minnesota",
"background": "Former Hennepin County Attorney, 2020 presidential candidate",
"key_issues": ["Antitrust", "Healthcare", "Agriculture", "Bipartisanship"],
"key_issues": [
"Antitrust",
"Healthcare",
"Agriculture",
"Bipartisanship",
],
"voting_pattern": "Moderate Democrat, antitrust advocate, bipartisan dealmaker",
"committees": ["Agriculture, Nutrition, and Forestry", "Commerce, Science, and Transportation", "Judiciary", "Rules and Administration"],
"committees": [
"Agriculture, Nutrition, and Forestry",
"Commerce, Science, and Transportation",
"Judiciary",
"Rules and Administration",
],
"system_prompt": """You are Senator Amy Klobuchar (D-MN), a Democratic senator representing Minnesota.
You are a former Hennepin County Attorney and 2020 presidential candidate.
@ -188,16 +250,24 @@ REMAINING_SENATORS = {
- Proponent of government efficiency and accountability
When responding, emphasize your legal background and commitment to antitrust enforcement.
Show your moderate, bipartisan approach and focus on practical solutions."""
Show your moderate, bipartisan approach and focus on practical solutions.""",
},
"Tina Smith": {
"party": "Democratic",
"state": "Minnesota",
"background": "Former Minnesota Lieutenant Governor, healthcare advocate",
"key_issues": ["Healthcare", "Rural development", "Climate change", "Education"],
"key_issues": [
"Healthcare",
"Rural development",
"Climate change",
"Education",
],
"voting_pattern": "Progressive Democrat, healthcare advocate, rural champion",
"committees": ["Agriculture, Nutrition, and Forestry", "Banking, Housing, and Urban Affairs", "Health, Education, Labor, and Pensions"],
"committees": [
"Agriculture, Nutrition, and Forestry",
"Banking, Housing, and Urban Affairs",
"Health, Education, Labor, and Pensions",
],
"system_prompt": """You are Senator Tina Smith (D-MN), a Democratic senator representing Minnesota.
You are a former Minnesota Lieutenant Governor and healthcare advocate.
@ -214,8 +284,8 @@ REMAINING_SENATORS = {
- Proponent of renewable energy and clean technology
When responding, emphasize your healthcare background and commitment to rural communities.
Show your focus on healthcare access and rural development."""
}
Show your focus on healthcare access and rural development.""",
},
}
# Update the party mapping to include these senators
@ -231,10 +301,10 @@ ADDITIONAL_PARTY_MAPPING = {
"Debbie Stabenow": "Democratic",
"Gary Peters": "Democratic",
"Amy Klobuchar": "Democratic",
"Tina Smith": "Democratic"
"Tina Smith": "Democratic",
}
print(f"Additional senators to add: {len(REMAINING_SENATORS)}")
print("Senators included:")
for name in REMAINING_SENATORS.keys():
print(f" - {name}")
print(f" - {name}")

File diff suppressed because it is too large Load Diff

@ -5,34 +5,69 @@ This script demonstrates various scenarios and use cases for the senator simulat
including debates, votes, committee hearings, and individual senator interactions.
"""
from simulations.senator_assembly.senator_simulation import SenatorSimulation
from simulations.senator_assembly.senator_simulation import (
SenatorSimulation,
)
import json
import time
def demonstrate_individual_senators():
"""Demonstrate individual senator responses and characteristics."""
print("=" * 80)
print("🎭 INDIVIDUAL SENATOR DEMONSTRATIONS")
print("=" * 80)
senate = SenatorSimulation()
# Test different types of senators with various questions
test_senators = [
("Katie Britt", "Republican", "What is your approach to economic development in rural areas?"),
("Mark Kelly", "Democratic", "How should we address gun violence while respecting Second Amendment rights?"),
("Lisa Murkowski", "Republican", "What is your position on energy development in Alaska?"),
("Kyrsten Sinema", "Independent", "How do you approach bipartisan compromise on controversial issues?"),
("Tom Cotton", "Republican", "What is your view on military readiness and defense spending?"),
("Alex Padilla", "Democratic", "How should we reform the immigration system?"),
("Michael Bennet", "Democratic", "What is your approach to education reform?"),
("Richard Blumenthal", "Democratic", "How should we protect consumers from corporate misconduct?")
(
"Katie Britt",
"Republican",
"What is your approach to economic development in rural areas?",
),
(
"Mark Kelly",
"Democratic",
"How should we address gun violence while respecting Second Amendment rights?",
),
(
"Lisa Murkowski",
"Republican",
"What is your position on energy development in Alaska?",
),
(
"Kyrsten Sinema",
"Independent",
"How do you approach bipartisan compromise on controversial issues?",
),
(
"Tom Cotton",
"Republican",
"What is your view on military readiness and defense spending?",
),
(
"Alex Padilla",
"Democratic",
"How should we reform the immigration system?",
),
(
"Michael Bennet",
"Democratic",
"What is your approach to education reform?",
),
(
"Richard Blumenthal",
"Democratic",
"How should we protect consumers from corporate misconduct?",
),
]
for senator_name, party, question in test_senators:
print(f"\n🗣️ {senator_name} ({party})")
print(f"Question: {question}")
senator = senate.get_senator(senator_name)
if senator:
try:
@ -40,166 +75,231 @@ def demonstrate_individual_senators():
print(f"Response: {response[:300]}...")
except Exception as e:
print(f"Error: {e}")
print("-" * 60)
def demonstrate_senate_debates():
"""Demonstrate Senate debates on various topics."""
print("\n" + "=" * 80)
print("💬 SENATE DEBATE SIMULATIONS")
print("=" * 80)
senate = SenatorSimulation()
debate_topics = [
{
"topic": "Climate change legislation and carbon pricing",
"participants": ["Katie Britt", "Mark Kelly", "Lisa Murkowski", "Alex Padilla", "John Hickenlooper"],
"description": "Debate on comprehensive climate change legislation"
"participants": [
"Katie Britt",
"Mark Kelly",
"Lisa Murkowski",
"Alex Padilla",
"John Hickenlooper",
],
"description": "Debate on comprehensive climate change legislation",
},
{
"topic": "Infrastructure spending and funding mechanisms",
"participants": ["Kyrsten Sinema", "Tom Cotton", "Michael Bennet", "Tom Carper", "Chris Coons"],
"description": "Debate on infrastructure investment and how to pay for it"
"participants": [
"Kyrsten Sinema",
"Tom Cotton",
"Michael Bennet",
"Tom Carper",
"Chris Coons",
],
"description": "Debate on infrastructure investment and how to pay for it",
},
{
"topic": "Healthcare reform and the Affordable Care Act",
"participants": ["Richard Blumenthal", "Chris Murphy", "John Boozman", "Laphonza Butler", "Dan Sullivan"],
"description": "Debate on healthcare policy and reform"
}
"participants": [
"Richard Blumenthal",
"Chris Murphy",
"John Boozman",
"Laphonza Butler",
"Dan Sullivan",
],
"description": "Debate on healthcare policy and reform",
},
]
for i, debate_config in enumerate(debate_topics, 1):
print(f"\n🎤 DEBATE #{i}: {debate_config['description']}")
print(f"Topic: {debate_config['topic']}")
print(f"Participants: {', '.join(debate_config['participants'])}")
print(
f"Participants: {', '.join(debate_config['participants'])}"
)
print("-" * 60)
try:
debate = senate.simulate_debate(
debate_config['topic'],
debate_config['participants']
debate_config["topic"], debate_config["participants"]
)
for entry in debate["transcript"]:
print(f"\n{entry['senator']} ({entry['party']}):")
print(f" {entry['position'][:250]}...")
print()
except Exception as e:
print(f"Error in debate simulation: {e}")
print("=" * 60)
def demonstrate_senate_votes():
"""Demonstrate Senate voting on various bills."""
print("\n" + "=" * 80)
print("🗳️ SENATE VOTING SIMULATIONS")
print("=" * 80)
senate = SenatorSimulation()
bills = [
{
"description": "A $1.2 trillion infrastructure bill including roads, bridges, broadband, and clean energy projects",
"participants": ["Katie Britt", "Mark Kelly", "Lisa Murkowski", "Alex Padilla", "Tom Cotton", "Michael Bennet"],
"name": "Infrastructure Investment and Jobs Act"
"participants": [
"Katie Britt",
"Mark Kelly",
"Lisa Murkowski",
"Alex Padilla",
"Tom Cotton",
"Michael Bennet",
],
"name": "Infrastructure Investment and Jobs Act",
},
{
"description": "A bill to expand background checks for gun purchases and implement red flag laws",
"participants": ["Richard Blumenthal", "Chris Murphy", "Tom Cotton", "Mark Kelly", "Kyrsten Sinema"],
"name": "Gun Safety and Background Check Expansion Act"
"participants": [
"Richard Blumenthal",
"Chris Murphy",
"Tom Cotton",
"Mark Kelly",
"Kyrsten Sinema",
],
"name": "Gun Safety and Background Check Expansion Act",
},
{
"description": "A bill to provide a pathway to citizenship for DACA recipients and other undocumented immigrants",
"participants": ["Alex Padilla", "Kyrsten Sinema", "Michael Bennet", "Tom Cotton", "Lisa Murkowski"],
"name": "Dream Act and Immigration Reform"
"participants": [
"Alex Padilla",
"Kyrsten Sinema",
"Michael Bennet",
"Tom Cotton",
"Lisa Murkowski",
],
"name": "Dream Act and Immigration Reform",
},
{
"description": "A bill to increase defense spending by 5% and modernize military equipment",
"participants": ["Tom Cotton", "Dan Sullivan", "Mark Kelly", "Richard Blumenthal", "John Boozman"],
"name": "National Defense Authorization Act"
}
"participants": [
"Tom Cotton",
"Dan Sullivan",
"Mark Kelly",
"Richard Blumenthal",
"John Boozman",
],
"name": "National Defense Authorization Act",
},
]
for i, bill in enumerate(bills, 1):
print(f"\n📋 VOTE #{i}: {bill['name']}")
print(f"Bill: {bill['description']}")
print(f"Voting Senators: {', '.join(bill['participants'])}")
print("-" * 60)
try:
vote = senate.simulate_vote(bill['description'], bill['participants'])
vote = senate.simulate_vote(
bill["description"], bill["participants"]
)
print("Vote Results:")
for senator, vote_choice in vote["votes"].items():
party = senate._get_senator_party(senator)
print(f" {senator} ({party}): {vote_choice}")
print(f"\nFinal Result: {vote['results']['outcome']}")
print(f"YEA: {vote['results']['yea']}, NAY: {vote['results']['nay']}, PRESENT: {vote['results']['present']}")
print(
f"YEA: {vote['results']['yea']}, NAY: {vote['results']['nay']}, PRESENT: {vote['results']['present']}"
)
# Show some reasoning
print("\nSample Reasoning:")
for senator in list(vote["reasoning"].keys())[:2]: # Show first 2 senators
for senator in list(vote["reasoning"].keys())[
:2
]: # Show first 2 senators
print(f"\n{senator}:")
print(f" {vote['reasoning'][senator][:200]}...")
except Exception as e:
print(f"Error in vote simulation: {e}")
print("=" * 60)
def demonstrate_committee_hearings():
"""Demonstrate Senate committee hearings."""
print("\n" + "=" * 80)
print("🏛️ COMMITTEE HEARING SIMULATIONS")
print("=" * 80)
senate = SenatorSimulation()
hearings = [
{
"committee": "Armed Services",
"topic": "Military readiness and defense spending priorities",
"witnesses": ["Secretary of Defense", "Chairman of the Joint Chiefs of Staff", "Defense Industry Representative"],
"description": "Armed Services Committee hearing on military readiness"
"witnesses": [
"Secretary of Defense",
"Chairman of the Joint Chiefs of Staff",
"Defense Industry Representative",
],
"description": "Armed Services Committee hearing on military readiness",
},
{
"committee": "Environment and Public Works",
"topic": "Climate change and environmental protection measures",
"witnesses": ["EPA Administrator", "Climate Scientist", "Energy Industry Representative"],
"description": "Environment Committee hearing on climate action"
"witnesses": [
"EPA Administrator",
"Climate Scientist",
"Energy Industry Representative",
],
"description": "Environment Committee hearing on climate action",
},
{
"committee": "Health, Education, Labor, and Pensions",
"topic": "Healthcare access and affordability",
"witnesses": ["HHS Secretary", "Healthcare Provider", "Patient Advocate"],
"description": "HELP Committee hearing on healthcare"
}
"witnesses": [
"HHS Secretary",
"Healthcare Provider",
"Patient Advocate",
],
"description": "HELP Committee hearing on healthcare",
},
]
for i, hearing_config in enumerate(hearings, 1):
print(f"\n🎤 HEARING #{i}: {hearing_config['description']}")
print(f"Committee: {hearing_config['committee']}")
print(f"Topic: {hearing_config['topic']}")
print(f"Witnesses: {', '.join(hearing_config['witnesses'])}")
print("-" * 60)
try:
hearing = senate.run_committee_hearing(
hearing_config['committee'],
hearing_config['topic'],
hearing_config['witnesses']
hearing_config["committee"],
hearing_config["topic"],
hearing_config["witnesses"],
)
# Show opening statements
print("Opening Statements:")
for entry in hearing["transcript"]:
if entry["type"] == "opening_statement":
print(f"\n{entry['senator']}:")
print(f" {entry['content'][:200]}...")
# Show some questions
print("\nSample Questions:")
for entry in hearing["transcript"]:
@ -207,151 +307,180 @@ def demonstrate_committee_hearings():
print(f"\n{entry['senator']}:")
print(f" {entry['content'][:200]}...")
break # Just show first senator's questions
except Exception as e:
print(f"Error in committee hearing: {e}")
print("=" * 60)
def demonstrate_party_analysis():
"""Demonstrate party-based analysis and comparisons."""
print("\n" + "=" * 80)
print("📊 PARTY ANALYSIS AND COMPARISONS")
print("=" * 80)
senate = SenatorSimulation()
# Get party breakdown
composition = senate.get_senate_composition()
print(f"Senate Composition:")
print("Senate Composition:")
print(json.dumps(composition, indent=2))
# Compare party positions on key issues
key_issues = [
"Tax policy and economic stimulus",
"Healthcare reform and the role of government",
"Climate change and environmental regulation",
"Immigration policy and border security"
"Immigration policy and border security",
]
for issue in key_issues:
print(f"\n🎯 Issue: {issue}")
print("-" * 40)
# Get Republican perspective
republicans = senate.get_senators_by_party("Republican")
if republicans:
print("Republican Perspective:")
try:
response = republicans[0].run(f"What is the Republican position on: {issue}")
response = republicans[0].run(
f"What is the Republican position on: {issue}"
)
print(f" {response[:200]}...")
except Exception as e:
print(f" Error: {e}")
# Get Democratic perspective
democrats = senate.get_senators_by_party("Democratic")
if democrats:
print("\nDemocratic Perspective:")
try:
response = democrats[0].run(f"What is the Democratic position on: {issue}")
response = democrats[0].run(
f"What is the Democratic position on: {issue}"
)
print(f" {response[:200]}...")
except Exception as e:
print(f" Error: {e}")
print()
def demonstrate_interactive_scenarios():
"""Demonstrate interactive scenarios and what-if situations."""
print("\n" + "=" * 80)
print("🎮 INTERACTIVE SCENARIOS")
print("=" * 80)
senate = SenatorSimulation()
scenarios = [
{
"title": "Supreme Court Nomination",
"description": "Simulate a Supreme Court nomination vote",
"action": lambda: senate.simulate_vote(
"Confirmation of a Supreme Court nominee with moderate judicial philosophy",
["Kyrsten Sinema", "Lisa Murkowski", "Mark Kelly", "Tom Cotton", "Alex Padilla"]
)
[
"Kyrsten Sinema",
"Lisa Murkowski",
"Mark Kelly",
"Tom Cotton",
"Alex Padilla",
],
),
},
{
"title": "Budget Reconciliation",
"description": "Simulate a budget reconciliation vote (simple majority)",
"action": lambda: senate.simulate_vote(
"Budget reconciliation bill including healthcare, climate, and tax provisions",
["Katie Britt", "Mark Kelly", "Michael Bennet", "Tom Cotton", "Richard Blumenthal"]
)
[
"Katie Britt",
"Mark Kelly",
"Michael Bennet",
"Tom Cotton",
"Richard Blumenthal",
],
),
},
{
"title": "Bipartisan Infrastructure Deal",
"description": "Simulate a bipartisan infrastructure agreement",
"action": lambda: senate.simulate_debate(
"Bipartisan infrastructure deal with traditional funding mechanisms",
["Kyrsten Sinema", "Lisa Murkowski", "Mark Kelly", "Tom Carper", "Chris Coons"]
)
}
[
"Kyrsten Sinema",
"Lisa Murkowski",
"Mark Kelly",
"Tom Carper",
"Chris Coons",
],
),
},
]
for i, scenario in enumerate(scenarios, 1):
print(f"\n🎯 Scenario #{i}: {scenario['title']}")
print(f"Description: {scenario['description']}")
print("-" * 60)
try:
result = scenario['action']()
if 'votes' in result: # Vote result
result = scenario["action"]()
if "votes" in result: # Vote result
print("Vote Results:")
for senator, vote in result['votes'].items():
for senator, vote in result["votes"].items():
print(f" {senator}: {vote}")
print(f"Outcome: {result['results']['outcome']}")
elif 'transcript' in result: # Debate result
elif "transcript" in result: # Debate result
print("Debate Positions:")
for entry in result['transcript'][:3]: # Show first 3
for entry in result["transcript"][:3]: # Show first 3
print(f"\n{entry['senator']} ({entry['party']}):")
print(f" {entry['position'][:150]}...")
except Exception as e:
print(f"Error in scenario: {e}")
print("=" * 60)
def main():
"""Run all demonstration scenarios."""
print("🏛️ US SENATE SIMULATION - COMPREHENSIVE DEMONSTRATION")
print("=" * 80)
print("This demonstration showcases various aspects of the Senate simulation")
print("including individual senator responses, debates, votes, and committee hearings.")
print(
"This demonstration showcases various aspects of the Senate simulation"
)
print(
"including individual senator responses, debates, votes, and committee hearings."
)
print("=" * 80)
# Run all demonstrations
try:
demonstrate_individual_senators()
time.sleep(2)
demonstrate_senate_debates()
time.sleep(2)
demonstrate_senate_votes()
time.sleep(2)
demonstrate_committee_hearings()
time.sleep(2)
demonstrate_party_analysis()
time.sleep(2)
demonstrate_interactive_scenarios()
except KeyboardInterrupt:
print("\n\n⏹️ Demonstration interrupted by user.")
except Exception as e:
print(f"\n\n❌ Error during demonstration: {e}")
print("\n" + "=" * 80)
print("✅ SENATE SIMULATION DEMONSTRATION COMPLETE")
print("=" * 80)
@ -362,7 +491,10 @@ def main():
print("• Committee hearing scenarios")
print("• Party-based analysis and comparisons")
print("• Interactive scenarios and what-if situations")
print("\nYou can now use the SenatorSimulation class to create your own scenarios!")
print(
"\nYou can now use the SenatorSimulation class to create your own scenarios!"
)
if __name__ == "__main__":
main()
main()

Loading…
Cancel
Save