Update app.py

app.py

@@ -2,7 +2,7 @@ import subprocess
 import json
 import os
 import gradio as gr
-# import tempfile
+import tempfile
 
 # Set JAVA_HOME environment variable
 os.environ['JAVA_HOME'] = '/usr/lib/jvm/java-11-openjdk-amd64'
@@ -78,17 +78,17 @@ def process_file(file):
     # Define paths
     pdffigures2_jar_path = "pdffigures2.jar"  # Path to pdffigures2 JAR file
     
-    # temp_input = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf')
-    # temp_input.write(file)
-    # temp_input_path = temp_input.name
-    # print("Path of the input file: ",temp_input_path)
+    temp_input = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf')
+    temp_input.write(file)
+    temp_input_path = temp_input.name
+    print("Path of the input file: ",temp_input_path)
 
     try:
         # Process the PDF and get JSON data
-        print(file.name)
-        extracted_data = grade_pdf(file.name, pdffigures2_jar_path)
+        # print(file.name)
+        extracted_data = grade_pdf(temp_input_path, pdffigures2_jar_path)
         # Clean up the temporary input file
-        # os.unlink(temp_input_path)
+        os.unlink(temp_input_path)
         
 
         # Return JSON as a formatted string
@@ -100,9 +100,13 @@ def process_file(file):
 def gradio_interface():
     interface = gr.Interface(
         fn=process_file,
-        inputs=gr.File(label="Upload PDF File", file_types=[".pdf"]),
+        
+        inputs=gr.File(label="Upload PDF File", file_types=[".pdf"], type="binary"),
+        
         outputs=gr.Textbox(label="Extracted JSON Data", lines=20),
+        
         title="PDF Grading Interface",
+        
         description="Upload a PDF file, and this tool will extract figures and captions as JSON."
     )
     return interface