Add display images

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
 import re
+import os
 import subprocess
 import argparse
 
+
 def remove_color_codes(text):
     color_pattern = re.compile(r'\x1b\[\d+m')
     clean_text = re.sub(color_pattern, '', text)
@@ -30,12 +32,26 @@ def to_html(document):
     formatted_document = '<br>'.join(formatted_lines)
     return formatted_document
 
+def add_images(file_name):
+    html = '<div align="center">'
+    for file in os.listdir("."):
+        if file.startswith(f"errors-{file_name}") and file.endswith(".png"):
+            html += f'<img src="file/{file}">'
+    html += '</div>'
+    return html
+
+def clear_cached_images(file_name):
+    for file in os.listdir("."):
+        if file.startswith(f"errors-{file_name}") and file.endswith(".png"):
+            os.remove(file)
 
 def upload_file(file, paper_type):
-    file_name = file.name.replace(" ", "\ ")
-    command = f"python3 aclpubcheck-main/aclpubcheck/formatchecker.py --paper_type {paper_type} {file_name}"
+    file_name_cmd = file.name.replace(" ", "\ ")
+    file_name = os.path.basename(file.name).split(".")[0]
+    clear_cached_images(file_name)
+    command = f"python3 aclpubcheck-main/aclpubcheck/formatchecker.py --paper_type {paper_type} {file_name_cmd}"
     out = subprocess.run(command, shell=True, stdout=subprocess.PIPE, text=True, stderr=subprocess.STDOUT)
-    return to_html(remove_color_codes(out.stdout))
+    return to_html(remove_color_codes(out.stdout)) + add_images(file_name)
 
 
 with gr.Blocks() as demo:
@@ -48,8 +64,8 @@ with gr.Blocks() as demo:
     gr.HTML(header)
     gr.Markdown("Drop or upload your paper here to have it checked for [*ACL conferences](https://www.aclweb.org/) guidelines.")
     dropdown = gr.Dropdown(
-            ["long", "short", "demo", "other"], label="Paper type", value="long"
-        )
+        ["long", "short", "demo", "other"], label="Paper type", value="long"
+    )
     file_output = gr.File(file_types=[".pdf"])
     button = gr.Button("Check your PDF!", variant="primary")
     out = gr.HTML()
@@ -57,9 +73,8 @@ with gr.Blocks() as demo:
         "This graphical interface is using the official [aclpubcheck tool](https://github.com/acl-org/aclpubcheck). Check the [Github repo for more information.](https://github.com/teelinsan/aclpubcheck-gui)")
     gr.Markdown(
         "No data is collected. If you prefer you can also duplicate this Space to run it privately. [![Duplicate Space](https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14)](https://huggingface.co/spaces/teelinsan/aclpubcheck?duplicate=true)")
-    
-    button.click(upload_file, [file_output, dropdown], out)
 
+    button.click(upload_file, [file_output, dropdown], out)
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()