Update main.py

main.py

@@ -826,6 +826,85 @@ def extract_firebase_path(public_url: str) -> str:
         return ""
 
 
+# Delete endpoints
+#projects
+@app.route('/api/projects/<string:story_id>', methods=['DELETE'])
+def delete_project(story_id):
+    """
+    Deletes the entire project (story) from the database (and optionally its assets in storage).
+    """
+    try:
+        # --- Authentication ---
+        auth_header = request.headers.get('Authorization', '')
+        if not auth_header or 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
+
+        # --- Fetch Story ---
+        story_ref = db.reference(f"stories/{story_id}")
+        story_data = story_ref.get()
+        if not story_data:
+            return jsonify({'error': 'Story not found'}), 404
+
+        if story_data.get('uid') != uid:
+            return jsonify({'error': 'Unauthorized'}), 403
+
+        # --- Delete the story from DB ---
+        story_ref.delete()
+
+        return jsonify({'success': True, 'message': f'Story {story_id} deleted.'}), 200
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+#videos
+@app.route('/api/videos/<string:story_id>', methods=['DELETE'])
+def delete_video(story_id):
+    """
+    Deletes only the video from a project (removes video_url from DB),
+    optionally deleting the file in storage.
+    """
+    try:
+        # --- Authentication ---
+        auth_header = request.headers.get('Authorization', '')
+        if not auth_header or 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
+
+        # --- Fetch Story ---
+        story_ref = db.reference(f"stories/{story_id}")
+        story_data = story_ref.get()
+        if not story_data:
+            return jsonify({'error': 'Story not found'}), 404
+
+        if story_data.get('uid') != uid:
+            return jsonify({'error': 'Unauthorized'}), 403
+
+        video_url = story_data.get('video_url')
+        if not video_url:
+            return jsonify({'error': 'No video to delete'}), 400
+
+        # Delete the video file from Firebase Storage
+        bucket = storage.bucket()
+        file_path = extract_firebase_path(video_url)
+        if file_path:
+            blob = bucket.blob(file_path)
+            blob.delete()
+
+        # --- Remove video_url from DB ---
+        story_ref.update({'video_url': None})
+
+        return jsonify({'success': True, 'message': f'Video removed from story {story_id}.'}), 200
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+        
 # ---------- Credit Request Endpoints ----------
 
 @app.route('/api/user/request-credits', methods=['POST'])