Update app.py

app.py

@@ -18,6 +18,7 @@ attack = TextFoolerJin2019.build(model_wrapper)
 
 # Function to run attack
 def run_attack(input_text):
+  def run_attack(input_text):
     try:
         # Attempt to modify the input text
         result = attack.attack(input_text, ground_truth_output=1)
@@ -26,11 +27,11 @@ def run_attack(input_text):
         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"
+        # Check the status of the result
+        if result.goal_status == textattack.shared.AttackResultStatus.SUCCEEDED:
+            attack_info += f"Attack Succeeded!\nOriginal Text: {input_text}\nModified Text: {result.attacked_text.text}\n"
+        elif result.goal_status == textattack.shared.AttackResultStatus.SKIPPED:
+            skipped_info += f"Skipped Text: {result.attacked_text.text}\n"
 
         # Format the output
         output = f"TextAttack Results:\n{attack_info}\nSkipped Details:\n{skipped_info}"
@@ -38,7 +39,7 @@ def run_attack(input_text):
     
     except Exception as e:
         return f"An error occurred: {str(e)}"
-    
+
 # Gradio UI
 gr.Interface(fn=run_attack, 
              inputs=gr.Textbox(lines=4, placeholder="Enter sentence to attack..."),