codebyam commited on
Commit
115622c
·
verified ·
1 Parent(s): d815f4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ History = []
11
  @app.get("/", response_class=HTMLResponse)
12
  async def read_root(response: Response):
13
  token = time.time()
14
- Tokens.append(token)
15
  response.set_cookie(key="token", value=token, httponly=True, samesite='strict') # Set cookie
16
  return '''<!DOCTYPE html>
17
  <html lang="en">
@@ -379,7 +379,7 @@ class ChatRequest(BaseModel):
379
  prompt: str
380
 
381
  @app.post("/response")
382
- async def handle_chat(chat_request: ChatRequest, token: int = Cookie(None)):
383
  if token in Tokens:
384
  i = Tokens.index(token)
385
  History[i].append({"role": "user", "content": chat_request.prompt})
 
11
  @app.get("/", response_class=HTMLResponse)
12
  async def read_root(response: Response):
13
  token = time.time()
14
+ Tokens.append(str(token))
15
  response.set_cookie(key="token", value=token, httponly=True, samesite='strict') # Set cookie
16
  return '''<!DOCTYPE html>
17
  <html lang="en">
 
379
  prompt: str
380
 
381
  @app.post("/response")
382
+ async def handle_chat(chat_request: ChatRequest, token: str = Cookie(None)):
383
  if token in Tokens:
384
  i = Tokens.index(token)
385
  History[i].append({"role": "user", "content": chat_request.prompt})