rendering fix

app.py

@@ -8,6 +8,7 @@ MODELS = ModelFactory.models()
 
 def app(seq, sub, model_name, acc):
     "Main application function"
+    global progress
     scoring = "masked-marginals" if acc else "wt-marginals"
 
     # Validate the input
@@ -20,15 +21,17 @@ def app(seq, sub, model_name, acc):
     try:
         data = Data(seq, sub, model_name, scoring).calculate(progress)
         if isinstance(data.image, str):
-            return ( Image(value=data.image, type='filepath', visible=True)
-                   , HTML()
+            return ( Image(value=data.image, type='filepath', height=None, visible=True)
+                   , HTML(value='')
                    , DownloadButton(value=data.csv, visible=True) )
         else:
             return ( Image(visible=False)
                    , HTML(value=data.image.to_html(), visible=True)
                    , DownloadButton(value=data.csv, visible=True) )
     except Exception as e:
-        raise Error(str(e))        
+        raise Error(str(e))
+    finally:
+        progress = Progress()
     
 # Create the Gradio interface
 with Blocks() as demo:
@@ -38,22 +41,24 @@ with Blocks() as demo:
         Markdown(open("header.md", 'r', encoding="utf-8").read())
         seq = Textbox( lines=2
                      , label="Sequence"
-                     , placeholder="FASTA sequence here..."
+                     , placeholder="FASTA sequence here…"
                      , value='' )
         sub = Textbox( lines=1
                      , label="Substitutions"
-                     , placeholder="Substitutions here..."
+                     , placeholder="Substitutions here…"
                      , value='' )
         model_name  = Dropdown(MODELS, label="Model", value="facebook/esm2_t30_150M_UR50D")
         acc_box     = Checkbox(value=True, label="Use higher accuracy scoring", interactive=True)
         run_btn     = Button(value="Run", variant="primary")
         dl_btn      = DownloadButton(label="Download raw data", visible=False)
+        out_html    = HTML(visible=False)
         progress    = Progress()
-        out_html    = HTML()
         out_img     = Image(visible=False)
-        run_btn.click( fn=app
-                     , inputs=[seq, sub, model_name, acc_box]
-                     , outputs=[out_img, out_html, dl_btn] )
+        run_btn.click( fn=lambda : (HTML(value=None, visible=False), Image(value=None, visible=True, height=128))
+                     , outputs=[out_html, out_img]
+                     ).then( fn=app
+                           , inputs=[seq, sub, model_name, acc_box]
+                           , outputs=[out_img, out_html, dl_btn] )
         ex = Examples(
             examples=[
                 [   "MVEQYLLEAIVRDARDGITISDCSRPDNPLVFVNDAFTRMTGYDAEEVIGKNCRFLQRGDINLSAVHTIKIAMLTHEPCLVTLKNYRKDGTIFWNELSLTPIINKNGLITHYLGIQKDVSAQVILNQTLHEENHLLKSNKEMLEYLVNIDALTGLHNRRFLEDQLVIQWKLASRHINTITIFMIDIDYFKAFNDTYGHTAGDEALRTIAKTLNNCFMRGSDFVARYGGEEFTILAIGMTELQAHEYSTKLVQKIENLNIHHKGSPLGHLTISLGYSQANPQYHNDQNLVIEQADRALYSAKVEGKNRAVAYREQ"

data.py

@@ -1,8 +1,9 @@
-from math import ceil
+from math import ceil, exp
 import matplotlib.pyplot as plt
 import pandas as pd
 from re import match
 import seaborn as sns
+import asyncio
 
 from model import ModelFactory
 
@@ -178,12 +179,20 @@ class Data:
             ax[i].set_yticklabels(ax[i].get_yticklabels(), rotation=0)
             ax[i].set_xticklabels(ax[i].get_xticklabels(), rotation=90)
         fig.tight_layout()
-  
+
     def calculate(self, progress):
         "run model and parse output"
+        async def _parse_output():
+            self.progress(0, desc="Rendering")
+            ren = asyncio.create_task(asyncio.to_thread(self.parse_output))
+            t = 0
+            while not ren.done():
+                await asyncio.sleep(0.33)
+                t += 1
+                self.progress((1 - exp(-0.1 * t)), desc="Rendering")
         self.progress = progress
         self.model.run_model(self)
-        self.parse_output()
+        asyncio.run(_parse_output())
         return self
 
     @property