Update main.py

main.py

@@ -5,7 +5,9 @@ from data_processor import DataProcessor  # Import the data analysis class
 from visualization import Visualization  # Import the data viz class
 from ai_analysis import AIAnalysis  # Import the ai analysis class
 from sidebar import Sidebar  # Import the Sidebar class
+from report import ReportGenerator
 
+# ... (rest of your imports and app setup)
 
 def main():
     # Initialize the app configuration
@@ -153,15 +155,17 @@ def main():
             # Download AI output
             ai_analysis.download_llm_output(recommendations, "llm_output.txt")
 
-            # Add the combined download button
-            st.divider()
-            st.subheader("Download Combined Report")
-            combined_pdf = create_combined_pdf(intervention_fig, student_metrics_fig, recommendations)
+            # Generate the PDF Report
+            report_gen = ReportGenerator()
+            combined_pdf = report_gen.create_combined_pdf(intervention_fig, student_metrics_fig, recommendations)
+        
+            # Add the download button
             st.download_button(
                 label="Download Combined Report (PDF)",
                 data=combined_pdf,
                 file_name="combined_report.pdf",
-                mime="application/pdf"
+                mime="application/pdf",
+                icon="📄"
             )
 
         except Exception as e: