Update app.py

app.py

@@ -2,7 +2,6 @@ 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")
@@ -71,41 +70,6 @@ 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 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():
     st.title('FlexTemp Personality Test')
@@ -133,7 +97,6 @@ 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"""
@@ -144,7 +107,7 @@ def show_mbti_quiz():
                     """
                     try:
                         response = openai.ChatCompletion.create(
-                            model="gpt-4",
+                            model="gpt-4o",
                             messages=[{"role": "system", "content": "You are a helpful assistant."},
                                       {"role": "user", "content": prompt}]
                         )
@@ -152,18 +115,6 @@ 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
-                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!")
 
@@ -175,4 +126,4 @@ def main():
         st.info("Please enter your OpenAI API Key to begin the quiz.")
 
 if __name__ == "__main__":
-    main()
+    main()