Update app.py

app.py

@@ -226,7 +226,11 @@ def analyze_composite(msg1, date1, msg2, date2, msg3, date3, *answers_and_none):
         return "Please enter at least one message."
     results = [(analyze_single_message(m), d) for m,d in active]
     abuse_scores = [r[0]['abuse_score'] for r in results]
-    top_labels = [r[0]['top_patterns'][0][0] if r[0]['top_patterns'] else 'None' for r,_ in results]
+    stage_list = [r[0]['stage'] for r,_ in results]
+    most_common_stage = max(set(stage_list), key=stage_list.count)
+    composite_abuse = int(round(sum(abuse_scores)/len(abuse_scores)))
+    # extract top label from each result dict
+    top_labels = [res['top_patterns'][0][0] if res['top_patterns'] else 'None' for res, _ in results]['top_patterns'][0][0] if r[0]['top_patterns'] else 'None' for r,_ in results]
     dates_used = [d or 'Undated' for _,d in results]
     # common stage
     stage_list = [r[0]['stage'] for r,_ in results]
@@ -269,4 +273,3 @@ iface = gr.Interface(
 )
 if __name__ == "__main__":
     iface.launch()
-