Update app.py

app.py

@@ -58,6 +58,7 @@ def inject_bfactors_and_save(pdb_file, scores_list, component_indices):
 
     return output_paths
 
+#--render the structure images
 def render_structure(pdb_path):
     with open(pdb_path, 'r') as f:
         pdb_data = f.read()
@@ -65,14 +66,15 @@ def render_structure(pdb_path):
     view.addModel(pdb_data, 'pdb')
     view.setStyle({'cartoon': {'color': 'bfactor'}})
     view.zoomTo()
+    
+    # Safely embed viewer + required 3Dmol.js script
+    viewer_html = view._make_html()
+    full_html = (
+        '<script src="https://3Dmol.org/build/3Dmol.js"></script>\n'
+        + viewer_html
+    )
+    return full_html
 
-    # Embed the 3Dmol.js script manually
-    html = f'''
-    <div id="viewer"></div>
-    <script src="https://3Dmol.org/build/3Dmol.js"></script>
-    {view._make_html().split("<script>")[1]}  <!-- extracts inner viewer JS only -->
-    '''
-    return html
 
 
 # Gradio UI