Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import openai
 from dotenv import load_dotenv
 import os
+from PIL import Image, ImageDraw, ImageFont
+import io
 
 # Load the OpenAI API Key
 api_key = st.text_input('Enter your OpenAI API Key', type="password")
@@ -70,6 +72,25 @@ def classic_mbti_weighted(responses):
             mbti_type += trait2
     return mbti_type
 
+# Function to create an image from text
+def create_image_from_text(text):
+    # Create an empty image with white background
+    img = Image.new('RGB', (600, 400), color='white')
+    draw = ImageDraw.Draw(img)
+
+    # Load font (You can customize font and size)
+    font = ImageFont.load_default()
+    
+    # Draw the text on the image
+    draw.text((10, 10), text, fill='black', font=font)
+    
+    # Save to a buffer
+    buf = io.BytesIO()
+    img.save(buf, format="PNG")
+    buf.seek(0)
+    
+    return buf
+
 # Streamlit component to display the quiz and handle responses
 def show_mbti_quiz():
     st.title('FlexTemp Personality Test')
@@ -94,11 +115,16 @@ def show_mbti_quiz():
             if st.button("Generate Personality Trait Information"):
                 st.subheader("Your MBTI Personality Type:")
                 mbti_type_classic = classic_mbti_weighted(responses)
-                st.write(f"Your MBTI type based on weighted answers: {mbti_type_classic}")
+                result_text = f"Your MBTI type based on weighted answers: {mbti_type_classic}"
+
+                # Display result text
+                st.write(result_text)
 
-                # You can add LLM-based prediction if needed here (example OpenAI-based model)
+                # Create an image of the result
+                image_buf = create_image_from_text(result_text)
+
+                # LLM-based prediction
                 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:
@@ -107,14 +133,33 @@ 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}]
                         )
                         mbti_type_llm = response['choices'][0]['message']['content']
-                        st.write(f"Your MBTI type according to AI: {mbti_type_llm}")
+                        llm_result_text = f"Your MBTI type according to AI: {mbti_type_llm}"
+
+                        # Create an image of the LLM result
+                        image_buf_llm = create_image_from_text(llm_result_text)
+
+                        # Display the generated LLM image
+                        st.image(image_buf_llm)
                     except Exception as e:
-                        st.error(f"Error occurred: {e}")
+                        st.error(f"Error occurred while using the AI model: {e}")
+                
+                # Save both the weighted result and LLM-based result as images
+                combined_text = result_text + "\n\n" + (llm_result_text if 'llm_result_text' in locals() else '')
+                combined_image_buf = create_image_from_text(combined_text)
+
+                # Display the combined result image
+                st.image(combined_image_buf)
+
+                # Optionally, save the image to disk
+                with open("personality_combined_result.png", "wb") as f:
+                    f.write(combined_image_buf.read())
+                st.success("Your result image has been saved as 'personality_combined_result.png'.")
+
         else:
             st.warning("Please answer all the questions!")