Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -46,17 +46,19 @@ def home():
|
|
| 46 |
PROCESSED_PDF_PATH = "static/processed.pdf"
|
| 47 |
|
| 48 |
def getpdfcontent(pdf_path):
|
| 49 |
-
#
|
| 50 |
if pdf_path and ('http' in pdf_path or 'dropbox' in pdf_path):
|
| 51 |
pdf_path = pdf_path.replace('dl=0', 'dl=1')
|
| 52 |
|
| 53 |
-
#
|
| 54 |
response = requests.get(pdf_path)
|
| 55 |
-
|
| 56 |
-
|
|
|
|
| 57 |
raise ValueError("No valid PDF content found.")
|
| 58 |
|
| 59 |
-
|
|
|
|
| 60 |
|
| 61 |
@app.route('/view-pdf', methods=['GET'])
|
| 62 |
def view_pdf():
|
|
|
|
| 46 |
PROCESSED_PDF_PATH = "static/processed.pdf"
|
| 47 |
|
| 48 |
def getpdfcontent(pdf_path):
|
| 49 |
+
# Handle Dropbox URLs
|
| 50 |
if pdf_path and ('http' in pdf_path or 'dropbox' in pdf_path):
|
| 51 |
pdf_path = pdf_path.replace('dl=0', 'dl=1')
|
| 52 |
|
| 53 |
+
# Get the PDF bytes
|
| 54 |
response = requests.get(pdf_path)
|
| 55 |
+
pdf_bytes = response.content
|
| 56 |
+
|
| 57 |
+
if not pdf_bytes or not pdf_bytes.startswith(b"%PDF"):
|
| 58 |
raise ValueError("No valid PDF content found.")
|
| 59 |
|
| 60 |
+
# Convert to BytesIO only after validation
|
| 61 |
+
return BytesIO(pdf_bytes)
|
| 62 |
|
| 63 |
@app.route('/view-pdf', methods=['GET'])
|
| 64 |
def view_pdf():
|