Spaces:
Running
Running
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 |
|