Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,10 +12,21 @@ def download_pdf(url):
|
|
| 12 |
file.write(response.content)
|
| 13 |
return file_path
|
| 14 |
|
|
|
|
|
|
|
| 15 |
def get_doi(pdf_path):
|
| 16 |
pdf2doi.config.set('verbose', False)
|
| 17 |
results = pdf2doi.pdf2doi(pdf_path)
|
| 18 |
-
validation_info = results
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
doi = validation_info.get('DOI', None)
|
| 20 |
title = validation_info.get('title', None)
|
| 21 |
url = validation_info.get('URL', None)
|
|
|
|
| 12 |
file.write(response.content)
|
| 13 |
return file_path
|
| 14 |
|
| 15 |
+
import json
|
| 16 |
+
|
| 17 |
def get_doi(pdf_path):
|
| 18 |
pdf2doi.config.set('verbose', False)
|
| 19 |
results = pdf2doi.pdf2doi(pdf_path)
|
| 20 |
+
validation_info = results.get('validation_info', {})
|
| 21 |
+
if isinstance(validation_info, str):
|
| 22 |
+
try:
|
| 23 |
+
validation_info = json.loads(validation_info) # Convert JSON string to dic
|
| 24 |
+
except json.JSONDecodeError as e:
|
| 25 |
+
print(f"Error decoding JSON: {e}")
|
| 26 |
+
return None, None, None
|
| 27 |
+
if not isinstance(validation_info, dict):
|
| 28 |
+
print("Validation info is not a dictionary")
|
| 29 |
+
return None, None, None
|
| 30 |
doi = validation_info.get('DOI', None)
|
| 31 |
title = validation_info.get('title', None)
|
| 32 |
url = validation_info.get('URL', None)
|