Update app.py

app.py

@@ -10,7 +10,6 @@ model = EsmForProteinFolding.from_pretrained("facebook/esmfold_v1")
 
 def generate_structure(sequence, style):
     try:
-        # 🔬 Generate structure
         inputs = tokenizer([sequence], return_tensors="pt", add_special_tokens=False)
 
         with torch.no_grad():
@@ -18,12 +17,12 @@ def generate_structure(sequence, style):
 
         pdb = model.output_to_pdb(outputs)[0]
 
-        # 💾 Save file
+        # Save file
         file_path = "output.pdb"
         with open(file_path, "w") as f:
             f.write(pdb)
 
-        # 🎨 Create 3D visualization
+        # 🔥 SAFE visualization
         view = py3Dmol.view(width=800, height=500)
         view.addModel(pdb, "pdb")
 
@@ -35,9 +34,15 @@ def generate_structure(sequence, style):
             view.setStyle({"sphere": {}})
 
         view.zoomTo()
+
+        # ✅ safer HTML export
         html = view._make_html()
+        html = f"""
+        <div style="width:100%; height:500px;">
+            {html}
+        </div>
+        """
 
-        # 📄 Preview (first 20 lines)
         preview = "\n".join(pdb.split("\n")[:20])
 
         return file_path, html, preview