Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -159,7 +159,64 @@ def relax_wrapper(structure_file, task_name, steps, fmax, charge, spin, relax_ce
|
|
| 159 |
# ==== UI ====
|
| 160 |
with gr.Blocks(theme=gr.themes.Ocean(), title="OrbMol Demo") as demo:
|
| 161 |
with gr.Tabs():
|
| 162 |
-
# --------
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 163 |
with gr.Tab("Single Point Energy"):
|
| 164 |
with gr.Row():
|
| 165 |
with gr.Column(scale=2):
|
|
@@ -207,7 +264,7 @@ with gr.Blocks(theme=gr.themes.Ocean(), title="OrbMol Demo") as demo:
|
|
| 207 |
[spe_out, spe_status, spe_viewer]
|
| 208 |
)
|
| 209 |
|
| 210 |
-
# -------- MD --------
|
| 211 |
with gr.Tab("Molecular Dynamics"):
|
| 212 |
with gr.Row():
|
| 213 |
with gr.Column(scale=2):
|
|
@@ -263,7 +320,7 @@ with gr.Blocks(theme=gr.themes.Ocean(), title="OrbMol Demo") as demo:
|
|
| 263 |
[md_status, md_traj, md_log, md_script, md_explain, md_viewer]
|
| 264 |
)
|
| 265 |
|
| 266 |
-
# -------- Relax --------
|
| 267 |
with gr.Tab("Relaxation / Optimization"):
|
| 268 |
with gr.Row():
|
| 269 |
with gr.Column(scale=2):
|
|
|
|
| 159 |
# ==== UI ====
|
| 160 |
with gr.Blocks(theme=gr.themes.Ocean(), title="OrbMol Demo") as demo:
|
| 161 |
with gr.Tabs():
|
| 162 |
+
# -------- HOME TAB (NUEVA) --------
|
| 163 |
+
with gr.Tab("Home"):
|
| 164 |
+
with gr.Row():
|
| 165 |
+
with gr.Column(scale=1):
|
| 166 |
+
gr.Markdown("## Learn more about OrbMol")
|
| 167 |
+
|
| 168 |
+
with gr.Accordion("What is OrbMol?", open=False):
|
| 169 |
+
gr.Markdown("""
|
| 170 |
+
OrbMol is a suite of quantum-accurate machine learning models for molecular predictions. Built on the **Orb-v3 architecture**, OrbMol provides fast and accurate calculations of energies, forces, and molecular properties at the level of advanced quantum chemistry methods.
|
| 171 |
+
""")
|
| 172 |
+
|
| 173 |
+
with gr.Accordion("Available Models", open=False):
|
| 174 |
+
gr.Markdown("""
|
| 175 |
+
**OMol** and **OMol-Direct**
|
| 176 |
+
- **Training dataset**: OMol25 (>100M calculations on small molecules, biomolecules, metal complexes, and electrolytes)
|
| 177 |
+
- **Level of theory**: ωB97M-V/def2-TZVPD with non-local dispersion; solvation treated explicitly
|
| 178 |
+
- **Inputs**: total charge & spin multiplicity
|
| 179 |
+
- **Applications**: biology, organic chemistry, protein folding, small-molecule drugs, organic liquids, homogeneous catalysis
|
| 180 |
+
- **Caveats**: trained only on aperiodic systems → periodic/inorganic cases may not work well
|
| 181 |
+
- **Difference**: OMol enforces energy–force consistency; OMol-Direct relaxes this for efficiency
|
| 182 |
+
|
| 183 |
+
**OMat**
|
| 184 |
+
- **Training dataset**: OMat24 (>100M inorganic calculations, from Materials Project, Alexandria, and far-from-equilibrium samples)
|
| 185 |
+
- **Level of theory**: PBE/PBE+U with Materials Project settings; VASP 54 pseudopotentials; no dispersion
|
| 186 |
+
- **Inputs**: No support for spin and charge
|
| 187 |
+
- **Applications**: inorganic discovery, photovoltaics, alloys, superconductors, electronic/optical materials
|
| 188 |
+
""")
|
| 189 |
+
|
| 190 |
+
with gr.Accordion("Supported File Formats", open=False):
|
| 191 |
+
gr.Markdown("""
|
| 192 |
+
OrbMol supports: `.xyz`, `.pdb`, `.cif`, `.traj`, `.mol`, `.sdf`
|
| 193 |
+
""")
|
| 194 |
+
|
| 195 |
+
with gr.Accordion("Resources & Support", open=False):
|
| 196 |
+
gr.Markdown("""
|
| 197 |
+
- [Orb-v3 paper](https://arxiv.org/abs/2504.06231)
|
| 198 |
+
- [Orb-Models GitHub repository](https://github.com/orbital-materials/orb-models)
|
| 199 |
+
""")
|
| 200 |
+
|
| 201 |
+
with gr.Column(scale=2):
|
| 202 |
+
gr.Image("logo_color_text.png",
|
| 203 |
+
show_share_button=False,
|
| 204 |
+
show_download_button=False,
|
| 205 |
+
show_label=False,
|
| 206 |
+
show_fullscreen_button=False)
|
| 207 |
+
|
| 208 |
+
gr.Markdown("# OrbMol — Quantum-Accurate Molecular Predictions")
|
| 209 |
+
gr.Markdown("""
|
| 210 |
+
Welcome to the OrbMol demo! Use the tabs above to access different functionalities:
|
| 211 |
+
|
| 212 |
+
1. **Single Point Energy**: Calculate energies and forces
|
| 213 |
+
2. **Molecular Dynamics**: Run MD simulations
|
| 214 |
+
3. **Relaxation / Optimization**: Optimize structures
|
| 215 |
+
|
| 216 |
+
Supported formats: `.xyz`, `.pdb`, `.cif`, `.traj`, `.mol`, `.sdf`
|
| 217 |
+
""")
|
| 218 |
+
|
| 219 |
+
# -------- SPE (IDÉNTICA A LA QUE FUNCIONABA) --------
|
| 220 |
with gr.Tab("Single Point Energy"):
|
| 221 |
with gr.Row():
|
| 222 |
with gr.Column(scale=2):
|
|
|
|
| 264 |
[spe_out, spe_status, spe_viewer]
|
| 265 |
)
|
| 266 |
|
| 267 |
+
# -------- MD (IDÉNTICA A LA QUE FUNCIONABA) --------
|
| 268 |
with gr.Tab("Molecular Dynamics"):
|
| 269 |
with gr.Row():
|
| 270 |
with gr.Column(scale=2):
|
|
|
|
| 320 |
[md_status, md_traj, md_log, md_script, md_explain, md_viewer]
|
| 321 |
)
|
| 322 |
|
| 323 |
+
# -------- Relax (IDÉNTICA A LA QUE FUNCIONABA) --------
|
| 324 |
with gr.Tab("Relaxation / Optimization"):
|
| 325 |
with gr.Row():
|
| 326 |
with gr.Column(scale=2):
|