Update app.py

app.py

@@ -151,8 +151,10 @@ def load_protein_sequences(fasta_file):
 def show_split(inputfile):
     seqs = load_protein_sequences(inputfile)
     htmls = []
+    pdb_paths = []
     for seq in seqs:
         pdb_path = f'{seq.replace(" ", "_").replace(",","")}.pdb'
+        pdb_paths.append(pdb_path)
         html = fold_protein_wpdb(seqs[seq], pdb_path)
         x = f"""<h3>>{seq}</h3>
         <br>
@@ -160,7 +162,7 @@ def show_split(inputfile):
         htmls.append(x+html)
     final = "\n<br>\n".join(htmls)
     realhtml = "<div>\n"+final+"\n</div>"
-    return realhtml
+    return realhtml, pdb_paths
 
 iface = gr.Interface(
     title="SingleProteinviz",
@@ -185,7 +187,7 @@ demo1 = gr.Interface(
     inputs=gr.File(
             label="FASTA File With Protein Sequences",
         ),
-    outputs= gr.HTML(label="Protein 3D models"),
+    outputs= [gr.HTML(label="Protein 3D models"), Molecule3D(label="Molecular 3D models", reps=reps, file_count="multiple")],
     examples = ["proteins.fasta"]
 )