Mini-patch #2

app.py

@@ -112,8 +112,8 @@ async def extract_text_pdf(document: DocumentID):
         else:
             for title in titles:
                 if title[0] == 1 or title[0] == 2:
-                    main_titles.append(title[1])
-        return {"pub_id": document.doc_id, "titles": [re.sub(r"\s+", " ", remove_punctuations(remove_in_betweens(t))).strip() for t in main_titles], "text": postprocess_text, "error": False} if len(main_titles) > 0 else {"pub_id": document.doc_id, "titles": "No titles found !", "text": postprocess_text, "error": False}
+                    main_titles.append((title[0], title[1]))
+        return {"pub_id": document.doc_id, "titles": [(t[0],re.sub(r"\s+", " ", remove_punctuations(remove_in_betweens(t[1]))).strip()) for t in main_titles], "text": postprocess_text, "error": False} if len(main_titles) > 0 else {"pub_id": document.doc_id, "titles": "No titles found !", "text": postprocess_text, "error": False}
     else:
         print("ID: " + document.doc_id)
         print("URL: " + f"http://arxiv.org/pdf/{document.doc_id}")