Update app.py

app.py

@@ -75,11 +75,11 @@ def parse_mol(output_json):
     for i, mol in enumerate(molecules_list):
         smiles = mol.get("smiles", "Unknown")
         label = mol.get("label", f"Mol {i+1}")
-        extra = mol.get("extra", "")
+        bbox = mol.get("bbox", [])
         # 可自定义格式
         mol_output = f"<b>Molecule:</b> {label}<br>" \
                      f"  SMILES: <span style='color:blue'>{smiles}</span><br>" \
-                     f"  Extra: {extra}<br><br>"
+                     f"  bbox: {bbox}<br><br>"
         detailed_output.append(mol_output)
         smiles_output.append(smiles)
     return detailed_output, smiles_output
@@ -222,7 +222,7 @@ with gr.Blocks() as demo:
                         try:
                             mol = Chem.MolFromSmiles(smiles_clean)
                             if mol:
-                                img = MolDraw.MolToImage(mol, size=(350, 150))
+                                img = Draw.MolToImage(mol, size=(350, 150))
                                 img_file = f"mol_{i}.png"
                                 img.save(img_file)
                                 components.append(gr.Image(value=img_file, label=f"RDKit Image of Molecule {i}"))