Update gradio_demo.py

gradio_demo.py

@@ -91,6 +91,7 @@ def generate_pdf(
     target_text: str = "",
     font_choice: str = 'auto',
     uploaded_font=None,
+    wrap_on_words: bool = True,
 ) -> Tuple[str, str]:
     """Generate selected PDF and return (pdf_path, extracted_text)
 
@@ -108,6 +109,8 @@ def generate_pdf(
     # resolve font path and create an attacker instance for this request
     font_path = _resolve_font_path(choice=font_choice, uploaded_file=uploaded_font)
     attacker = PDFAttacker(font_path=font_path)
+    # apply wrap mode
+    attacker.wrap_on_words = wrap_on_words
 
     try:
         if mode == 'normal':
@@ -144,20 +147,21 @@ def build_demo():
                 generate = gr.Button('Generate PDF')
                 
                 # Font selection: presets + custom upload
-                font_choice = gr.Dropdown(choices=['auto', 'DejaVu Serif', 'Liberation Serif', 'FreeSerif', 'Custom'], value='auto', label='Font')
+                font_choice = gr.Dropdown(choices=['auto', 'DejaVu Serif', 'Liberation Serif', 'FreeSerif', 'Arial', 'Helvetica', 'Times New Roman', 'Roboto', 'Courier', 'Custom'], value='auto', label='Font')
                 upload_font = gr.File(label='Upload TTF/OTF (optional)', file_count='single')
+                wrap_on_words = gr.Checkbox(label='Wrap on words', value=True)
 
         download_file = gr.File(label='Download generated PDF')
         extracted_preview = gr.Textbox(lines=8, label='Extracted text preview')
 
-        def _on_generate(text, mode, attack_factor, target_text, font_choice, upload_font):
-            path, extracted = generate_pdf(text=text, mode=mode, attack_factor=attack_factor, target_text=target_text, font_choice=font_choice, uploaded_font=upload_font)
+        def _on_generate(text, mode, attack_factor, target_text, font_choice, upload_font, wrap_on_words):
+            path, extracted = generate_pdf(text=text, mode=mode, attack_factor=attack_factor, target_text=target_text, font_choice=font_choice, uploaded_font=upload_font, wrap_on_words=wrap_on_words)
             if not path:
                 # Return empty file and error message in preview
                 return None, extracted
             return path, extracted
 
-        generate.click(fn=_on_generate, inputs=[txt, mode, attack_factor, target_text, font_choice, upload_font], outputs=[download_file, extracted_preview])
+        generate.click(fn=_on_generate, inputs=[txt, mode, attack_factor, target_text, font_choice, upload_font, wrap_on_words], outputs=[download_file, extracted_preview])
 
     return demo