Update app.py

app.py

@@ -156,7 +156,8 @@ def visualize_protein_3d(pdb_data: str, title="Protein 3D Structure"):
     if not pdb_data:
         return None, "Cannot generate 3D view: No PDB data provided."
     try:
-        viewer = py3Dmol.view(width='100%', height=600)
+        js_url = 'https://cdn.jsdelivr.net/npm/[email protected]/build/3Dmol.js'
+        viewer = py3Dmol.view(width='100%', height=600, js=js_url)
         viewer.setBackgroundColor('#1C1C1C') 
         viewer.addModel(pdb_data, "pdb")
         viewer.setStyle({'cartoon': {'color': 'spectrum', 'thickness': 0.8}})
@@ -486,7 +487,8 @@ def visualize_molecule_2d_3d(smiles: str, name: str):
         AllChem.MMFFOptimizeMolecule(mol_3d)
         sdf_data = Chem.MolToMolBlock(mol_3d)
 
-        viewer = py3Dmol.view(width=400, height=300)
+        js_url = 'https://cdn.jsdelivr.net/npm/[email protected]/build/3Dmol.js'
+        viewer = py3Dmol.view(width=400, height=300, js=js_url)
         viewer.setBackgroundColor('#1C1C1C') 
         viewer.addModel(sdf_data, "sdf")
         viewer.setStyle({'stick': {}, 'sphere': {'scale': 0.25}})
@@ -518,7 +520,8 @@ def visualize_protein_ligand_interaction(pdb_data: str, pdb_id: str, ligand_resn
         return None, "Cannot generate interaction view: No PDB data provided."
     
     try:
-        viewer = py3Dmol.view(width='100%', height=650)
+        js_url = 'https://cdn.jsdelivr.net/npm/[email protected]/build/3Dmol.js'
+        viewer = py3Dmol.view(width='100%', height=650, js=js_url)
         viewer.setBackgroundColor('#1C1C1C')
         
         # Add the protein structure