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,6 +94,41 @@ 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)
+
+        # Save the PDF to a file
+        pdf_file_path = '/mnt/data/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')
@@ -164,6 +199,17 @@ 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!")
 
@@ -171,7 +217,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**: 
@@ -189,4 +235,4 @@ def main():
         st.info("Please enter your OpenAI API Key to begin the quiz.")
 
 if __name__ == "__main__":
-    main()
+    main()