Update app.py

app.py

@@ -125,19 +125,18 @@ def extract_full_paper_with_labels(pdf_path, progress=None):
 def process_pdf_file(pdf_file, api_key, repo_address):
     if pdf_file is None:
         return None, "No PDF file uploaded."
-    # Extract content from PDF.
-    # pdf_file can be a file-like object or a dict depending on how Gradio returns it.
+    
+    # Determine file path (Gradio returns a file object or dict)
     file_path = pdf_file.name if hasattr(pdf_file, "name") else pdf_file['name']
     result = extract_full_paper_with_labels(file_path)
     
-    # Convert the result dictionary into a DataFrame and write it to a parquet file.
+    # Convert the result dictionary to a DataFrame and write it as a parquet file.
     df = pd.DataFrame([result])
     base = os.path.splitext(result['filename'])[0]
     parquet_filename = f"{base}.parquet"
     df.to_parquet(parquet_filename, index=False)
     
     repo_status = ""
-    # If API key and repo address are provided, attempt to upload the parquet file.
     if api_key and repo_address:
         api = HfApi()
         try:
@@ -153,10 +152,9 @@ def process_pdf_file(pdf_file, api_key, repo_address):
     else:
         repo_status = "API key or repo address not provided, skipping repo upload."
     
-    # Return the parquet file for local download and the status message.
     return parquet_filename, repo_status
 
-# Function to clear only file-related inputs/outputs, preserving the API key and repo address.
+# Clear only file-related inputs/outputs, preserving API key and repo address.
 def clear_files():
     return None, None, ""
 
@@ -178,7 +176,6 @@ with gr.Blocks() as demo:
         inputs=[pdf_file_input, api_key_input, repo_address_input],
         outputs=[download_file_output, repo_status_output]
     )
-    # The clear button now only clears file-related components; API key and Repo Address remain unchanged.
     clear_button.click(
         clear_files,
         inputs=None,
@@ -186,3 +183,4 @@ with gr.Blocks() as demo:
     )
 
 demo.launch()
+