Spaces:
Sleeping
Sleeping
File size: 4,221 Bytes
8311f5b d0f5098 28a5857 8311f5b 28a5857 8311f5b d0f5098 957c035 d0f5098 8311f5b d0f5098 f2f5171 28a5857 f2f5171 d0f5098 f2f5171 8311f5b 28a5857 d0f5098 8311f5b 28a5857 8311f5b 28a5857 8311f5b d0f5098 28a5857 d0f5098 8311f5b d0f5098 8311f5b 28a5857 957c035 d0f5098 8311f5b d0f5098 8311f5b d95cbea d0f5098 d95cbea d0f5098 d95cbea 8311f5b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 |
import random
import gradio as gr
# from fastapi.security import APIKeyHeader
from fastapi import FastAPI, Request, Depends
from fastapi.staticfiles import StaticFiles
from fastapi.responses import HTMLResponse, RedirectResponse
from fastapi.templating import Jinja2Templates
from loguru import logger
from app import Summarizer, TextRequest
from app import (
EN_SENTIMENT_MODEL,
EN_SUMMARY_MODEL,
)
from app import DEFAULT_EN_TEXT
from models.forms import VerificationForm
from models.exceptions import MyHTTPException, my_http_exception_handler
from models.security import AuthUsers
app = FastAPI()
pipe = Summarizer()
users = AuthUsers()
app.mount("/static", StaticFiles(directory="static"), name="static")
templates = Jinja2Templates(directory="templates")
# auth_header = APIKeyHeader(name="Authorization", auto_error=True)
# router = APIRouter(
# prefix="/v1",
# tags=["API v1"],
# dependencies=[Security(auth_header)],
# )
# app.include_router(router)
app.add_exception_handler(MyHTTPException, my_http_exception_handler)
@app.middleware("http")
async def add_process_time_header(request: Request, call_next):
response = await call_next(request)
token = request.cookies.get("Authorization")
dest = request.url.path.split("/")[-1]
resp_headers = dict(response.headers)
cont_type = resp_headers.get("content-type")
if (
not cont_type
or "text/html" not in cont_type
or dest
in [
"docs",
"verify_page",
"verify",
]
):
return response
if not token or token == "":
return RedirectResponse("/verify_page", status_code=307)
return response
@app.get("/verify_page", response_class=HTMLResponse)
async def verify_page(request: Request):
captcha_id = random.randint(1, 5)
return templates.TemplateResponse(
request=request,
name="verification.html",
context={"captcha_id": captcha_id},
)
@app.post("/verify")
async def verify(request: Request, checked_user: bool = Depends(users.get_cookie_data)):
form = VerificationForm(request)
await form.load_data()
if await form.is_valid():
logger.info("Form is valid")
response = RedirectResponse("/index", status_code=302)
if not checked_user:
user_token = users.generate_user_token()
users.add_user_session(user_token)
response.set_cookie(key="Authorization", value=user_token)
logger.info(f"Issued token: {user_token}")
return response
logger.info("Validation error")
return await verify_page(request)
@app.get("/")
async def get_main_page():
return RedirectResponse("/index", status_code=302)
def get_summary(text: TextRequest, lang: str, request: gr.Request):
token = request.cookies["Authorization"]
if users.verify_user(token):
users.add_user_session(token)
return pipe.get_summary(text, lang)
logger.info("User not verified")
return "Sorry. You are not verified."
with gr.Blocks() as demo:
with gr.Column(scale=2, min_width=600):
en_sum_description = gr.Markdown(value=f"Model for Summary: {EN_SUMMARY_MODEL}")
en_sent_description = gr.Markdown(
value=f"Model for Sentiment: {EN_SENTIMENT_MODEL}"
)
verify_href = gr.Markdown(
value="Available only for verified users.[Click here for verification.](/verify_page)"
)
en_inputs = gr.Textbox(
label="Input",
lines=5,
value=DEFAULT_EN_TEXT,
placeholder=DEFAULT_EN_TEXT,
)
# en_lang = gr.Textbox(value="en", visible=False)
en_lang = gr.Radio(["en", "ru"], value="en", label="Language")
en_outputs = gr.Textbox(
label="Output",
lines=5,
placeholder="Summary and Sentiment would be here...",
)
en_inbtn = gr.Button("Proceed")
en_inbtn = en_inbtn.click(
get_summary,
[en_inputs, en_lang],
[en_outputs],
)
# demo.launch(server_name="127.0.0.1", server_port=8080, share=False)
# mounting at the root path
app = gr.mount_gradio_app(app, demo, path="/index")
|