Mihaiii commited on
Commit
115c7b7
·
verified ·
1 Parent(s): ea4da82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -18,14 +18,17 @@ def welcome():
18
  </div>
19
  """
20
 
21
- @app.post('/set_log_level')
22
- def set_log_level():
 
 
 
23
  try:
24
  logger = logging.getLogger()
25
- level = request.json.get('level', '').upper()
26
  if level in ['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL']:
27
  logger.setLevel(level)
28
- return f"Log level changed to {level}", 200
29
- return "Invalid log level", 400
30
  except Exception as e:
31
- return str(e), 500
 
18
  </div>
19
  """
20
 
21
+ class LogLevelRequest(BaseModel):
22
+ level: str
23
+
24
+ @app.post("/set_log_level")
25
+ async def set_log_level(request: LogLevelRequest):
26
  try:
27
  logger = logging.getLogger()
28
+ level = request.level.upper()
29
  if level in ['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL']:
30
  logger.setLevel(level)
31
+ return {"message": f"Log level changed to {level}"}
32
+ raise HTTPException(status_code=400, detail="Invalid log level")
33
  except Exception as e:
34
+ raise HTTPException(status_code=500, detail=str(e))