Update app.py

app.py

@@ -82,7 +82,12 @@ if uploaded_file is not None:
             election=str(df["instance"].iloc[0])
             statements = str(df["statement"].nunique())
             statement_n = df["statement"].nunique()
-            parties = str(df["party_name"].nunique())
+            try:
+                parties = str(df["party_name"].nunique())
+            except KeyError:
+                parties = str(df["party"].nunique())
+                df["party_name"] = df["party"]
+                
             st.success("You have uploaded the party answers for the VOTO instance: "+election+" ("+statements+" statements; "+parties+" Parties)")
             
             expander1 = st.expander("Statement Selection")