Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ from io import BytesIO
|
|
| 22 |
from pathlib import Path
|
| 23 |
import os
|
| 24 |
from utils.block_relation_builder import block_builder, variable_adder_main
|
| 25 |
-
|
| 26 |
# ============================== #
|
| 27 |
# INITIALIZE CLIP EMBEDDER #
|
| 28 |
# ============================== #
|
|
@@ -2259,6 +2259,17 @@ def download_sb3(project_id):
|
|
| 2259 |
download_name=sb3_path.name
|
| 2260 |
)
|
| 2261 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2262 |
# API endpoint
|
| 2263 |
@app.route('/process_pdf', methods=['POST'])
|
| 2264 |
def process_pdf():
|
|
@@ -2289,7 +2300,7 @@ def process_pdf():
|
|
| 2289 |
temp_dir = tempfile.mkdtemp()
|
| 2290 |
saved_pdf_path = os.path.join(temp_dir, filename)
|
| 2291 |
pdf_file.save(saved_pdf_path)
|
| 2292 |
-
|
| 2293 |
# logger.info(f"Created project folder: {project_folder}")
|
| 2294 |
logger.info(f"Saved uploaded PDF to: {saved_pdf_path}")
|
| 2295 |
|
|
|
|
| 22 |
from pathlib import Path
|
| 23 |
import os
|
| 24 |
from utils.block_relation_builder import block_builder, variable_adder_main
|
| 25 |
+
global pdf_doc
|
| 26 |
# ============================== #
|
| 27 |
# INITIALIZE CLIP EMBEDDER #
|
| 28 |
# ============================== #
|
|
|
|
| 2259 |
download_name=sb3_path.name
|
| 2260 |
)
|
| 2261 |
|
| 2262 |
+
@app.route("/download_pdf/<project_id>", methods=["GET"])
|
| 2263 |
+
def download_sb3(project_id):
|
| 2264 |
+
if not pdf_doc.exists():
|
| 2265 |
+
return jsonify({"error": "Scratch project file not found"}), 404
|
| 2266 |
+
|
| 2267 |
+
return send_file(
|
| 2268 |
+
pdf_doc,
|
| 2269 |
+
as_attachment=True,
|
| 2270 |
+
download_name=pdf_doc.name
|
| 2271 |
+
)
|
| 2272 |
+
|
| 2273 |
# API endpoint
|
| 2274 |
@app.route('/process_pdf', methods=['POST'])
|
| 2275 |
def process_pdf():
|
|
|
|
| 2300 |
temp_dir = tempfile.mkdtemp()
|
| 2301 |
saved_pdf_path = os.path.join(temp_dir, filename)
|
| 2302 |
pdf_file.save(saved_pdf_path)
|
| 2303 |
+
pdf_doc = saved_pdf_path
|
| 2304 |
# logger.info(f"Created project folder: {project_folder}")
|
| 2305 |
logger.info(f"Saved uploaded PDF to: {saved_pdf_path}")
|
| 2306 |
|