Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from textattack.attack_recipes import TextFoolerJin2019
 from textattack.models.wrappers import HuggingFaceModelWrapper
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import textattack
 import torch
 
 # Load Hugging Face model (e.g., distilbert for demo)
@@ -17,8 +18,21 @@ attack = TextFoolerJin2019.build(model_wrapper)
 
 # Function to run attack
 def run_attack(input_text):
+    # Perform the attack
     result = attack.attack(input_text, ground_truth_output=1)
-    return str(result)
+
+    # Extract detailed information
+    skipped_info = ""
+    for goal_function_result in result:
+        if goal_function_result.goal_status == textattack.shared.AttackResultStatus.SKIPPED:
+            skipped_info += f"Skipped Text: {goal_function_result.attacked_text.text}\n"
+    
+    # Include model information
+    model_info = f"Model: {model_name}\n"
+
+    # Format the output
+    output = f"{model_info}\nResult:\n{str(result)}\n\nSkipped Details:\n{skipped_info}"
+    return output
 
 # Gradio UI
 gr.Interface(fn=run_attack,