Update app.py

app.py

@@ -17,10 +17,9 @@ def get_chemical_info(chemical_name):
 
         # Atoms
         output.append("\nAtoms:")
-        atom_dict = {}
-        for atom in compound.atoms:
-            atom_dict[atom.idx] = atom.element
-            output.append(f"  Atom {atom.idx}: Element = {atom.element}")
+        atom_dict = {i + 1: atom.element for i, atom in enumerate(compound.atoms)}
+        for idx, element in atom_dict.items():
+            output.append(f"  Atom {idx}: Element = {element}")
 
         # Bonds
         output.append("\nBonds:")
@@ -34,7 +33,6 @@ def get_chemical_info(chemical_name):
     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,