Add application file

app.py

@@ -30,8 +30,8 @@ tts_kwargs = {
 
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
 
-class User(BaseModel):
-    username: str
+class Auth(BaseModel):
+    api_key: str
     password: str
 
 # Password hashing context
@@ -44,46 +44,46 @@ def verify_password(plain_password, hashed_password):
     return pwd_context.verify(plain_password, hashed_password)
 
 # Load username and password from environment variables
-username = os.getenv("XCHE_API_KEY")
+api_key = os.getenv("XCHE_API_KEY")
 password = os.getenv("XCHE_PASSWORD")
 
 # In-memory storage for simplicity; in production use a database
-fake_users_db = {
-    username: {
-        "username": username,
+fake_data_db = {
+    api_key: {
+        "api_key": api_key,
         "password": get_password_hash(password)  # Pre-hashed password
     }
 }
 
-def get_user(db, username: str):
-    if username in db:
-        user_dict = db[username]
-        return User(**user_dict)
+def get_api_key(db, api_key: str):
+    if api_key in db:
+        api_dict = db[api_key]
+        return Auth(**api_dict)
 
-def authenticate_user(fake_db, username: str, password: str):
-    user = get_user(fake_db, username)
-    if not user:
+def authenticate(fake_db, api_key: str, password: str):
+    api_data = get_api_key(fake_db, api_key)
+    if not api_data:
         return False
-    if not verify_password(password, user.password):
+    if not verify_password(password, api_data.password):
         return False
-    return user
+    return api_data
 
 @app.post("/token")
 async def login(form_data: OAuth2PasswordRequestForm = Depends()):
-    user = authenticate_user(fake_users_db, form_data.username, form_data.password)
-    if not user:
+    api_data = authenticate(fake_data_db, form_data.api_key, form_data.password)
+    if not api_data:
         raise HTTPException(
             status_code=400,
-            detail="Incorrect username or password",
+            detail="Incorrect API KEY or Password",
             headers={"WWW-Authenticate": "Bearer"},
         )
-    return {"access_token": user.username, "token_type": "bearer"}
+    return {"access_token": api_data.api_key, "token_type": "bearer"}
 
 def check_api_token(token: str = Depends(oauth2_scheme)):
-    user = get_user(fake_users_db, token)
-    if not user:
+    api_data = get_api_key(fake_data_db, token)
+    if not api_data:
         raise HTTPException(status_code=403, detail="Invalid or missing API Key")
-    return user
+    return api_data
 
 def trim_memory():
     libc = ctypes.CDLL("libc.so.6")
@@ -98,7 +98,7 @@ def init_models():
     return models
 
 @app.post("/create_audio")
-async def tts(request: str, user: User = Depends(check_api_token)):
+async def tts(request: str, api_data: Auth = Depends(check_api_token)):
 
     print(request)