Corin1998 commited on
Commit
0588a0e
·
verified ·
1 Parent(s): 0e336e6

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +2 -9
app/main.py CHANGED
@@ -1,36 +1,29 @@
1
  from fastapi import FastAPI, Request, Depends, UploadFile, Form
2
- from fastapi.responses import HTMLResponse, RedirectResponse
3
- from fastapi.staticfiles import StaticFiles
4
  from fastapi.templating import Jinja2Templates
5
  from sqlalchemy.orm import Session
6
 
7
-
8
  from app.database import Base, engine, get_db
9
  from app import models
10
  from app.routers import content, deliver, health
11
 
12
-
13
  app = FastAPI(title="PR/IR MiniSaaS")
14
  app.include_router(content.router)
15
  app.include_router(deliver.router)
16
  app.include_router(health.router)
17
 
18
-
19
  templates = Jinja2Templates(directory="app/templates")
20
 
21
-
22
  @app.on_event("startup")
23
  def on_startup():
24
  Base.metadata.create_all(bind=engine)
25
 
26
-
27
  @app.get("/", response_class=HTMLResponse)
28
  async def index(request: Request, db: Session = Depends(get_db)):
29
  drafts = db.query(models.Draft).order_by(models.Draft.id.desc()).limit(20).all()
30
  return templates.TemplateResponse("index.html", {"request": request, "drafts": drafts, "Tone": models.Tone, "ContentType": models.ContentType})
31
 
32
-
33
  @app.get("/draft/{draft_id}", response_class=HTMLResponse)
34
  async def detail(draft_id: int, request: Request, db: Session = Depends(get_db)):
35
  d = db.get(models.Draft, draft_id)
36
- return templates.TemplateResponse("detail.html", {"request": request, "d": d})
 
1
  from fastapi import FastAPI, Request, Depends, UploadFile, Form
2
+ from fastapi.responses import HTMLResponse
 
3
  from fastapi.templating import Jinja2Templates
4
  from sqlalchemy.orm import Session
5
 
 
6
  from app.database import Base, engine, get_db
7
  from app import models
8
  from app.routers import content, deliver, health
9
 
 
10
  app = FastAPI(title="PR/IR MiniSaaS")
11
  app.include_router(content.router)
12
  app.include_router(deliver.router)
13
  app.include_router(health.router)
14
 
 
15
  templates = Jinja2Templates(directory="app/templates")
16
 
 
17
  @app.on_event("startup")
18
  def on_startup():
19
  Base.metadata.create_all(bind=engine)
20
 
 
21
  @app.get("/", response_class=HTMLResponse)
22
  async def index(request: Request, db: Session = Depends(get_db)):
23
  drafts = db.query(models.Draft).order_by(models.Draft.id.desc()).limit(20).all()
24
  return templates.TemplateResponse("index.html", {"request": request, "drafts": drafts, "Tone": models.Tone, "ContentType": models.ContentType})
25
 
 
26
  @app.get("/draft/{draft_id}", response_class=HTMLResponse)
27
  async def detail(draft_id: int, request: Request, db: Session = Depends(get_db)):
28
  d = db.get(models.Draft, draft_id)
29
+ return templates.TemplateResponse("detail.html", {"request": request, "d": d})