Update main.py

main.py

@@ -358,13 +358,12 @@ def delete_sozo_project(project_id):
 
 @app.route('/api/sozo/projects/<string:project_id>/generate-report', methods=['POST'])
 def generate_sozo_report(project_id):
-    logger.info(f"POST /api/sozo/projects/{project_id}/generate-report - Generating report")
+    logger.info(f"POST /api/sozo/projects/{project_id}/generate-report - Received request")
     
     try:
         token = request.headers.get('Authorization', '').split(' ')[1]
         uid = verify_token(token)
         if not uid:
-            logger.warning(f"Unauthorized access attempt to generate report for project {project_id}")
             return jsonify({'error': 'Unauthorized'}), 401
         
         project_ref = db.reference(f'sozo_projects/{project_id}')
@@ -373,11 +372,20 @@ def generate_sozo_report(project_id):
         if not project_data or project_data.get('uid') != uid:
             return jsonify({'error': 'Project not found or unauthorized'}), 404
         
+        # --- THE AUTHORITATIVE LOCK ---
+        current_status = project_data.get('status')
+        if current_status in ['generating_report', 'generating_video']:
+            logger.warning(f"User {uid} attempted to generate a report for project {project_id} which is already in progress (status: {current_status}).")
+            return jsonify({'error': 'Report generation is already in progress for this project.'}), 409 # 409 Conflict
+
+        # Set the lock BEFORE starting the heavy work
+        project_ref.update({'status': 'generating_report'})
+        logger.info(f"Project {project_id} status locked to 'generating_report'.")
+        
         blob_path = f"sozo_projects/{uid}/{project_id}/data{Path(project_data['originalFilename']).suffix}"
         blob = bucket.blob(blob_path)
         file_bytes = blob.download_as_bytes()
         
-        # This function now returns data_context, which we need to capture
         draft_data = generate_report_draft(
             io.BytesIO(file_bytes),
             project_data['originalFilename'],
@@ -387,12 +395,11 @@ def generate_sozo_report(project_id):
             bucket
         )
         
-        # **CORRECTION HERE: Save the data_context to Firebase**
         update_data = {
-            'status': 'draft',
+            'status': 'draft', # Unlock by setting the new status
             'rawMarkdown': draft_data.get('raw_md'),
             'chartUrls': draft_data.get('chartUrls'),
-            'dataContext': draft_data.get('data_context') # Save the context for the video generator
+            'dataContext': draft_data.get('data_context')
         }
         
         project_ref.update(update_data)
@@ -406,7 +413,11 @@ def generate_sozo_report(project_id):
         
     except Exception as e:
         logger.error(f"CRITICAL error generating report for project {project_id}: {traceback.format_exc()}")
-        db.reference(f'sozo_projects/{project_id}').update({'status': 'failed', 'error': str(e)})
+        # Make sure to unlock the project on failure
+        db.reference(f'sozo_projects/{project_id}').update({
+            'status': 'failed',
+            'error': str(e)
+        })
         return jsonify({'error': str(e)}), 500