Update main.py

main.py

@@ -1,15 +1,14 @@
 # main.py
 import uvicorn
 from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+from pydantic import BaseModel, conint, constr
 import threading
 import requests
 import time
 import logging
 import socket
 import random
-import struct
-from typing import Optional, Literal
+from typing import Optional
 from concurrent.futures import ThreadPoolExecutor
 
 logging.basicConfig(level=logging.INFO)
@@ -27,7 +26,7 @@ class AttackConfig(BaseModel):
     port: int = 80
     duration: int = 30  # -1 for unlimited
     threads: int = 100  # -1 for unlimited
-    attack_type: Literal["layer7", "tcp", "udp"] = "layer7"
+    attack_type: constr(regex='^(layer7|tcp|udp)$') = "layer7"
     payload_size: int = 1024  # For Layer 4 attacks
 
 def generate_payload(size: int) -> bytes: