Update main.py

main.py

@@ -369,6 +369,91 @@ def get_sozo_project(project_id):
         logger.error(f"Error fetching project {project_id}: {str(e)}")
         return jsonify({'error': str(e)}), 500
 
+@app.route('/api/sozo/projects/<string:project_id>', methods=['DELETE'])
+def delete_sozo_project(project_id):
+    logger.info(f"DELETE /api/sozo/projects/{project_id} - Deleting project")
+    
+    try:
+        token = request.headers.get('Authorization', '').split(' ')[1]
+        uid = verify_token(token)
+        if not uid:
+            logger.warning(f"Unauthorized access attempt to delete project {project_id}")
+            return jsonify({'error': 'Unauthorized'}), 401
+        
+        logger.info(f"User {uid} attempting to delete project {project_id}")
+        
+        # Get project data first to verify ownership and get file paths
+        project_ref = db.reference(f'sozo_projects/{project_id}')
+        project_data = project_ref.get()
+        
+        if not project_data:
+            logger.warning(f"Project {project_id} not found for deletion")
+            return jsonify({'error': 'Project not found'}), 404
+        
+        if project_data.get('uid') != uid:
+            logger.warning(f"User {uid} unauthorized to delete project {project_id}")
+            return jsonify({'error': 'Unauthorized to delete this project'}), 403
+        
+        logger.info(f"Project {project_id} verified for deletion by user {uid}")
+        
+        # Delete files from storage
+        deleted_files = []
+        try:
+            # Delete original data file
+            if 'originalDataUrl' in project_data:
+                blob_path = "/".join(project_data['originalDataUrl'].split('/')[4:])
+                blob = bucket.blob(blob_path)
+                if blob.exists():
+                    blob.delete()
+                    deleted_files.append(blob_path)
+                    logger.info(f"Deleted original data file: {blob_path}")
+            
+            # Delete chart files if they exist
+            chart_urls = project_data.get('chartUrls', {})
+            if isinstance(chart_urls, dict):
+                for chart_name, chart_url in chart_urls.items():
+                    try:
+                        chart_blob_path = "/".join(chart_url.split('/')[4:])
+                        chart_blob = bucket.blob(chart_blob_path)
+                        if chart_blob.exists():
+                            chart_blob.delete()
+                            deleted_files.append(chart_blob_path)
+                            logger.info(f"Deleted chart file: {chart_blob_path}")
+                    except Exception as chart_error:
+                        logger.warning(f"Error deleting chart {chart_name}: {str(chart_error)}")
+            
+            # Delete entire project folder if it exists
+            project_folder = f"sozo_projects/{uid}/{project_id}/"
+            blobs = bucket.list_blobs(prefix=project_folder)
+            for blob in blobs:
+                if blob.name not in deleted_files:  # Avoid duplicate deletions
+                    blob.delete()
+                    deleted_files.append(blob.name)
+                    logger.info(f"Deleted additional file: {blob.name}")
+            
+            logger.info(f"Deleted {len(deleted_files)} files from storage for project {project_id}")
+            
+        except Exception as storage_error:
+            logger.error(f"Error deleting storage files for project {project_id}: {str(storage_error)}")
+            # Continue with database deletion even if storage cleanup fails
+        
+        # Delete project from database
+        project_ref.delete()
+        logger.info(f"Project {project_id} deleted from database")
+        
+        logger.info(f"Project {project_id} successfully deleted by user {uid}")
+        
+        return jsonify({
+            'success': True,
+            'message': f'Project {project_id} deleted successfully',
+            'deleted_files': len(deleted_files)
+        }), 200
+        
+    except Exception as e:
+        logger.error(f"Error deleting project {project_id}: {str(e)}")
+        logger.error(f"Traceback: {traceback.format_exc()}")
+        return jsonify({'error': str(e)}), 500
+
 @app.route('/api/sozo/projects/<string:project_id>/markdown', methods=['PUT'])
 def update_sozo_markdown(project_id):
     try: