Update app.py

app.py

@@ -17,14 +17,16 @@ def get_chemical_info(chemical_name):
 
         # Atoms
         output.append("\nAtoms:")
-        for i, atom in enumerate(compound.atoms, start=1):
-            output.append(f"  Atom {i}: Element = {atom.element}")
+        atom_dict = {}
+        for atom in compound.atoms:
+            atom_dict[atom.idx] = atom.element
+            output.append(f"  Atom {atom.idx}: Element = {atom.element}")
 
         # Bonds
         output.append("\nBonds:")
         for bond in compound.bonds:
-            atom1 = bond.atom1.element
-            atom2 = bond.atom2.element
+            atom1 = atom_dict.get(bond.aid1, f"Atom {bond.aid1}")
+            atom2 = atom_dict.get(bond.aid2, f"Atom {bond.aid2}")
             output.append(f"  Bond between {atom1} and {atom2}, Order: {bond.order}")
 
         return "\n".join(output)