Update app.py

app.py

@@ -2,8 +2,6 @@ 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")
@@ -72,25 +70,6 @@ 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')
@@ -115,16 +94,11 @@ def show_mbti_quiz():
             if st.button("Generate Personality Trait Information"):
                 st.subheader("Your MBTI Personality Type:")
                 mbti_type_classic = classic_mbti_weighted(responses)
-                result_text = f"Your MBTI type based on weighted answers: {mbti_type_classic}"
-
-                # Display result text
-                st.write(result_text)
+                st.write(f"Your MBTI type based on weighted answers: {mbti_type_classic}")
 
-                # Create an image of the result
-                image_buf = create_image_from_text(result_text)
-
-                # LLM-based prediction
+                # You can add LLM-based prediction if needed here (example OpenAI-based model)
                 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:
@@ -133,33 +107,14 @@ 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}]
                         )
                         mbti_type_llm = response['choices'][0]['message']['content']
-                        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)
+                        st.write(f"Your MBTI type according to AI: {mbti_type_llm}")
                     except Exception as 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'.")
-
+                        st.error(f"Error occurred: {e}")
         else:
             st.warning("Please answer all the questions!")
 
@@ -171,4 +126,4 @@ def main():
         st.info("Please enter your OpenAI API Key to begin the quiz.")
 
 if __name__ == "__main__":
-    main()
+    main()