Update app.py

app.py

@@ -15,19 +15,24 @@ def get_chemical_info(chemical_name):
         output.append(f"Isotope Atom Count: {compound.isotope_atom_count}")
         output.append(f"Charge: {compound.charge}")
 
+        # Atoms
         output.append("\nAtoms:")
         for i, atom in enumerate(compound.atoms, start=1):
             output.append(f"  Atom {i}: Element = {atom.element}")
 
+        # Bonds
         output.append("\nBonds:")
         for bond in compound.bonds:
-            output.append(f"  Bond between Atom {bond.a1} and Atom {bond.a2}, Order: {bond.order}")
+            atom1 = bond.atom1.element
+            atom2 = bond.atom2.element
+            output.append(f"  Bond between {atom1} and {atom2}, Order: {bond.order}")
 
         return "\n".join(output)
 
     except IndexError:
         return f"No information found for '{chemical_name}'. Please try a more precise name."
 
+
 # Gradio Interface
 demo = gr.Interface(
     fn=get_chemical_info,