hacky fix

views/questions_screen.py

@@ -45,7 +45,7 @@ def get_previous_ratings(model_name, query_key, current_index):
         response_from_session = st.session_state.responses[current_index]
         try:
             previous_ratings = response_from_session.model_ratings.get(model_name, {})
-        except AttributeError:
+        except (AttributeError, KeyError):
             previous_ratings = response_from_session["model_ratings"].get(model_name, {})
 
     stored_query_ratings = {}
@@ -53,17 +53,17 @@ def get_previous_ratings(model_name, query_key, current_index):
         if "query_v" in query_key:
             try:
                 stored_query_ratings = previous_ratings.query_v_ratings
-            except AttributeError:
+            except (AttributeError, KeyError):
                 stored_query_ratings = previous_ratings["query_v_ratings"]
         elif "query_p0" in query_key:
             try:
                 stored_query_ratings = previous_ratings.query_p0_ratings
-            except AttributeError:
+            except (AttributeError, KeyError):
                 stored_query_ratings = previous_ratings["query_p0_ratings"]
         elif "query_p1" in query_key:
             try:
                 stored_query_ratings = previous_ratings.query_p1_ratings
-            except AttributeError:
+            except (AttributeError, KeyError):
                 stored_query_ratings = previous_ratings["query_p1_ratings"]
 
     return stored_query_ratings if stored_query_ratings else {}