Omar ID EL MOUMEN commited on
Commit
d588a4d
·
1 Parent(s): 9a324f3

Fix attempt #1

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -52,7 +52,7 @@ class Query(BaseModel):
52
  class DocumentID(BaseModel):
53
  doc_id: str
54
 
55
- class URL(BaseModel):
56
  url: str
57
 
58
  @app.post("/search")
@@ -130,8 +130,8 @@ async def extract_random_arxiv_pdf(query: Query):
130
  return await extract_arxiv_pdf(random.choice(list(pubs["message"].keys())))
131
 
132
  @app.post("/extract_pdf/url")
133
- async def extract_pdf(url: URL):
134
- pdf_req = requests.get(url, verify=False)
135
  if pdf_req.status_code == 200:
136
  pdf_data = BytesIO(pdf_req.content)
137
  doc = fitz.open(stream=pdf_data, filetype="pdf")
 
52
  class DocumentID(BaseModel):
53
  doc_id: str
54
 
55
+ class WebPDF(BaseModel):
56
  url: str
57
 
58
  @app.post("/search")
 
130
  return await extract_arxiv_pdf(random.choice(list(pubs["message"].keys())))
131
 
132
  @app.post("/extract_pdf/url")
133
+ async def extract_pdf(url: WebPDF):
134
+ pdf_req = requests.get(url)
135
  if pdf_req.status_code == 200:
136
  pdf_data = BytesIO(pdf_req.content)
137
  doc = fitz.open(stream=pdf_data, filetype="pdf")