annabossler commited on
Commit
ea0059e
·
verified ·
1 Parent(s): 32f7c9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +60 -3
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
- # -------- SPE --------
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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):