Correctly overwrite API auth files

main.py

@@ -1,7 +1,30 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Header, HTTPException, Depends
+from sqlalchemy.orm import Session
+import datetime
+import uuid
+from database import SessionLocal, APIKey
 
 app = FastAPI()
 
-@app.get("/")
-def read_root():
-    return {"Hello": "World"}
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+@app.get('/get-api-key')
+def get_api_key(db: Session = Depends(get_db)):
+    new_key = str(uuid.uuid4())
+    expires_at = datetime.datetime.utcnow() + datetime.timedelta(hours=24)
+    db_key = APIKey(key=new_key, expires_at=expires_at)
+    db.add(db_key)
+    db.commit()
+    return {'api_key': new_key, 'expires_at': expires_at}
+
+@app.get('/hello')
+def read_hello(x_api_key: str = Header(...), db: Session = Depends(get_db)):
+    db_key = db.query(APIKey).filter(APIKey.key == x_api_key).first()
+    if not db_key or db_key.expires_at < datetime.datetime.utcnow():
+        raise HTTPException(status_code=403, detail='Invalid or expired API Key')
+    return {'message': 'Hello World'}