Update gradio_demo.py

gradio_demo.py

@@ -13,7 +13,33 @@ import gradio as gr
 from pdf_attacker import PDFAttacker
 
 
-attacker = PDFAttacker()
+def _resolve_font_path(choice: str, uploaded_file) -> str:
+    """Return a font path given a dropdown choice or uploaded file.
+
+    If choice is 'auto' return None so PDFAttacker will pick a reasonable default.
+    """
+    if choice == 'auto' or not choice:
+        return None
+
+    # known presets mapped to common system paths
+    presets = {
+        'DejaVu Serif': '/usr/share/fonts/truetype/dejavu/DejaVuSerif.ttf',
+        'Liberation Serif': '/usr/share/fonts/truetype/liberation/LiberationSerif-Regular.ttf',
+        'FreeSerif': '/usr/share/fonts/truetype/freefont/FreeSerif.ttf',
+    }
+
+    if choice in presets:
+        p = presets[choice]
+        return p if os.path.exists(p) else None
+
+    # custom uploaded file: gradio returns a local path-like string or dict
+    if choice == 'Custom' and uploaded_file:
+        # uploaded_file may be a dict-like object or a str path
+        if isinstance(uploaded_file, dict) and 'name' in uploaded_file:
+            return uploaded_file['name']
+        return uploaded_file
+
+    return None
 
 # Theme customization per request
 theme = gr.themes.Soft(
@@ -63,6 +89,8 @@ def generate_pdf(
     mode: str,
     attack_factor: float = 0.7,
     target_text: str = "",
+    font_choice: str = 'auto',
+    uploaded_font=None,
 ) -> Tuple[str, str]:
     """Generate selected PDF and return (pdf_path, extracted_text)
 
@@ -77,6 +105,10 @@ def generate_pdf(
     # Clean input text
     clean_text = " ".join(text.split())
 
+    # 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)
+
     try:
         if mode == 'normal':
             attacker.create_normal_pdf(text=clean_text, output_path=output_path)
@@ -110,18 +142,22 @@ def build_demo():
                 attack_factor = gr.Slider(minimum=0.0, maximum=1.0, step=0.05, value=0.7, label='Attack factor (attacked mode)')
                 target_text = gr.Textbox(lines=2, label='Target text (targeted mode)')
                 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')
+                upload_font = gr.File(label='Upload TTF/OTF (optional)', file_count='single')
 
         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):
-            path, extracted = generate_pdf(text=text, mode=mode, attack_factor=attack_factor, target_text=target_text)
+        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)
             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], outputs=[download_file, extracted_preview])
+        generate.click(fn=_on_generate, inputs=[txt, mode, attack_factor, target_text, font_choice, upload_font], outputs=[download_file, extracted_preview])
 
     return demo