Update main.py

main.py

@@ -155,9 +155,10 @@ def main():
             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_hf_llm(llm_input)
                     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)
@@ -182,30 +183,25 @@ def main():
             # )
 
             # 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
-                )
+            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
-                )
+            # Add the download button for the PDF
+            st.download_button(
+                label="Download MTSS.ai 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()