Update app.py

app.py

@@ -140,7 +140,7 @@ def fold_protein_wpdb(test_protein, pdb_path):
     with open(pdb_path, "w") as f:
         f.write("".join(pdb))
     html = molecule(pdb_path)
-    return html, pdb_path
+    return html
 
 def load_protein_sequences(fasta_file):
     protein_sequences = {} 
@@ -165,22 +165,6 @@ iface = gr.Interface(
     ]
 )
 
-# with gr.Blocks() as demo1:
-#     input_seqs = gr.File(label="FASTA File With Protein Sequences")
-#     @gr.render(inputs=input_seqs)
-#     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, pdb = fold_protein_wpdb(seqs[seq], pdb_path)
-#                 print(f"Prediction for {seq} is over")
-#                 gr.HTML(html, label=f"{seq} structural representation")   
-#                 Molecule3D(pdb, label=f"{seq} molecular representation")
-
 with gr.Blocks() as demo1:
     input_seqs = gr.File(label="FASTA File With Protein Sequences")
     @gr.render(inputs=input_seqs)
@@ -192,10 +176,11 @@ with gr.Blocks() as demo1:
             print("Loaded sequences")
             for seq in seqs:
                 pdb_path = f'{seq.replace(" ", "_").replace(",","")}.pdb'
-                html = f"<h1>{seq}</h1><br><h2>{seqs[seq]}</h2>"
+                html, pdb = fold_protein_wpdb(seqs[seq], pdb_path)
                 print(f"Prediction for {seq} is over")
                 gr.HTML(html, label=f"{seq} structural representation")
 
+
 demo = gr.TabbedInterface([iface, demo1], ["Single Protein Structure Prediction", "Bulk Protein Structure Prediction"])
 
 demo.launch(server_name="0.0.0.0", share=False)