Update app.py

app.py

@@ -148,8 +148,22 @@ def load_protein_sequences(fasta_file):
         protein_sequences[record.id] = str(record.seq)
     return protein_sequences
 
+def show_split(inputfile):
+    seqs = load_protein_sequences(inputfile)
+    htmls = []
+    for seq in seqs:
+        pdb_path = f'{seq.replace(" ", "_").replace(",","")}.pdb'
+        html = fold_protein_wpdb(seqs[seq], pdb_path)
+        x = f"""<h3>>{seq}</h3>
+        <br>
+        """
+        htmls.append(x+html)
+    final = "\n<br>\n".join(htmls)
+    realhtml = "<div>\n"+final+"\n</div>"
+    return realhtml
+
 iface = gr.Interface(
-    title="Proteinviz",
+    title="SingleProteinviz",
     fn=fold_protein,
     inputs=gr.Textbox(
             label="Protein Sequence",
@@ -165,20 +179,14 @@ iface = gr.Interface(
     ]
 )
 
-with gr.Blocks() as demo1:
-    input_seqs = gr.File(label="FASTA File With Protein Sequences")
-    @gr.render(inputs=input_seqs, concurrency_limit=100)
-    def show_split(inputfile):
-        if inputfile is None:
-            gr.Markdown("## No Input Provided")
-        else:
-            seqs = load_protein_sequences(inputfile)
-            print("Loaded sequences")
-            for seq in seqs:
-                pdb_path = f'{seq.replace(" ", "_").replace(",","")}.pdb'
-                html = fold_protein_wpdb(seqs[seq], pdb_path)
-                print(f"Prediction for {seq} is over")
-                gr.HTML(html, label=f"{seq} structural representation")
+demo1 = gr.Interface(
+    title="BulkProteinviz",
+    fn=show_split,
+    inputs=gr.File(
+            label="FASTA File With Protein Sequences",
+        ),
+    outputs= gr.HTML(label="Protein 3D models")
+)
 
 
 demo = gr.TabbedInterface([iface, demo1], ["Single Protein Structure Prediction", "Bulk Protein Structure Prediction"])