Update main.py

main.py

@@ -7,12 +7,14 @@ 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
     app_config = AppConfig()
 
+    # Initialize the session state 
+    if 'ai_recommendations' not in st.session_state:
+        st.session_state.ai_recommendations = None
+
     # Initialize the sidebar
     sidebar = Sidebar()
     sidebar.display()
@@ -141,36 +143,69 @@ def main():
                 with st.expander(f"{student_name} Decision Tree", expanded=False):
                     st.graphviz_chart(tree_diagram.source)
 
-            # Prepare input for the language model
-            llm_input = ai_analysis.prepare_llm_input(student_metrics_df)
+            # # Prepare input for the language model
+            # llm_input = ai_analysis.prepare_llm_input(student_metrics_df)
 
-            # Generate Notes and Recommendations using Hugging Face LLM
-            with st.spinner("Generating AI analysis..."):
-                # recommendations = ai_analysis.prompt_response_from_hf_llm(llm_input)
-                recommendations = ai_analysis.prompt_response_from_mistral_llm(llm_input)
+            # # Generate Notes and Recommendations using LLM
+            # with st.spinner("Generating MTSS.ai analysis..."):
+            #     # recommendations = ai_analysis.prompt_response_from_hf_llm(llm_input)
+            #     recommendations = ai_analysis.prompt_response_from_mistral_llm(llm_input)
 
-            st.subheader("AI Analysis")
-            st.markdown(recommendations)
+            # Generate Notes and Recommendations using LLM
+            if st.session_state.ai_recommendations is None:
+                with st.spinner("Generating MTSS.ai analysis..."):
+                    llm_input = ai_analysis.prepare_llm_input(student_metrics_df)
+                    recommendations = ai_analysis.prompt_response_from_mistral_llm(llm_input)
+                    st.session_state.ai_recommendations = recommendations
+
+            # Display the recommendations
+            st.subheader("MTSS.ai Analysis")
+            # st.markdown(recommendations)
+            st.markdown(st.session_state.ai_recommendations)
 
             # Download AI output
-            ai_analysis.download_llm_output(recommendations, "llm_output.txt")
+            # ai_analysis.download_llm_output(recommendations, "MTSSai_Report.txt")
+            ai_analysis.download_llm_output(st.session_state.ai_recommendations, "MTSSai_Report.txt")
 
-            # Generate the PDF Report
-            report_gen = ReportGenerator()
-            combined_pdf = report_gen.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",
-                icon="📄", 
-                use_container_width=True
-            )
+            # # Add the download button
+            # st.download_button(
+            #     label="Download Combined Report (PDF)",
+            #     data=combined_pdf,
+            #     file_name="combined_report.pdf",
+            #     mime="application/pdf",
+            #     icon="📄", 
+            #     use_container_width=True
+            # )
+
+            # Generate the PDF Report using the stored recommendations
+            if st.button("Generate PDF Report"):
+                report_gen = ReportGenerator()
+                combined_pdf = report_gen.create_combined_pdf(
+                    intervention_fig, 
+                    student_metrics_fig, 
+                    st.session_state.ai_recommendations
+                )
+                
+                # Add the download button for the PDF
+                st.download_button(
+                    label="Download Combined Report (PDF)",
+                    data=combined_pdf,
+                    file_name="combined_report.pdf",
+                    mime="application/pdf",
+                    icon="📄", 
+                    use_container_width=True
+                )
 
         except Exception as e:
             st.error(f"Error processing the file: {str(e)}")
 
 if __name__ == '__main__':
-    main()
+    main()
+
+
+
+