Update app.py

app.py

@@ -3,7 +3,6 @@ import openai
 import json
 import os
 from dotenv import load_dotenv
-from fpdf import FPDF
 
 # Load the OpenAI API Key
 api_key = st.text_input('Enter your OpenAI API Key', type="password")
@@ -95,33 +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
-def generate_pdf_report(participant_name, mbti_type_classic, mbti_type_llm):
-    pdf = FPDF()
-    pdf.set_auto_page_break(auto=True, margin=15)
-    pdf.add_page()
-
-    # Set fonts and title
-    pdf.set_font("Arial", 'B', 16)
-    pdf.cell(200, 10, txt="MBTI Personality Report", ln=True, align="C")
-
-    # Add participant name
-    pdf.ln(10)
-    pdf.set_font("Arial", '', 12)
-    pdf.cell(200, 10, txt=f"Participant Name: {participant_name}", ln=True)
-
-    # Add MBTI information
-    pdf.ln(10)
-    pdf.set_font("Arial", 'I', 12)
-    pdf.cell(200, 10, txt=f"Your MBTI type based on weighted answers: {mbti_type_classic}", ln=True)
-    pdf.cell(200, 10, txt=f"Your MBTI type according to AI: {mbti_type_llm}", ln=True)
-
-    # Save the PDF to a temporary file
-    pdf_output_path = "/tmp/mbti_report.pdf"
-    pdf.output(pdf_output_path)
-
-    return pdf_output_path
-
 # Streamlit component to display the quiz and handle responses
 def show_mbti_quiz():
     st.title('FlexTemp Personality Test')
@@ -149,10 +121,8 @@ 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 = "Not Available"  # Placeholder for now
-
-                # Run the LLM (GPT-4, for example) model to generate a personality type.
                 if api_key:
+                    # Run the LLM (GPT-4, for example) model to generate a personality type.
                     prompt = f"""
                     Determine a person's personality type based on their answers to the following Myers-Briggs Type Indicator (MBTI) questions:
                     The person has answered the following questions:
@@ -194,16 +164,6 @@ def show_mbti_quiz():
                     mime="application/json"
                 )
 
-                # Generate and add the button for the PDF report
-                pdf_report_path = generate_pdf_report(participant_name, mbti_type_classic, mbti_type_llm)
-                with open(pdf_report_path, "rb") as pdf_file:
-                    st.download_button(
-                        label="Download Generated Report",
-                        data=pdf_file,
-                        file_name="MBTI_Report.pdf",
-                        mime="application/pdf"
-                    )
-
         else:
             st.warning("Please answer all the questions!")
 
@@ -211,7 +171,8 @@ def show_mbti_quiz():
 def main():
     # Add instructions to the sidebar
     with st.sidebar.expander("How This App Works", expanded=False):
-        st.write(""" ### FlexTemp Personality Test
+        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**: 
             - Each question corresponds to a trait in the MBTI system. 
@@ -220,7 +181,7 @@ def main():
             2. **LLM-Based Prediction**:
             - Optionally, you can also get your MBTI type based on the answers using a language model (LLM) like GPT-4. This provides an additional prediction that may offer insights into your personality.
             - The LLM is trained on vast amounts of data and can generate responses based on patterns from psychological research and real-world interactions.
-        """)
+            """)
     
     if api_key:
         show_mbti_quiz()
@@ -228,4 +189,4 @@ def main():
         st.info("Please enter your OpenAI API Key to begin the quiz.")
 
 if __name__ == "__main__":
-    main()
+    main()