Update app.py

app.py

@@ -18,21 +18,26 @@ 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)
+    try:
+        # Attempt to modify the input text
+        result = attack.attack(input_text, ground_truth_output=1)
 
-    # 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"
+        # Initialize a message to describe what happened
+        attack_info = ""
+        skipped_info = ""
+
+        for goal_function_result in result:
+            if goal_function_result.goal_status == textattack.shared.AttackResultStatus.SUCCEEDED:
+                attack_info += f"Attack Succeeded!\nOriginal Text: {input_text}\nModified Text: {goal_function_result.attacked_text.text}\n"
+            elif goal_function_result.goal_status == textattack.shared.AttackResultStatus.SKIPPED:
+                skipped_info += f"Skipped Text: {goal_function_result.attacked_text.text}\n"
 
-    # Format the output
-    output = f"{model_info}\nResult:\n{str(result)}\n\nSkipped Details:\n{skipped_info}"
-    return output
+        # Format the output
+        output = f"TextAttack Results:\n{attack_info}\nSkipped Details:\n{skipped_info}"
+        return output
+    
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
     
 # Gradio UI
 gr.Interface(fn=run_attack,