Update main.py

main.py

@@ -610,6 +610,55 @@ def view_videos():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
+@app.route('/api/download/story_asset/<story_id>', methods=['GET'])
+def download_story_asset(story_id):
+    try:
+        # --- Authentication ---
+        auth_header = request.headers.get('Authorization', '')
+        if not auth_header.startswith('Bearer '):
+            return jsonify({'error': 'Missing or invalid token'}), 401
+        token = auth_header.split(' ')[1]
+        uid = verify_token(token)
+        if not uid:
+            return jsonify({'error': 'Invalid or expired token'}), 401
+
+        # --- Check if the Story Exists and Belongs to the User ---
+        stories_ref = db.reference('stories')
+        story_record = stories_ref.child(story_id).get()
+
+        if not story_record:
+            return jsonify({'error': 'Story not found'}), 404
+        if story_record.get('uid') != uid:
+            return jsonify({'error': 'Unauthorized'}), 403
+
+        # --- Determine the Asset Type ---
+        video_url = story_record.get("video_url", None)
+        audio_url = story_record.get("audio_url", None)  # Add if audio files exist
+        image_url = story_record.get("image_url", None)  # Add if images exist
+
+        # Prioritize video > audio > image
+        file_url = video_url or audio_url or image_url
+        if not file_url:
+            return jsonify({'error': 'No asset found for this story'}), 404
+
+        # --- If Stored in Firebase Storage, Generate a Signed URL ---
+        if "firebasestorage.googleapis.com" in file_url:
+            bucket = storage.bucket()
+            blob_path = file_url.split('/o/')[-1].split('?')[0]  # Extract path
+            blob = bucket.blob(blob_path)
+            signed_url = blob.generate_signed_url(expiration=600)  # Valid for 10 minutes
+            return jsonify({"download_url": signed_url})
+
+        # --- If Stored Locally, Serve the File ---
+        local_path = os.path.join("uploads", file_url)  # Adjust path accordingly
+        if os.path.exists(local_path):
+            return send_file(local_path, as_attachment=True)
+
+        return jsonify({'error': 'File not found'}), 404
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
 # ---------- Credit Request Endpoints ----------
 
 @app.route('/api/user/request-credits', methods=['POST'])