Update main.py

main.py

@@ -560,6 +560,59 @@ def update_narration_audio(project_id):
     except Exception as e:
         logger.error(f"CRITICAL ERROR updating narration for {project_id}: {traceback.format_exc()}")
         return jsonify({'error': str(e)}), 500
+
+# In main.py, add this new endpoint
+
+@app.route('/api/sozo/projects/<string:project_id>/generate-slides', methods=['POST'])
+def generate_sozo_slides(project_id):
+    logger.info(f"POST /api/sozo/projects/{project_id}/generate-slides - Generating slides")
+    try:
+        token = request.headers.get('Authorization', '').split(' ')
+        uid = verify_token(token)
+        if not uid:
+            return jsonify({'error': 'Unauthorized'}), 401
+        
+        project_ref = db.reference(f'sozo_projects/{project_id}')
+        project_data = project_ref.get()
+        
+        if not project_data or project_data.get('uid') != uid:
+            return jsonify({'error': 'Project not found or unauthorized'}), 404
+        
+        raw_md = project_data.get('rawMarkdown')
+        chart_urls = project_data.get('chartUrls', {})
+        
+        if not raw_md:
+            return jsonify({'error': 'Report must be generated before slides can be created.'}), 400
+            
+        # The planner AI needs an LLM instance
+        llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", google_api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.2)
+
+        slides_data = generate_slides_from_report(
+            raw_md,
+            chart_urls,
+            uid,
+            project_id,
+            bucket,
+            llm
+        )
+        
+        if not slides_data:
+            raise Exception("Slide generation failed in core function.")
+            
+        # Save the slides data to the project in Firebase
+        project_ref.update({'slides': slides_data})
+        
+        logger.info(f"Project {project_id} successfully updated with {len(slides_data)} slides.")
+        
+        return jsonify({
+            'success': True,
+            'slides': slides_data
+        }), 200
+        
+    except Exception as e:
+        logger.error(f"CRITICAL error generating slides for project {project_id}: {traceback.format_exc()}")
+        db.reference(f'sozo_projects/{project_id}').update({'status': 'failed_slides', 'error': str(e)})
+        return jsonify({'error': str(e)}), 500
 # -----------------------------------------------------------------------------
 # 5. UNIVERSAL ENDPOINTS (Waitlist, Feedback, Credits)
 # -----------------------------------------------------------------------------