filtering the issues

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import tempfile
 
 # Set JAVA_HOME environment variable
-os.environ['JAVA_HOME'] = '/usr/lib/jvm/java-11-openjdk-amd64'
+# os.environ['JAVA_HOME'] = '/usr/lib/jvm/java-11-openjdk-amd64'
 
 def grade_pdf(input_pdf, pdffigures2_jar):
     """
@@ -52,12 +52,31 @@ def grade_pdf(input_pdf, pdffigures2_jar):
             with open(output_json_file, "r") as f:
                 regions = json.load(f)
 
+            issueList = list()
+
             for region in regions:
+                issues = dict()
                 if(region["captionBoundary"]["y1"]>region["regionBoundary"]["y1"]):
                     region["captionLocation"]="Below"
                 else:
                     region["captionLocation"]="Above"
-            return regions
+
+                if(region["captionLocation"]=="Below" and region["figType"] == "Table"):
+                    issues["page_number"]=region["page"]
+                    issues["caption_coordinate"]=region["captionBoundary"]
+                    issues["fig_type"] = region["figType"]
+                    issues["caption_location"] = region["captionLocation"]
+                    issues["description"]="Location of the caption for tables must be above the table."
+                    issueList.append(issues)
+                    
+                if(region["captionLocation"]=="Above" and region["figType"] == "Figure"):
+                    issues["page_number"]=region["page"]
+                    issues["caption_coordinate"]=region["captionBoundary"]
+                    issues["fig_type"] = region["figType"]
+                    issues["description"]="Location of the caption for figures must be below the table."
+                    issueList.append(issues)
+                    
+            return issueList
 
         else:
             raise Exception(f"JSON output file not found: {output_json_file}")