Update app_main.py

app_main.py

@@ -455,8 +455,6 @@ def index():
     return render_template('app_index.html')
 
 # API endpoint
-
-
 @app.route('/process_pdf', methods=['POST'])
 def process_pdf():
     try:
@@ -495,6 +493,31 @@ def process_pdf():
         logger.exception("❌ Failed to process PDF")
         return jsonify({"error": f"❌ Failed to process PDF: {str(e)}"}), 500
 
+# --- New endpoint to download the .sb3 file ---
+@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)
+
+    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
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)