Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
 import openai
 import json
-from fpdf import FPDF
 import os
+from dotenv import load_dotenv
 
 # Load the OpenAI API Key
 api_key = st.text_input('Enter your OpenAI API Key', type="password")
@@ -94,49 +94,6 @@ def save_personality_to_output_json(username, mbti_type_classic, mbti_type_llm):
     with open("Output.json", "w") as json_file:
         json.dump(output_data, json_file, indent=4)
 
-# Function to generate a PDF report from Output.json data
-def generate_pdf_report():
-    # Load the Output.json data
-    try:
-        with open("Output.json", "r") as json_file:
-            output_data = json.load(json_file)
-        
-        # Create PDF object
-        pdf = FPDF()
-        pdf.set_auto_page_break(auto=True, margin=15)
-        pdf.add_page()
-
-        # Set font
-        pdf.set_font("Arial", size=12)
-
-        # Title of the report
-        pdf.cell(200, 10, txt="FlexTemp Personality Test Report", ln=True, align='C')
-
-        # Add username
-        pdf.ln(10)
-        pdf.cell(200, 10, txt=f"Name: {output_data['username']}", ln=True)
-
-        # Add MBTI types
-        pdf.cell(200, 10, txt=f"MBTI Type (Classic): {output_data['mbti_type_classic']}", ln=True)
-        pdf.cell(200, 10, txt=f"MBTI Type (AI Prediction): {output_data['mbti_type_llm']}", ln=True)
-
-        # Add full Output.json content (including AI prediction and all data)
-        pdf.ln(10)
-        pdf.cell(200, 10, txt="Full Output Information:", ln=True)
-
-        # Break the full AI prediction into lines to avoid overflow
-        full_output = json.dumps(output_data, indent=4)
-        pdf.multi_cell(0, 10, full_output)
-
-        # Save the PDF to a file
-        pdf_file_path = 'flex_temp_personality_report.pdf'
-        pdf.output(pdf_file_path)
-
-        return pdf_file_path
-    except Exception as e:
-        st.error(f"Error occurred while generating the PDF report: {e}")
-        return None
-
 # Streamlit component to display the quiz and handle responses
 def show_mbti_quiz():
     st.title('FlexTemp Personality Test')
@@ -174,7 +131,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}]
                         )
@@ -207,17 +164,6 @@ def show_mbti_quiz():
                     mime="application/json"
                 )
 
-                # Add the Download PDF button
-                pdf_file_path = generate_pdf_report()
-
-                if pdf_file_path:
-                    st.download_button(
-                        label="Download PDF Report",
-                        data=open(pdf_file_path, "rb").read(),
-                        file_name="FlexTemp_Personality_Report.pdf",
-                        mime="application/pdf"
-                    )
-
         else:
             st.warning("Please answer all the questions!")
 
@@ -225,7 +171,7 @@ def show_mbti_quiz():
 def main():
     # Add instructions to the sidebar
     with st.sidebar.expander("How This App Works", expanded=False):
-        st.write("""  
+        st.write("""
             ### FlexTemp Personality Test
             This app is designed to help you determine your MBTI personality type based on your answers to a series of questions. The process works as follows:
             1. **Weighted MBTI Scoring**: 
@@ -243,4 +189,4 @@ def main():
         st.info("Please enter your OpenAI API Key to begin the quiz.")
 
 if __name__ == "__main__":
-    main()
+    main()