Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import pipeline
 from typing import List, Dict
 
 # Initialize the Hugging Face pipeline (make sure to replace with your model name)
-model_name = "your_huggingface_model_name"
+model_name = "your_huggingface_model_name"  # Ensure to use a valid model
 try:
     generator = pipeline("text-generation", model=model_name)
 except Exception as e:
@@ -12,11 +12,9 @@ except Exception as e:
 def generate_attack(prompt: str, history: List[Dict[str, str]]) -> List[str]:
     """
     Simulates a Blackhat AI scenario by generating attack strategies and potential impacts.
-
     Args:
         prompt (str): The user's input to the simulator.
         history (List[Dict]): The user's message history with timestamps.
-
     Returns:
         List[str]: A list of attack responses from the AI.
     """
@@ -34,6 +32,7 @@ def generate_attack(prompt: str, history: List[Dict[str, str]]) -> List[str]:
         if "assistant" in val:
             messages.append({"role": "assistant", "content": val["assistant"]})
 
+    # Append the current user prompt
     messages.append({"role": "user", "content": prompt})
 
     # Generate a response using the Hugging Face model