Invalid prediction check

app.py

@@ -122,7 +122,7 @@ def run_ui():
         st.stop()
 
     elif submitted or st.session_state.valid_inputs_received:
-        pubmed_query = GPTHelper.gpt35_rephrase(fact)  # Call gpt3.5 to rephrase fact as a PubMed query.
+        pubmed_query = GPTHelper.gpt35_rephrase(fact)  # Call gpt3.5 to rephrase the fact as a PubMed query.
         pubmed = load_pubmed_fetcher()
 
         with st.spinner('Fetching articles...'):
@@ -166,13 +166,15 @@ def run_ui():
             elif prediction == 'Undetermined':
                 predictions.append(prediction)
             else:
+                # If GPT3.5 returns an invalid response. Has not happened during testing.
+                predictions.append('Invalid')
                 logging.warning(f'Unexpected prediction: {prediction}')
 
             percent_complete += step/100
             fact_checking_bar.progress(round(percent_complete, 2), text=progress_text)
         fact_checking_bar.empty()
         df['Prediction'] = predictions
-
+        df = df[df.Prediction != 'Invalid']  # Drop rows with invalid prediction.
         # Prepare DataFrame for plotly sunburst chart.
         totals = df.groupby('Prediction').size().to_dict()
         df['Total'] = df['Prediction'].map(totals)