Update app.py

app.py

@@ -9,6 +9,7 @@ import random
 from silverspeak import (
     random_attack as random_homoglyphs_attack,
 )
+from silverspeak.homoglyphs.utils import TypesOfHomoglyphs
 
 
 def random_deletion_attack(original_text, percentage=0.1, random_seed=42):
@@ -44,17 +45,18 @@ inputs = [
         value=0.05,
         label="Percentage of characters to replace",
     ),
-    gr.Checkbox(label="Zero-Width", value=False),
+    gr.Dropdown(value=['identical'], multiselect=True, choices=[x.value for x in TypesOfHomoglyphs])
+    gr.Checkbox(label="Zero-Width", value=False, show=False),
 ]
 outputs = gr.Textbox(label="Output text", show_copy_button=True)
 title = "Silverspeak"
-description = """Here, we utilize homoglyph and zero-width character based attacks to evade AI-generated content detection systems.
+description = """Here, we utilize homoglyph attacks to evade AI-generated content detection systems.
 
 For more information, please refer to the research paper: [Creo, Aldan and Shushanta Pudasaini. “Evading AI-Generated Content Detectors using Homoglyphs.” (2024)](https://arxiv.org/abs/2109.06255)."""
 
-def attack_fn(text, replace_chars, percentage, replace_spaces):
+def attack_fn(text, replace_chars, percentage, types_of_homoglyphs, replace_spaces):
     if replace_chars:
-        text = random_homoglyphs_attack(text=text, percentage=percentage, types_of_homoglyphs_to_use=["identical"])
+        text = random_homoglyphs_attack(text=text, percentage=percentage, types_of_homoglyphs_to_use=types_of_homoglyphs)
     if replace_spaces:
         text = random_deletion_attack(original_text=text, percentage=percentage)
     return text