Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 
+
 def get_pdb(pdb_code="", filepath=""):
     if pdb_code is None or pdb_code == "":
         try:
@@ -10,6 +11,7 @@ def get_pdb(pdb_code="", filepath=""):
         os.system(f"wget -qnc https://files.rcsb.org/view/{pdb_code}.pdb")
         return f"{pdb_code}.pdb"
 
+
 def read_mol(molpath):
     with open(molpath, "r") as fp:
         lines = fp.readlines()
@@ -17,7 +19,7 @@ def read_mol(molpath):
     for l in lines:
         mol += l
     return mol
-    
+
 
 def molecule(input_pdb):
 
@@ -71,19 +73,22 @@ def molecule(input_pdb):
     allow-scripts allow-same-origin allow-popups 
     allow-top-navigation-by-user-activation allow-downloads" allowfullscreen="" 
     allowpaymentrequest="" frameborder="0" srcdoc='{x}'></iframe>"""
-    
-  def update(inp, file):
+
+
+def update(inp, file):
     pdb_path = get_pdb(inp, file)
     return molecule(pdb_path)
-  
-  demo = gr.Blocks()
-  
-  with demo:
+
+
+demo = gr.Blocks()
+
+with demo:
     gr.Markdown("#PDB viewer using 3Dmol.js")
-    inp = gr.Textbox(placeholder="PDB Code or upload file below", label="Input structure")
+    inp = gr.Textbox(
+        placeholder="PDB Code or upload file below", label="Input structure"
+    )
     file = gr.File(file_count="single")
     btn = gr.Button("View structure")
     mol = gr.HTML()
-    btn.click(fn=update,inputs=[inp, file],outputs=mol]
-    
- demo.launch()
+    btn.click(fn=update, inputs=[inp, file], outputs=mol)
+demo.launch()