Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, request, jsonify, render_template, send_from_directory
+from flask import Flask, request, jsonify, render_template, send_from_directory, send_file
 import cv2, json,base64,io,os,tempfile,logging, re
 import numpy as np
 from unstructured.partition.pdf import partition_pdf
@@ -2180,29 +2180,15 @@ def index():
 
 @app.route("/download_sb3/<project_id>", methods=["GET"])
 def download_sb3(project_id):
-    """
-    Allows users to download the generated .sb3 Scratch project file.
-    """
-    sb3_filename = f"{project_id}.sb3"
-    #sb3_filepath = os.path.join("generated_projects", sb3_filename)
-    sb3_filepath = GEN_PROJECT_DIR / sb3_filename
-    
-    try:
-        if os.path.exists(sb3_filepath):
-            logger.info(f"Serving SB3 file for project ID: {project_id}")
-            # send_from_directory serves the file and handles content-disposition for download
-            return send_from_directory(
-                directory="generated_projects",
-                path=sb3_filename,
-                as_attachment=True,  # This makes the browser download the file
-                download_name=sb3_filename # This sets the filename for the download
-            )
-        else:
-            logger.warning(f"SB3 file not found for ID: {project_id}")
-            return jsonify({"error": "Scratch project file not found"}), 404
-    except Exception as e:
-        logger.error(f"Error serving SB3 file for ID {project_id}: {e}")
-        return jsonify({"error": "Failed to retrieve Scratch project file"}), 500
+    sb3_path = GEN_PROJECT_DIR / f"{project_id}.sb3"
+    if not sb3_path.exists():
+        return jsonify({"error": "Scratch project file not found"}), 404
+
+    return send_file(
+        sb3_path,
+        as_attachment=True,
+        download_name=sb3_path.name
+    )
 
 # API endpoint
 @app.route('/process_pdf', methods=['POST'])