update new pages with verification

.gitignore

@@ -1,3 +1,4 @@
 __pycache__
 .venv
+.env
 flagged

app.py

@@ -46,7 +46,7 @@ DEFAULT_RU_TEXT = """В результате взрыва на заправке,
 """
 
 
-class Request(BaseModel):
+class TextRequest(BaseModel):
     text: str
 
 
@@ -121,7 +121,7 @@ class Summarizer:
         }
         return summary[lang], sentiment[lang]
 
-    def summarize(self, req: Request, lang: str = "en") -> Result:
+    def summarize(self, req: TextRequest, lang: str = "en") -> Result:
         sum_pipe, sent_pipe = self.get_pipe(lang)
         response_summary = sum_pipe(req.text)
         logger.info(response_summary)
@@ -134,9 +134,12 @@ class Summarizer:
         )
         return result
 
-    def summ(self, req: Request, lang: str = "en") -> str:
+    def summ(self, req: TextRequest, lang: str = "en") -> str:
         return self.summarize(req, lang).to_str()
 
+    def verified_summarize(self, req: TextRequest, lang: str = "en") -> str:
+        return "verified"
+
 
 if __name__ == "__main__":
     pipe = Summarizer()

main.py

@@ -1,6 +1,13 @@
+import os
 import gradio as gr
