Update app.py

app.py

@@ -98,10 +98,13 @@ def generate_pdf_report(participant_name, responses, mbti_type_classic, mbti_typ
     pdf.cell(200, 10, txt=f"Your MBTI Type (Weighted): {mbti_type_classic}", ln=True)
     pdf.cell(200, 10, txt=f"Your MBTI Type (AI Prediction): {mbti_type_llm}", ln=True)
 
-    # Save PDF
-    file_path = f"{participant_name}_MBTI_report.pdf"  # Save in the current directory
-    pdf.output(file_path)
-    return file_path
+    # Save PDF to a memory buffer
+    from io import BytesIO
+    pdf_output = BytesIO()
+    pdf.output(pdf_output)
+    pdf_output.seek(0)  # Reset pointer to the start of the buffer
+
+    return pdf_output
 
 # Streamlit component to display the quiz and handle responses
 def show_mbti_quiz():
@@ -141,7 +144,7 @@ def show_mbti_quiz():
                     """
                     try:
                         response = openai.ChatCompletion.create(
-                            model="gpt-4o",
+                            model="gpt-4",
                             messages=[{"role": "system", "content": "You are a helpful assistant."},
                                       {"role": "user", "content": prompt}]
                         )
@@ -151,8 +154,15 @@ def show_mbti_quiz():
                         st.error(f"Error occurred: {e}")
                 
                 # Generate PDF Report
-                file_path = generate_pdf_report(participant_name, responses, mbti_type_classic, mbti_type_llm)
-                st.download_button("Download your MBTI report", file_path)
+                pdf_output = generate_pdf_report(participant_name, responses, mbti_type_classic, mbti_type_llm)
+                
+                # Create a download button for the PDF
+                st.download_button(
+                    label="Download your MBTI report",
+                    data=pdf_output,
+                    file_name=f"{participant_name}_MBTI_report.pdf",
+                    mime="application/pdf"
+                )
 
         else:
             st.warning("Please answer all the questions!")