Moved code to subfolder

backend/Dockerfile

@@ -0,0 +1,12 @@
+FROM python:3.8
+WORKDIR /app
+
+ADD requirements.txt /app/requirements.txt
+
+RUN pip install --upgrade -r requirements.txt
+
+EXPOSE 8080
+
+COPY ./ /app
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8080"]

backend/app.py

@@ -0,0 +1,29 @@
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
+from config.config import initiate_database
+from routes.request import router as RequestRouter
+from routes.admin import router as AdminRouter
+from config import Settings
+
+app = FastAPI()
+
+@app.on_event("startup")
+async def start_database():
+    await initiate_database()
+
+origins = ["*"]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
+app.include_router(RequestRouter, tags=["Requests"])
+
+if Settings().ENV == "dev":
+    app.include_router(AdminRouter, tags=["Admin"])

backend/azure/__init__.py

@@ -0,0 +1 @@
+from .functions import translate

backend/azure/functions.py

@@ -0,0 +1,27 @@
+import requests
+import uuid
+from config import Settings
+
+
+def translate(data):
+    constructed_url = Settings().AZURE_ENDPOINT + '/translate'
+
+    params = {
+        'api-version': '3.0',
+        'from': data.language_from,
+        'to': [data.language_to]
+    }
+
+    headers = {
+        'Ocp-Apim-Subscription-Key': Settings().AZURE_KEY,
+        'Ocp-Apim-Subscription-Region': Settings().AZURE_LOCATION,
+        'Content-type': 'application/json',
+        'X-ClientTraceId': str(uuid.uuid4())
+    }
+
+    body = [{
+        'text': data.requested_text
+    }]
+
+    request = requests.post(constructed_url, params=params, headers=headers, json=body)
+    return request.json()

backend/config/__init__.py

@@ -0,0 +1 @@
+from .config import Settings

backend/config/config.py

@@ -0,0 +1,23 @@
+from beanie import init_beanie
+from motor.motor_asyncio import AsyncIOMotorClient
+from pydantic import BaseSettings
+from typing import Literal
+
+from models.request import Request
+
+
+class Settings(BaseSettings):
+    DATABASE_URL: str
+    AZURE_KEY: str
+    AZURE_ENDPOINT: str
+    AZURE_LOCATION: str
+    ENV: Literal["dev", "prod"]
+
+    class Config:
+        env_file = ".env"
+        orm_mode = True
+
+
+async def initiate_database():
+    client = AsyncIOMotorClient(Settings().DATABASE_URL)
+    await init_beanie(database=client.get_default_database(), document_models=[Request])

backend/database/database.py

@@ -0,0 +1,18 @@
+from typing import List
+from models.request import Request
+
+
+request_collection = Request
+
+
+async def add_request(item: Request) -> Request:
+    return await item.create()
+
+async def retrieve_requests() -> List[Request]:
+    return await request_collection.all().to_list()
+
+async def retrieve_request(text: str, language_from: str, language_to: str) -> List[Request]:
+    return await request_collection.find({"requested_text": text, "language_from": language_from, "language_to": language_to}).to_list()
+
+async def clear_requests() -> None:
+    return await request_collection.delete_all()

backend/models/request.py

@@ -0,0 +1,40 @@
+from beanie import Document
+from pydantic import BaseModel
+
+
+class Request(Document):
+    id: str
+    requested_text: str
+    response_text: str
+    language_from: str
+    language_to: str
+    unixdate: int
+
+    class Settings:
+        name = "request"
+
+    class Config:
+        schema_extra = {
+            "example": {
+                "id": "9e90b529-05d0-4b0e-a8f8-de17e72e553c",
+                "requested_text": "I would really like to drive your car around the block a few times!",
+                "response_text": "Ich würde wirklich gerne mit Ihrem Auto ein paar Mal um den Block fahren!",
+                "language_from": "en",
+                "language_to": "de",
+                "date": 1678225669
+            }
+        }
+
+class RequestData(BaseModel):
+    requested_text: str
+    language_from: str
+    language_to: str
+
+    class Config:
+        schema_extra = {
+            "example": {
+                "requested_text": "I would really like to drive your car around the block a few times!",
+                "language_from": "en",
+                "language_to": "de",
+            }
+        }

backend/requirements.txt

@@ -0,0 +1,10 @@
+requests==2.28.2
+uuid==1.30
+motor==3.1.1
+uvicorn==0.20.0
+pymongo==4.3.3
+fastapi==0.93.0
+pydantic==1.10.5
+passlib==1.7.4
+beanie==1.17.0
+python-dotenv==1.0.0

backend/routes/admin.py

@@ -0,0 +1,14 @@
+from fastapi import APIRouter
+
+from database.database import *
+from models.request import *
+
+router = APIRouter()
+
+@router.post("/cleardb", response_description="Clears database cache")
+async def cleardb():
+    await clear_requests()
+
+@router.get("/dbcache", response_description="Show database contents", response_model=List[Request])
+async def showcache():
+    return await retrieve_requests()

backend/routes/request.py

@@ -0,0 +1,37 @@
+from fastapi import APIRouter
+import uuid
+import time
+
+from azure import translate as azure_translate
+from database.database import *
+from models.request import *
+
+router = APIRouter()
+
+
+@router.post("/translate", response_description="Translate text", response_model=Request)
+async def translate(data: RequestData):
+    # Check if this request is saved in the database
+    check_db = await retrieve_request(data.requested_text, data.language_from, data.language_to)
+
+    if len(check_db) == 0: # Not in database, use azure api
+        print("Not in database, use Azure api")
+        # Make api request
+        response = azure_translate(data)
+        print(response)
+
+        # Create object
+        item = Request(
+            id=str(uuid.uuid4()),
+            requested_text=data.requested_text,
+            response_text=response[0]['translations'][0]['text'],
+            language_from=data.language_from,
+            language_to=data.language_to,
+            unixdate=int(time.time())
+        )
+
+        # Save it to the database and return it
+        return await add_request(item=item)
+    else:
+        print("Already in database, serving the cached result")
+        return check_db[0]