Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -57,11 +57,13 @@ def generate(item: Item):
|
|
| 57 |
async def generate_text(item: Item):
|
| 58 |
# Reject if not authenticated
|
| 59 |
apiKey = os.environ.get("API_KEY")
|
| 60 |
-
|
| 61 |
if not auth_header.startswith('Bearer ' + apiKey):
|
| 62 |
-
return jsonify({
|
|
|
|
|
|
|
|
|
|
| 63 |
|
| 64 |
return {
|
| 65 |
-
"response": generate(item)
|
| 66 |
-
"token": auth_header
|
| 67 |
}
|
|
|
|
| 57 |
async def generate_text(item: Item):
|
| 58 |
# Reject if not authenticated
|
| 59 |
apiKey = os.environ.get("API_KEY")
|
| 60 |
+
authHeader = requests.headers['Authorization']
|
| 61 |
if not auth_header.startswith('Bearer ' + apiKey):
|
| 62 |
+
return jsonify({
|
| 63 |
+
"message": "Invalid or missing Bearer token",
|
| 64 |
+
"token": authHeader
|
| 65 |
+
}), 401
|
| 66 |
|
| 67 |
return {
|
| 68 |
+
"response": generate(item)
|
|
|
|
| 69 |
}
|