Update gradio_demo.py

gradio_demo.py

@@ -21,16 +21,54 @@ def _resolve_font_path(choice: str, uploaded_file) -> str:
     if choice == 'auto' or not choice:
         return None
 
-    # known presets mapped to common system paths
+    # known presets mapped to candidate system paths (try first existing)
     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',
+        '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',
+        ],
+        'DejaVu Sans': [
+            '/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf',
+        ],
+        'Arial': [
+            '/usr/share/fonts/truetype/msttcorefonts/Arial.ttf',
+            '/usr/share/fonts/truetype/msttcorefonts/arial.ttf',
+            '/usr/share/fonts/truetype/arial/arial.ttf',
+            '/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf',
+        ],
+        'Helvetica': [
+            '/usr/share/fonts/truetype/urw-base35/Helvetica.ttf',
+            '/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf',
+            '/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf',
+        ],
+        'Times New Roman': [
+            '/usr/share/fonts/truetype/msttcorefonts/Times_New_Roman.ttf',
+            '/usr/share/fonts/truetype/msttcorefonts/Times_New_Roman.ttf',
+            '/usr/share/fonts/truetype/liberation/LiberationSerif-Regular.ttf',
+        ],
+        'Roboto': [
+            '/usr/share/fonts/truetype/roboto/Roboto-Regular.ttf',
+            '/usr/share/fonts/truetype/roboto/Roboto-Regular.ttf',
+        ],
+        'Courier': [
+            '/usr/share/fonts/truetype/liberation/LiberationMono-Regular.ttf',
+            '/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf',
+        ],
+        'Times': [
+            '/usr/share/fonts/truetype/liberation/LiberationSerif-Regular.ttf',
+        ],
     }
 
     if choice in presets:
-        p = presets[choice]
-        return p if os.path.exists(p) else None
+        for p in presets[choice]:
+            if os.path.exists(p):
+                return p
+        return None
 
     # custom uploaded file: gradio returns a local path-like string or dict
     if choice == 'Custom' and uploaded_file:
@@ -92,7 +130,7 @@ def generate_pdf(
     font_choice: str = 'auto',
     uploaded_font=None,
     wrap_on_words: bool = True,
-) -> Tuple[str, str]:
+) -> Tuple[str, str, str]:
     """Generate selected PDF and return (pdf_path, extracted_text)
 
     Inputs: text, mode: 'normal'|'attacked'|'targeted', attack_factor, target_text
@@ -112,6 +150,10 @@ def generate_pdf(
     # apply wrap mode
     attacker.wrap_on_words = wrap_on_words
 
+    # Build a contextual status string for the UI
+    resolved_font = font_path or "(auto/default)"
+    status_lines = [f"Font resolved to: {resolved_font}", f"Wrap on words: {wrap_on_words}"]
+
     try:
         if mode == 'normal':
             attacker.create_normal_pdf(text=clean_text, output_path=output_path)
@@ -125,12 +167,12 @@ def generate_pdf(
 
     except Exception as e:
         # Surface errors to the UI
-        return "", f"Error generating PDF: {e}"
+        return "", f"Error extracting text: {e}", f"Error: {e}"
 
     # Extract text to show how the copied/extracted text looks
     extracted = _extract_text_from_pdf(output_path)
 
-    return output_path, extracted
+    return output_path, extracted, "\n".join(status_lines)
 
 
 def build_demo():
@@ -153,15 +195,16 @@ def build_demo():
 
         download_file = gr.File(label='Download generated PDF')
         extracted_preview = gr.Textbox(lines=8, label='Extracted text preview')
+        status_box = gr.Textbox(lines=4, label='Status')
 
         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)
+            path, extracted, status = 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
+                return None, extracted, status
+            return path, extracted, status
 
-        generate.click(fn=_on_generate, inputs=[txt, mode, attack_factor, target_text, font_choice, upload_font, wrap_on_words], 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, status_box])
 
     return demo