Update app.py

app.py

@@ -66,10 +66,9 @@ css = """
 """
 st.markdown(css, unsafe_allow_html=True)
 
-# Allow the user to choose whether to rate a movie or a sitcom.
-rating_type = st.radio("Select Rating Type", ["Movie", "Sitcom"], index=0)
+# Let the user select which type of show they are rating.
+rating_type = st.radio("Select Rating Type", ["Movie", "Sitcom", "Reality TV"], index=0)
 
-# Define criteria based on the rating type.
 if rating_type == "Movie":
     criteria = {
         "Writing": [
@@ -115,8 +114,9 @@ if rating_type == "Movie":
             {"name": "Emotional Impact", "description": "Strength of connection, intensity of response."}
         ]
     }
-    scoring_weights = None  # Use flat averaging for movies.
-else:
+    scoring_weights = None  # Flat averaging for movies.
+
+elif rating_type == "Sitcom":
     criteria = {
         "Writing & Humor": [
             {"name": "Dialogue & Humor", "description": "Witty exchanges, comedic timing, and joke quality."},
@@ -140,7 +140,6 @@ else:
             {"name": "Rewatchability", "description": "Longevity of appeal and repeat enjoyment."}
         ]
     }
-    # Define group weights for sitcoms (weights sum to 1)
     scoring_weights = {
         "Writing & Humor": 0.3,
         "Performance & Chemistry": 0.3,
@@ -149,9 +148,34 @@ else:
         "Audience Engagement": 0.1
     }
 
+else:  # Reality TV (e.g., Survivor)
+    criteria = {
+        "Social & Strategy": [
+            {"name": "Alliance & Social Dynamics", "description": "Quality of contestant interactions, alliances, and social gameplay."},
+            {"name": "Strategy & Decision Making", "description": "Smart, tactical decisions and risk management."}
+        ],
+        "Challenges & Endurance": [
+            {"name": "Physical Performance", "description": "Effectiveness and endurance in physical challenges."},
+            {"name": "Mental Agility", "description": "Problem-solving skills and performance in mental challenges."}
+        ],
+        "Production & Narrative": [
+            {"name": "Editing & Storytelling", "description": "Crafting an engaging narrative from real events."},
+            {"name": "Cinematography & Environment", "description": "Quality of visuals and effective use of natural settings."}
+        ],
+        "Host & Engagement": [
+            {"name": "Host Performance", "description": "Charisma, control, and the ability to engage contestants and audience."},
+            {"name": "Overall Entertainment", "description": "The show's excitement, drama, and rewatchability."}
+        ]
+    }
+    scoring_weights = {
+        "Social & Strategy": 0.3,
+        "Challenges & Endurance": 0.3,
+        "Production & Narrative": 0.2,
+        "Host & Engagement": 0.2
+    }
+
 def calculate_score(criteria, weights=None):
     if weights is None:
-         # For movies: flat average of all ratings.
          ratings = []
          for group, subcriteria in criteria.items():
               for idx in range(len(subcriteria)):
@@ -164,10 +188,9 @@ def calculate_score(criteria, weights=None):
          if not valid:
              return "Please rate at least one criterion."
          avg = sum(valid) / len(valid)
-         final_score = (avg + 1) * 5  # Maps average in [-1, 1] to [0, 10]
+         final_score = (avg + 1) * 5  # Map from [-1, 1] to [0, 10]
          return f"Final Score: {final_score:.2f} / 10"
     else:
-         # For sitcoms: compute a weighted average by criteria group.
          overall = 0
          for group, subcriteria in criteria.items():
               group_ratings = []
@@ -186,10 +209,13 @@ def calculate_score(criteria, weights=None):
          final_score = (overall + 1) * 5
          return f"Final Score: {final_score:.2f} / 10"
 
-header_text = "Movie Rating System" if rating_type == "Movie" else "Sitcom Rating System"
+header_text = (
+    "Movie Rating System" if rating_type == "Movie"
+    else "Sitcom Rating System" if rating_type == "Sitcom"
+    else "Reality TV Rating System"
+)
 st.markdown(f"<h1 class='app-header'>{header_text}</h1>", unsafe_allow_html=True)
 
-# Loop over each criterion and display the checkbox, text, and three horizontal buttons for rating.
 for group, subcriteria in criteria.items():
     st.markdown(f"<div class='category-header'>{group}</div>", unsafe_allow_html=True)
     for idx, crit in enumerate(subcriteria):
@@ -198,7 +224,6 @@ for group, subcriteria in criteria.items():
             st.session_state[f"rating_{key_prefix}"] = None
 
         with st.container():
-            # Two columns: checkbox on the left and content on the right.
             cols = st.columns([0.5, 7])
             checkbox_col = cols[0]
             content_col = cols[1]
@@ -211,7 +236,6 @@ for group, subcriteria in criteria.items():
             )
             content_col.markdown(combined, unsafe_allow_html=True)
 
-            # Wrap the buttons in a container for styling.
             content_col.markdown('<div class="binary-btns-container">', unsafe_allow_html=True)
             btn_cols = content_col.columns(3, gap="small")
             if btn_cols[0].button("👎", key=f"btn_down_{key_prefix}"):