import pygame

import random

# Initialize pygame

pygame.init()

# Game constants

SCREEN_WIDTH = 800

SCREEN_HEIGHT = 600

WHITE = (255, 255, 255)

BLACK = (0, 0, 0)

RED = (255, 0, 0)

GREEN = (0, 255, 0)

BLUE = (0, 0, 255)

YELLOW = (255, 255, 0)

# Create the screen object

screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))

pygame.display.set_caption("Avengers Survival Game")

# Player (Avenger) base class

class Avenger(pygame.sprite.Sprite):

def __init__(self, name, color, x, y, health=100):

super().__init__()

self.name = name
 self.image = pygame.Surface((50, 50))

self.image.fill(color)

self.rect = self.image.get_rect()

self.rect.center = (x, y)

self.health = health

self.speed = 5

def update(self, keys):

if keys[pygame.K_LEFT]:

self.rect.x -= self.speed

if keys[pygame.K_RIGHT]:

self.rect.x += self.speed

if keys[pygame.K_UP]:

self.rect.y -= self.speed

if keys[pygame.K_DOWN]:

self.rect.y += self.speed

def attack(self):

pass # Will be overridden in subclasses

# Specific Avengers class with unique abilities

# Iron Man's Repulsor Beam

class IronMan(Avenger):

def __init__(self, x, y):

 super().__init__("Iron Man", BLUE, x, y)

def attack(self):

repulsor = RepulsorBeam(self.rect.centerx, self.rect.top)

all_sprites.add(repulsor)

repulsor_group.add(repulsor)

# Thor's Lightning Strike

class Thor(Avenger):

def __init__(self, x, y):

super().__init__("Thor", YELLOW, x, y)

def attack(self):

lightning = Lightning(self.rect.centerx, self.rect.top)

all_sprites.add(lightning)

lightning_group.add(lightning)

# Captain America's Shield Throw

class CaptainAmerica(Avenger):

def __init__(self, x, y):

super().__init__("Captain America", WHITE, x, y)

def attack(self):

shield = Shield(self.rect.centerx, self.rect.top)

all_sprites.add(shield)
 shield_group.add(shield)

# Hulk's Smash

class Hulk(Avenger):

def __init__(self, x, y):

super().__init__("Hulk", GREEN, x, y)

def attack(self):

smash = Smash(self.rect.centerx, self.rect.top)

all_sprites.add(smash)

smash_group.add(smash)

# Black Widow's Widow's Bite (Electric Strike)

class BlackWidow(Avenger):

def __init__(self, x, y):

super().__init__("Black Widow", RED, x, y)

def attack(self):

bite = WidowBite(self.rect.centerx, self.rect.top)

all_sprites.add(bite)

bite_group.add(bite)

# Hawkeye's Arrow

class Hawkeye(Avenger):

def __init__(self, x, y):

 super().__init__("Hawkeye", BLACK, x, y)

def attack(self):

arrow = Arrow(self.rect.centerx, self.rect.top)

all_sprites.add(arrow)

arrow_group.add(arrow)

# Projectile classes for each power

# Iron Man's Repulsor Beam

class RepulsorBeam(pygame.sprite.Sprite):

def __init__(self, x, y):

super().__init__()

self.image = pygame.Surface((5, 20))

self.image.fill(WHITE) # Repulsor beam color

self.rect = self.image.get_rect()

self.rect.center = (x, y)

self.speed = 10

def update(self):

self.rect.y -= self.speed

if self.rect.bottom < 0:

self.kill()

# Thor's Lightning
class Lightning(pygame.sprite.Sprite):

def __init__(self, x, y):

super().__init__()

self.image = pygame.Surface((5, 20))

self.image.fill(RED)

self.rect = self.image.get_rect()

self.rect.center = (x, y)

self.speed = 10

def update(self):

self.rect.y -= self.speed

if self.rect.bottom < 0:

self.kill()

# Captain America's Shield

class Shield(pygame.sprite.Sprite):

def __init__(self, x, y):

super().__init__()

self.image = pygame.Surface((10, 30))

self.image.fill(BLUE)

self.rect = self.image.get_rect()

self.rect.center = (x, y)

self.speed = 8

def update(self):
 self.rect.y -= self.speed

if self.rect.bottom < 0:

self.kill()

# Hulk's Smash

class Smash(pygame.sprite.Sprite):

def __init__(self, x, y):

super().__init__()

self.image = pygame.Surface((50, 50))

self.image.fill(GREEN)

self.rect = self.image.get_rect()

self.rect.center = (x, y)

self.speed = 10

def update(self):

self.rect.y -= self.speed

if self.rect.bottom < 0:

self.kill()

# Black Widow's Widow Bite

class WidowBite(pygame.sprite.Sprite):

def __init__(self, x, y):

super().__init__()

self.image = pygame.Surface((5, 20))

self.image.fill(RED)
 self.rect = self.image.get_rect()

self.rect.center = (x, y)

self.speed = 8

def update(self):

self.rect.y -= self.speed

if self.rect.bottom < 0:

self.kill()

# Hawkeye's Arrow

class Arrow(pygame.sprite.Sprite):

def __init__(self, x, y):

super().__init__()

self.image = pygame.Surface((5, 20))

self.image.fill(BLACK)

self.rect = self.image.get_rect()

self.rect.center = (x, y)

self.speed = 10

def update(self):

self.rect.y -= self.speed

if self.rect.bottom < 0:

self.kill()

# Initialize player characters

ironman = IronMan(SCREEN_WIDTH // 2, SCREEN_HEIGHT - 50)

thor = Thor(SCREEN_WIDTH // 3, SCREEN_HEIGHT - 50)

cap_america = CaptainAmerica(SCREEN_WIDTH // 4, SCREEN_HEIGHT - 50)

hulk = Hulk(SCREEN_WIDTH // 5, SCREEN_HEIGHT - 50)

black_widow = BlackWidow(SCREEN_WIDTH // 6, SCREEN_HEIGHT - 50)

hawkeye = Hawkeye(SCREEN_WIDTH // 7, SCREEN_HEIGHT - 50)

all_sprites = pygame.sprite.Group()

all_sprites.add(ironman, thor, cap_america, hulk, black_widow, hawkeye)

# Groups for projectiles

repulsor_group = pygame.sprite.Group()

lightning_group = pygame.sprite.Group()

shield_group = pygame.sprite.Group()

smash_group = pygame.sprite.Group()

bite_group = pygame.sprite.Group()

arrow_group = pygame.sprite.Group()

# Game loop

running = True

while running:

screen.fill(BLACK)

# Handle events

for event in pygame.event.get():

 if event.type == pygame.QUIT:

running = False

# Key press handling for movement

keys = pygame.key.get_pressed()

# Update player positions

all_sprites.update(keys)

# Example attacks (Thor's lightning, Iron Man's repulsor)

if keys[pygame.K_SPACE]:

ironman.attack() # Iron Man attack (Repulsor)

if keys[pygame.K_RETURN]:

thor.attack() # Thor attack (Lightning)

# Draw everything

all_sprites.draw(screen)

repulsor_group.draw(screen)

lightning_group.draw(screen)

shield_group.draw(screen)

smash_group.draw(screen)

bite_group.draw(screen)

arrow_group.draw(screen)

pygame.display.flip()
 pygame.time.Clock().tick(60)

# Quit pygame

pygame.quit()