Add application file

app.py

@@ -20,7 +20,6 @@ from passlib.context import CryptContext
 
 app = FastAPI()
 
-
 # Set environment variable for Hugging Face cache directory
 os.environ["HF_HOME"] = "/app/.cache"
 
@@ -31,8 +30,8 @@ tts_kwargs = {
 
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
 
-class Auth(BaseModel):
-    api_key: str
+class User(BaseModel):
+    username: str
     password: str
 
 # Password hashing context
@@ -44,45 +43,47 @@ def get_password_hash(password):
 def verify_password(plain_password, hashed_password):
     return pwd_context.verify(plain_password, hashed_password)
 
-api_key = os.getenv("XCHE_API_KEY")
+# Load username and password from environment variables
+username = os.getenv("XCHE_API_KEY")
 password = os.getenv("XCHE_PASSWORD")
 
-fake_data_db = {
-    api_key: {
-        "username": api_key,
+# In-memory storage for simplicity; in production use a database
+fake_users_db = {
+    username: {
+        "username": username,
         "password": get_password_hash(password)  # Pre-hashed password
     }
 }
 
-def get_api_key(db, api_key: str):
-    if api_key in db:
-        api_dict = db[api_key]
-        return Auth(**api_dict)
+def get_user(db, username: str):
+    if username in db:
+        user_dict = db[username]
+        return User(**user_dict)
 
-def authenticate(fake_db, api_key: str, password: str):
-    api_key = get_api_key(fake_db, api_key)
-    if not api_key:
+def authenticate_user(fake_db, username: str, password: str):
+    user = get_user(fake_db, username)
+    if not user:
         return False
-    if not verify_password(password, api_key.password):
+    if not verify_password(password, user.password):
         return False
-    return api_key
+    return user
 
 @app.post("/token")
 async def login(form_data: OAuth2PasswordRequestForm = Depends()):
-    api_key = authenticate(fake_data_db, form_data.username, form_data.password)
-    if not api_key:
+    user = authenticate_user(fake_users_db, form_data.username, form_data.password)
+    if not user:
         raise HTTPException(
             status_code=400,
-            detail="Incorrect API KEY or Password",
+            detail="Incorrect username or password",
             headers={"WWW-Authenticate": "Bearer"},
         )
-    return {"access_token": api_key.username, "token_type": "bearer"}
+    return {"access_token": user.username, "token_type": "bearer"}
 
 def check_api_token(token: str = Depends(oauth2_scheme)):
-    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
+    user = get_user(fake_users_db, token)
+    if not user:
+        raise HTTPException(status_code=403, detail="Invalid or missing API Key")
+    return user
 
 def trim_memory():
     libc = ctypes.CDLL("libc.so.6")
@@ -97,7 +98,9 @@ def init_models():
     return models
 
 @app.post("/create_audio")
-async def tts(request: str, api_key: Auth = Depends(check_api_token)):
+async def tts(request: str, user: User = Depends(check_api_token)):
+
+    print(request)
 
     accented_text = accentification(request)
     plussed_text = stress_replace_and_shift(accented_text)