Added error handling.

app.py

@@ -1,9 +1,7 @@
 import gradio as gr
-import matplotlib.pyplot as plt
 import functools
 import requests
 import math
-import plotly as py
 import plotly.express as px
 
 
@@ -55,6 +53,8 @@ def detect(text):
     x = requests.post(GET_RESULT, data={'text': text}, headers=headers).json()
     response = x['result']
     status = x['status']
+    if status != 'OK':
+        return gr.update(visible=False), gr.update(visible=False), status, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
     original_score = response[0]
     perturbed_scores = response[1]
@@ -171,7 +171,7 @@ def detect(text):
     PERTURBATION_LOGPS = perturbed_scores
     DATAFRAME_PAGE = 0
 
-    return fig, update_perturbations_dataframe(), result, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+    return gr.update(value=fig, visible=True), update_perturbations_dataframe(), gr.update(value=result, visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 
 def generate(text):
@@ -187,7 +187,7 @@ def update_perturbations_dataframe():
     perturbed_texts = PERTURBATIONS[DATAFRAME_PAGE * PERTURBATIONS_PER_PAGE: (DATAFRAME_PAGE + 1) * PERTURBATIONS_PER_PAGE]
     perturbed_scores = PERTURBATION_LOGPS[DATAFRAME_PAGE * PERTURBATIONS_PER_PAGE: (DATAFRAME_PAGE + 1) * PERTURBATIONS_PER_PAGE]
     data = [[t, s] for t, s in zip(perturbed_texts, perturbed_scores)]
-    return gr.Dataframe.update(data)
+    return gr.Dataframe.update(data, visible=True)
 
 
 
@@ -290,8 +290,8 @@ with gr.Blocks() as demo:
                 with gr.Column(scale=8):
                     pass
             detect_results_text = gr.Markdown()
-            results_plot = gr.Plot()
-            perturbations_dataframe = gr.DataFrame(label="Perturbed texts", headers=['Perturbed Text', 'Log Prob'], datatype=["str", "number"], wrap=True, max_rows=5)
+            results_plot = gr.Plot(visible=False)
+            perturbations_dataframe = gr.DataFrame(label="Perturbed texts", headers=['Perturbed Text', 'Log Prob'], datatype=["str", "number"], wrap=True, max_rows=5, visible=False)
 
             page_label = gr.Markdown("Page 1", visible=False)
             next_page_button = gr.Button("Next Page", visible=False)