Add application file

app.py

@@ -20,6 +20,7 @@ from passlib.context import CryptContext
 
 app = FastAPI()
 
+
 # Set environment variable for Hugging Face cache directory
 os.environ["HF_HOME"] = "/app/.cache"
 
@@ -30,8 +31,8 @@ tts_kwargs = {
 
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
 
-class User(BaseModel):
-    username: str
+class Auth(BaseModel):
+    api_key: str
     password: str
 
 # Password hashing context
@@ -43,52 +44,45 @@ def get_password_hash(password):
 def verify_password(plain_password, hashed_password):
     return pwd_context.verify(plain_password, hashed_password)
 
-# Load secret key from Hugging Face secrets
-username = os.getenv("XCHE_SECRET_AUTH")
-password = os.getenv("XCHE_SECRET_PASSWORD")
-
-
-print(username)
+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_key = get_api_key(fake_db, api_key)
+    if not api_key:
         return False
-    if not verify_password(password, user.password):
+    if not verify_password(password, api_key.password):
         return False
-    return user
+    return api_key
 
 @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_key = authenticate(fake_data_db, form_data.api_key, form_data.password)
+    if not api_key:
         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_key.api_key, "token_type": "bearer"}
 
 def check_api_token(token: str = Depends(oauth2_scheme)):
-    user = get_user(fake_users_db, token)
-    if not user:
-        raise HTTPException(status_code=403, detail="Invalid or missing API Key")
-    return user
+    api_key = get_api_key(fake_data_db, token)
+    if not api_key:
+        raise HTTPException(status_code=403, detail="Invalid or missing API KEY")
+    return api_key
 
 def trim_memory():
     libc = ctypes.CDLL("libc.so.6")
@@ -103,9 +97,7 @@ def init_models():
     return models
 
 @app.post("/create_audio")
-async def tts(request: str, user: User = Depends(check_api_token)):
-
-    print(request)
+async def tts(request: str, api_key: Auth = Depends(check_api_token)):
 
     accented_text = accentification(request)
     plussed_text = stress_replace_and_shift(accented_text)