prthm11 commited on
Commit
65e5476
·
verified ·
1 Parent(s): 6c22188

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -2
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