Added API KEY for api call verification

QuoteGenerator.py

@@ -26,7 +26,6 @@ class QuoteGenerator():
                        top_p:float=0.9, top_k:int=5):
         
         tags = self.preprocess_tags(tags)
-        print("Tags = ", tags)
         output = self.quote_generator(tags, min_length=min_length, max_length=max_length,
                                       temperature=1.0, top_k=5, top_p=top_p, early_stopping=True,
                                       num_beams=4)

main.py

@@ -1,24 +1,29 @@
-from fastapi import FastAPI, Request
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.middleware.trustedhost import TrustedHostMiddleware
+from fastapi import FastAPI, Request, Depends, HTTPException, status
+from fastapi.security import OAuth2PasswordBearer
 from QuoteGenerator import QuoteGenerator
 from typing import Union
 from pydantic import BaseModel
 import time
+import os
+
+# API to key to validate the Referer
+API_KEY = os.getenv('API_KEY')
+
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
+
+# Function to check of the incoming API call is from valid host or not
+def api_key_auth(api_key:str = Depends(oauth2_scheme)):
+    if api_key != API_KEY:
+        raise HTTPException(
+            status_code = status.HTTP_401_UNAUTHORIZED,
+            detail="Forbidden Access"
+        )
 
 class QuoteRequest(BaseModel):
     tags: Union[None, str]
 
 app = FastAPI()
 
-origins = ['quote-generator.streamlit.app']
-
-# Middleware to valiate host
-app.add_middleware(
-    TrustedHostMiddleware,
-    allowed_hosts=origins
-)
-
 #Middleware to note time
 @app.middleware("http")
 async def note_response_time(request: Request, call_next):
@@ -31,11 +36,11 @@ async def note_response_time(request: Request, call_next):
 quote_generator = QuoteGenerator()
 quote_generator.load_generator()
 
-@app.get("/")
+@app.post("/", dependencies=[Depends(api_key_auth)])
 def root(request: Request):
     return {"message": "This is the website for quote-generator"}
 
-@app.post("/generate_quote")
+@app.post("/generate_quote", dependencies=[Depends(api_key_auth)])
 def generate_quote(req: QuoteRequest):
     print("Tags: ", req.tags)
     generated_quote_oup = quote_generator.generate_quote(req.tags)