Skip to content

Commit

Permalink
Add more templates to webapp
Browse files Browse the repository at this point in the history
  • Loading branch information
caufieldjh committed May 17, 2024
1 parent bb06196 commit 470047f
Showing 1 changed file with 23 additions and 1 deletion.
24 changes: 23 additions & 1 deletion src/ontogpt/webapp/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@
from starlette.staticfiles import StaticFiles
from starlette.templating import Jinja2Templates

from ontogpt.engines.knowledge_engine import DATAMODELS
from ontogpt.engines.spires_engine import SPIRESEngine
from ontogpt.io.html_exporter import HTMLExporter
from ontogpt.io.template_loader import get_template_details
Expand All @@ -19,6 +18,29 @@
static_dir = this_path / "static"
html_dir = this_path / "html"

DATAMODELS = [
"biological_process.BiologicalProcess",
"biotic_interaction.BioticInteraction",
"cell_type.CellTypeDocument",
"condition",
"ctd.ChemicalToDiseaseDocument",
"data_sheets_schema",
"diagnostic_procedure.DiagnosticProceduretoPhenotypeAssociation",
"dietitian_notes",
"drug.DrugMechanism",
"emapa_simple",
"environmental_sample.Study",
"figure",
"go_simple",
"gocam.GoCamAnnotations",
"mendelian_disease.MendelianDisease",
"maxo",
"mondo_simple",
"phenotype.Trait",
"reaction.Reaction",
"recipe.Recipe",
"treatment.DiseaseTreatmentSummary",
]

class Query(BaseModel):
text: str
Expand Down

0 comments on commit 470047f

Please sign in to comment.