Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import openai
 from dotenv import load_dotenv
 import os
+from fpdf import FPDF
 
 # Load the OpenAI API Key
 api_key = st.text_input('Enter your OpenAI API Key', type="password")
@@ -70,6 +71,38 @@ def classic_mbti_weighted(responses):
             mbti_type += trait2
     return mbti_type
 
+# Function to generate PDF report
+def generate_pdf_report(participant_name, responses, mbti_type_classic, mbti_type_llm):
+    pdf = FPDF()
+    pdf.set_auto_page_break(auto=True, margin=15)
+    pdf.add_page()
+
+    # Add a title
+    pdf.set_font("Arial", 'B', 16)
+    pdf.cell(200, 10, txt="FlexTemp Personality Test Report", ln=True, align="C")
+
+    # Add participant name
+    pdf.set_font("Arial", size=12)
+    pdf.ln(10)
+    pdf.cell(200, 10, txt=f"Name: {participant_name}", ln=True)
+
+    # Add the responses
+    pdf.ln(10)
+    pdf.cell(200, 10, txt="Your Answers:", ln=True)
+    for i, question in enumerate(questions):
+        response = responses[i]
+        pdf.cell(200, 10, txt=f"Q{i+1}: {question['text']} - {response}", ln=True)
+
+    # Add MBTI results
+    pdf.ln(10)
+    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"/mnt/data/{participant_name}_MBTI_report.pdf"
+    pdf.output(file_path)
+    return file_path
+
 # Streamlit component to display the quiz and handle responses
 def show_mbti_quiz():
     st.title('FlexTemp Personality Test')
@@ -97,6 +130,7 @@ def show_mbti_quiz():
                 st.write(f"Your MBTI type based on weighted answers: {mbti_type_classic}")
 
                 # You can add LLM-based prediction if needed here (example OpenAI-based model)
+                mbti_type_llm = ""
                 if api_key:
                     # Run the LLM (GPT-4, for example) model to generate a personality type.
                     prompt = f"""
@@ -107,7 +141,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}]
                         )
@@ -115,6 +149,11 @@ def show_mbti_quiz():
                         st.write(f"Your MBTI type according to AI: {mbti_type_llm}")
                     except Exception as e:
                         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)
+
         else:
             st.warning("Please answer all the questions!")
 
@@ -126,4 +165,4 @@ def main():
         st.info("Please enter your OpenAI API Key to begin the quiz.")
 
 if __name__ == "__main__":
-    main()
+    main()