-from fastapi import FastAPI
-from app import Summarizer, Request, Result
+from pathlib import Path
+from fastapi import FastAPI, Request
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import HTMLResponse
+from fastapi.templating import Jinja2Templates
+from dotenv import load_dotenv
+
+from app import Summarizer, TextRequest, Result
 from app import (
     EN_SENTIMENT_MODEL,
     EN_SUMMARY_MODEL,
@@ -9,18 +16,58 @@ from app import (
 )
 from app import DEFAULT_EN_TEXT, DEFAULT_RU_TEXT
 
+load_dotenv()
+
+SITE_KEY = os.getenv("SITE_KEY")
+SECRET_KEY = os.getenv("SECRET_KEY")
+VERIFY_URL = "https://www.google.com/recaptcha/api/siteverify"
+
+# create FastAPI app
 app = FastAPI()
 pipe = Summarizer()
 
+# create a static directory to store the static files
+static_dir = Path("./static")
+static_dir.mkdir(parents=True, exist_ok=True)
+
+# mount FastAPI StaticFiles server
+app.mount("/static", StaticFiles(directory=static_dir), name="static")
+templates = Jinja2Templates(directory="templates")
+
+
+@app.get("/verify_page", response_class=HTMLResponse)
+async def verify_page(request: Request):
+    return templates.TemplateResponse(
+        request=request, name="verification.html", context={"site_key": SITE_KEY}
+    )
+
+
+@app.get("/bad_request", response_class=HTMLResponse)
+async def bad_request(request: Request):
+    return templates.TemplateResponse("bad_request.html", {"request": request})
+
+
+@app.post("/verify")
+async def verify(request: Request):
+    # verify_response = requests.post(
+    #     url=VERIFY_URL,
+    #     data={
+    #         "secret": SECRET_KEY,
+    #         "response": request.form["g-recaptcha-response"],
+    #     },
+    # )
+    # print(verify_response.json())
+    return templates.TemplateResponse("bad_request.html", {"request": request})
+
 
 @app.post("/summ_ru", response_model=Result)
-async def ru_summ_api(request: Request):
+async def ru_summ_api(request: TextRequest):
     results = pipe.summarize(request.text, lang="ru")
     return results
 
 
 @app.post("/summ_en", response_model=Result)
-async def en_summ_api(request: Request):
+async def en_summ_api(request: TextRequest):
     results = pipe.summarize(request.text, lang="en")
     return results
 

poetry.lock

@@ -723,14 +723,14 @@ testing = ["pytest (>=6)", "pytest-black (>=0.3.7)", "pytest-checkdocs (>=2.4)",
 
 [[package]]
 name = "jinja2"
-version = "3.1.2"
+version = "3.1.3"
 description = "A very fast and expressive template engine."
 category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
-    {file = "Jinja2-3.1.2-py3-none-any.whl", hash = "sha256:6088930bfe239f0e6710546ab9c19c9ef35e29792895fed6e6e31a023a182a61"},
-    {file = "Jinja2-3.1.2.tar.gz", hash = "sha256:31351a702a408a9e7595a8fc6150fc3f43bb6bf7e319770cbc0db9df9437e852"},
+    {file = "Jinja2-3.1.3-py3-none-any.whl", hash = "sha256:7d6d50dd97d52cbc355597bd845fabfbac3f551e1f99619e39a35ce8c370b5fa"},
+    {file = "Jinja2-3.1.3.tar.gz", hash = "sha256:ac8bd6544d4bb2c9792bf3a159e80bba8fda7f07e81bc3aed565432d5925ba90"},
 ]
 
 [package.dependencies]
@@ -1701,6 +1701,21 @@ files = [
 [package.dependencies]
 six = ">=1.5"
 
+[[package]]
+name = "python-dotenv"
+version = "1.0.1"
+description = "Read key-value pairs from a .env file and set them as environment variables"
+category = "main"
+optional = false
+python-versions = ">=3.8"
+files = [
+    {file = "python-dotenv-1.0.1.tar.gz", hash = "sha256:e324ee90a023d808f1959c46bcbc04446a10ced277783dc6ee09987c37ec10ca"},
+    {file = "python_dotenv-1.0.1-py3-none-any.whl", hash = "sha256:f7b63ef50f1b690dddf550d03497b66d609393b40b564ed0d674909a68ebf16a"},
+]
+
+[package.extras]
+cli = ["click (>=5.0)"]
+
 [[package]]
 name = "python-multipart"
 version = "0.0.6"
@@ -2835,4 +2850,4 @@ dev = ["black (>=19.3b0)", "pytest (>=4.6.2)"]
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10"
-content-hash = "5648f003dcb355b02c351874f25f204380d23b153110cec7f6177e81fd260dbf"
+content-hash = "e9de680b1e2f7d363af6cedea3cbb3873dddd31a084cffe71ff63cb146b7ed22"

pyproject.toml

@@ -18,6 +18,8 @@ protobuf = "^4.25.1"
 loguru = "^0.7.2"
 fastapi = "^0.109.0"
 uvicorn = "^0.27.0"
+python-dotenv = "^1.0.1"
+jinja2 = "^3.1.3"
 
 
 [tool.poetry.group.dev.dependencies]

templates/{home.html → bad_request.html}

@@ -2,10 +2,11 @@
 <html lang="en">
     <head>
         <meta charset="UTF-8"/>
+        <meta name="viewport" content="width=device-width, initial-scale=1.0"/>
         <link rel="stylesheet" href="{{ url_for('static',path='/css/style.css')}}"/> 
-        <title>Home Page</title>
+        <title>Bad request</title>
     </head>
     <body>
-        <h1>Welcome!</h1>
+        <h1>Sorry, your request not allowed</h1>
     </body>
 </html>

templates/error.html

@@ -1,11 +0,0 @@
-<!DOCTYPE html>
-<html lang="en">
-  <head>
-    <meta charset="UTF-8" />
-    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>Error</title>
-  </head>
-  <body>
-    <h2>{{error}}</h2>
-  </body>
-</html>

templates/verification.html

@@ -0,0 +1,50 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <meta http-equiv="X-UA-Compatible" content="ie=edge">
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap@5.2.2/dist/css/bootstrap.min.css"/>
+    <link rel="stylesheet" href="{{ url_for('static',path='/css/style.css')}}"/> 
+    <script async src="https://www.google.com/recaptcha/api.js"></script>
+    <script>
+      function onSubmit(token) {
+        document.getElementById("verify_form").submit();
+      }
+    </script>
+    <script>
+      function onClick() {
+        grecaptcha.ready(function() {
+          grecaptcha.execute('{{ site_key }}', {action: 'submit'}).then(function(token) {
+              // Add your logic to submit to your backend server here.
+          });
+        });
+      }
+    </script>
+    <title>Verification Page</title>
+  </head>
+  <body>
+    <div class="row justify-content-center col-md-4">
+      <div class="align-self-center">
+        <h1 class="text-primary">Verification Page</h1>
+        <form id="verify_form" action="{{ url_for('verify') }}" method="post">
+          <div class="form-group my-3">
+            <input type="text" class="form-control" name="captcha" placeholder="Type captcha">
+          </div>
+          <button type="submit" class="btn btn-primary">Submit</button>
+
+          <!-- reCaptcha v3 -->
+          <!-- <button type="submit" class="g-recaptcha btn btn-primary" 
+            data-sitekey="{{ site_key }}" data-callback="onSubmit" 
+            data-action="submit">Submit
+          </button> -->
+          <!-- reCaptcha v2 -->
+          <!-- <div class="g-recaptcha" data-sitekey="{{ site_key }}"></div> -->
+          <!-- <div id="html_element"></div> -->
+          <!-- <button type="submit" class="btn btn-primary" onclick='onClick()'>Submit</button> -->
+        </form>
+      </div>
+    </div>
+    <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.2.2/dist/css/bootstrap.min.css"></script>
+  </body>
+</html>