Update app.py

app.py

@@ -1,115 +1,170 @@
 import gradio as gr
 
-# Define the list of items with their questions and whether they are reverse-scored
-items = [
-    {'BFI': 'BFI_1', 'question': 'Talks a lot', 'reverse': False},
-    {'BFI': 'BFI_2', 'question': 'Notices other people\'s weak points', 'reverse': True},
-    {'BFI': 'BFI_3', 'question': 'Does things carefully and completely', 'reverse': False},
-    {'BFI': 'BFI_4', 'question': 'Is sad, depressed', 'reverse': False},
-    {'BFI': 'BFI_5', 'question': 'Is original, comes up with new ideas', 'reverse': False},
-    {'BFI': 'BFI_6', 'question': 'Keeps their thoughts to themselves', 'reverse': True},
-    {'BFI': 'BFI_7', 'question': 'Is helpful and not selfish with others', 'reverse': False},
-    {'BFI': 'BFI_8', 'question': 'Can be kind of careless', 'reverse': True},
-    {'BFI': 'BFI_9', 'question': 'Is relaxed, handles stress well', 'reverse': True},
-    {'BFI': 'BFI_10', 'question': 'Is curious about lots of different things', 'reverse': False},
-    {'BFI': 'BFI_11', 'question': 'Has a lot of energy', 'reverse': False},
-    {'BFI': 'BFI_12', 'question': 'Starts arguments with others', 'reverse': True},
-    {'BFI': 'BFI_13', 'question': 'Is a good, hard worker', 'reverse': False},
-    {'BFI': 'BFI_14', 'question': 'Can be tense; not always easy going', 'reverse': False},
-    {'BFI': 'BFI_15', 'question': 'Clever; thinks a lot', 'reverse': False},
-    {'BFI': 'BFI_16', 'question': 'Makes things exciting', 'reverse': False},
-    {'BFI': 'BFI_17', 'question': 'Forgives others easily', 'reverse': False},
-    {'BFI': 'BFI_18', 'question': 'Isn\'t very organized', 'reverse': True},
-    {'BFI': 'BFI_19', 'question': 'Worries a lot', 'reverse': False},
-    {'BFI': 'BFI_20', 'question': 'Has a good, active imagination', 'reverse': False},
-    {'BFI': 'BFI_21', 'question': 'Tends to be quiet', 'reverse': True},
-    {'BFI': 'BFI_22', 'question': 'Usually trusts people', 'reverse': False},
-    {'BFI': 'BFI_23', 'question': 'Tends to be lazy', 'reverse': True},
-    {'BFI': 'BFI_24', 'question': 'Doesn\'t get upset easily; steady', 'reverse': True},
-    {'BFI': 'BFI_25', 'question': 'Is creative and inventive', 'reverse': False},
-    {'BFI': 'BFI_26', 'question': 'Has a good, strong personality', 'reverse': False},
-    {'BFI': 'BFI_27', 'question': 'Can be cold and distant with others', 'reverse': True},
-    {'BFI': 'BFI_28', 'question': 'Keeps working until things are done', 'reverse': False},
-    {'BFI': 'BFI_29', 'question': 'Can be moody', 'reverse': False},
-    {'BFI': 'BFI_30', 'question': 'Likes artistic and creative experiences', 'reverse': False},
-    {'BFI': 'BFI_31', 'question': 'Is kind of shy', 'reverse': True},
-    {'BFI': 'BFI_32', 'question': 'Kind and considerate to almost everyone', 'reverse': False},
-    {'BFI': 'BFI_33', 'question': 'Does things quickly and carefully', 'reverse': False},
-    {'BFI': 'BFI_34', 'question': 'Stays calm in difficult situations', 'reverse': True},
-    {'BFI': 'BFI_35', 'question': 'Likes work that is the same every time', 'reverse': True},
-    {'BFI': 'BFI_36', 'question': 'Is outgoing; likes to be with people', 'reverse': False},
-    {'BFI': 'BFI_37', 'question': 'Is sometimes rude to others', 'reverse': True},
-    {'BFI': 'BFI_38', 'question': 'Makes plans and sticks to them', 'reverse': False},
-    {'BFI': 'BFI_39', 'question': 'Gets nervous easily', 'reverse': False},
-    {'BFI': 'BFI_40', 'question': 'Likes to think and play with ideas', 'reverse': False},
-    {'BFI': 'BFI_41', 'question': 'Doesn\'t like artistic things (plays, music)', 'reverse': True},
-    {'BFI': 'BFI_42', 'question': 'Likes to cooperate; goes along with others', 'reverse': False},
-    {'BFI': 'BFI_43', 'question': 'Has trouble paying attention', 'reverse': True},
-    {'BFI': 'BFI_44', 'question': 'Knows a lot about art, music, and books', 'reverse': False},
+# Define the 44 BFI questions
+questions = [
+    "1. Talks a lot",
+    "2. Notices other people’s weak points",
+    "3. Does things carefully and completely",
+    "4. Is sad, depressed",
+    "5. Is original, comes up with new ideas",
+    "6. Keeps their thoughts to themselves",
+    "7. Is helpful and not selfish with others",
+    "8. Can be kind of careless",
+    "9. Is relaxed, handles stress well",
+    "10. Is curious about lots of different things",
+    "11. Has a lot of energy",
+    "12. Starts arguments with others",
+    "13. Is a good, hard worker",
+    "14. Can be tense; not always easy going",
+    "15. Clever; thinks a lot",
+    "16. Makes things exciting",
+    "17. Forgives others easily",
+    "18. Isn’t very organized",
+    "19. Worries a lot",
+    "20. Has a good, active imagination",
+    "21. Tends to be quiet",
+    "22. Usually trusts people",
+    "23. Tends to be lazy",
+    "24. Doesn’t get upset easily; steady",
+    "25. Is creative and inventive",
+    "26. Has a good, strong personality",
+    "27. Can be cold and distant with others",
+    "28. Keeps working until things are done",
+    "29. Can be moody",
+    "30. Likes artistic and creative experiences",
+    "31. Is kind of shy",
+    "32. Kind and considerate to almost everyone",
+    "33. Does things quickly and carefully",
+    "34. Stays calm in difficult situations",
+    "35. Likes work that is the same every time",
+    "36. Is outgoing; likes to be with people",
+    "37. Is sometimes rude to others",
+    "38. Makes plans and sticks to them",
+    "39. Get nervous easily",
+    "40. Likes to think and play with ideas",
+    "41. Doesn’t like artistic things (plays, music)",
+    "42. Likes to cooperate; goes along with others",
+    "43. Has trouble paying attention",
+    "44. Knows a lot about art, music and books"
 ]
 
-# Define which items belong to which traits
-traits = {
-    'Extraversion': {
-        'positive': ['BFI_1', 'BFI_11', 'BFI_16', 'BFI_26', 'BFI_36'],
-        'negative': ['BFI_6', 'BFI_21', 'BFI_31'],
-    },
-    'Agreeableness': {
-        'positive': ['BFI_7', 'BFI_17', 'BFI_22', 'BFI_32', 'BFI_42'],
-        'negative': ['BFI_2', 'BFI_12', 'BFI_27', 'BFI_37'],
-    },
-    'Conscientiousness': {
-        'positive': ['BFI_3', 'BFI_13', 'BFI_28', 'BFI_33', 'BFI_38'],
-        'negative': ['BFI_8', 'BFI_18', 'BFI_23', 'BFI_43'],
-    },
-    'Neuroticism': {
-        'positive': ['BFI_4', 'BFI_14', 'BFI_19', 'BFI_29', 'BFI_39'],
-        'negative': ['BFI_9', 'BFI_24', 'BFI_34'],
-    },
-    'Openness': {
-        'positive': ['BFI_5', 'BFI_10', 'BFI_15', 'BFI_20', 'BFI_25', 'BFI_30', 'BFI_40', 'BFI_44'],
-        'negative': ['BFI_35', 'BFI_41'],
+# Scoring function based on the provided SPSS syntax
+def compute_bfi_scores(*args):
+    responses = list(args)
+    # Convert 'No response' to None, else to int
+    processed = []
+    for r in responses:
+        if r == "No response":
+            processed.append(None)
+        else:
+            processed.append(int(r))
+    
+    # Define traits with their respective items and scoring parameters
+    traits = {
+        'Extraversion': {
+            'positive': [1, 11, 16, 26, 36],
+            'reverse': [6, 21, 31],
+            'threshold': 1,
+            'formula_pos_mult': 5,
+            'formula_reverse_mult': 3,
+            'formula_reverse_const': 12
+        },
+        'Agreeableness': {
+            'positive': [7, 17, 22, 32, 42],
+            'reverse': [2, 12, 27, 37],
+            'threshold': 1,
+            'formula_pos_mult': 5,
+            'formula_reverse_mult':4,
+            'formula_reverse_const':16
+        },
+        'Conscientiousness': {
+            'positive': [3, 13, 28, 33, 38],
+            'reverse': [8, 18, 23, 43],
+            'threshold': 1,
+            'formula_pos_mult':5,
+            'formula_reverse_mult':4,
+            'formula_reverse_const':16
+        },
+        'Neuroticism':{
+            'positive':[4, 14, 19, 29, 39],
+            'reverse':[9, 24, 34],
+            'threshold':1,
+            'formula_pos_mult':5,
+            'formula_reverse_mult':3,
+            'formula_reverse_const':12
+        },
+        'Openness':{
+            'positive':[5, 10, 15, 20, 25, 30, 40, 44],
+            'reverse':[35, 41],
+            'threshold':2,
+            'formula_pos_mult':8,
+            'formula_reverse_mult':2,
+            'formula_reverse_const':8
+        }
     }
-}
 
-# Define the scoring function
-def compute_trait_scores(*args):
-    responses = {}
-    for i, item in enumerate(items):
-        BFI = item['BFI']
-        response = int(args[i])
-        responses[BFI] = response
+    scores = {}
 
-    trait_scores = {}
-    for trait in traits:
-        positive_items = traits[trait]['positive']
-        negative_items = traits[trait]['negative']
+    for trait, info in traits.items():
+        pos_items = [processed[i-1] for i in info['positive']]
+        rev_items = [processed[i-1] for i in info['reverse']]
 
-        positive_sum = sum(responses[item] for item in positive_items)
-        negative_sum = sum(6 - responses[item] for item in negative_items)
+        missing_pos = pos_items.count(None)
+        missing_rev = rev_items.count(None)
+        total_missing = missing_pos + missing_rev
 
-        trait_scores[trait] = positive_sum + negative_sum
+        if total_missing > info['threshold']:
+            scores[trait] = "Incomplete"
+        else:
+            # Compute means, ignoring None
+            pos_values = [x for x in pos_items if x is not None]
+            rev_values = [x for x in rev_items if x is not None]
 
-    return trait_scores
+            mean_pos = sum(pos_values) / len(pos_values) if pos_values else 0
+            mean_rev = sum(rev_values) / len(rev_values) if rev_values else 0
+
+            # Apply the scoring formula
+            score = (mean_pos * info['formula_pos_mult']) + (info['formula_reverse_const'] - (mean_rev * info['formula_reverse_mult']))
+            score = round(score, 2)
+            scores[trait] = score
+
+    # Prepare the output in Markdown format
+    output = "## Your Big Five Personality Traits Scores\n\n"
+    for trait, score in scores.items():
+        output += f"**{trait}**: {score}\n\n"
+
+    return output
 
 # Create the Gradio interface
-inputs = []
-for item in items:
-    question = item['question']
-    input_component = gr.Radio(
-        label=question,
-        choices=['1 (Disagree a lot)', '2', '3', '4', '5 (Agree a lot)'],
-        type='value',
-    )
-    inputs.append(input_component)
+def create_interface():
+    with gr.Blocks() as demo:
+        gr.Markdown("## Big Five Inventory (BFI) Quiz")
+        gr.Markdown("Please rate the following statements on a scale from **1 (Disagree a lot)** to **5 (Agree a lot)**. If you prefer not to respond to a particular statement, select **'No response'**.")
+
+        # Create a grid layout for better organization
+        with gr.Row():
+            with gr.Column(scale=1):
+                pass  # Empty column for spacing
+            with gr.Column(scale=10):
+                dropdowns = []
+                for q in questions:
+                    dropdown = gr.Dropdown(
+                        choices=["No response", 1, 2, 3, 4, 5],
+                        label=q,
+                        value="No response"
+                    )
+                    dropdowns.append(dropdown)
+        
+        # Submit button
+        submit_btn = gr.Button("Submit")
+        submit_btn.click(compute_bfi_scores, inputs=dropdowns, outputs="result")
+
+        # Results display
+        result = gr.Markdown()
+        demo.add_component(result, "result")
 
-iface = gr.Interface(
-    fn=compute_trait_scores,
-    inputs=inputs,
-    outputs=gr.outputs.JSON(label="Your Big Five Personality Scores"),
-    title='Big Five Personality Test',
-    description='Answer the following questions to see your Big Five personality trait scores.',
-    allow_flagging='never'
-)
+    return demo
 
-iface.launch()
+# Launch the interface
+demo = create_interface()
+demo.launch()