Spaces:
Sleeping
Sleeping
File size: 6,126 Bytes
d1a66a2 06f7537 da71e44 2f1cacb 10f0001 d1a66a2 edb525a b8cc90f 5bfb5e0 06f7537 d1a66a2 eb44025 d56d550 d51772c d56d550 eb44025 d56d550 5b59bf9 d56d550 eb44025 d56d550 eb44025 d56d550 eb44025 d56d550 5bfb5e0 d56d550 5bfb5e0 d56d550 5bfb5e0 d56d550 027633d d56d550 2f1cacb d56d550 da71e44 d1a66a2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 |
from fastapi import APIRouter , Request ,HTTPException , WebSocket
from controllers import websocket_controller as wc
from controllers import ws_controller as w
from controllers import ws_controller_1 as w1
from controllers import ner_ai_controller as ai
from services.chat_client_NER import ChatClient
import logging
import aiohttp
import time
from models import supabase_models as sp
router = APIRouter(prefix="/websockets")
#-----New Code -----#
@router.websocket("/ws")
async def get_data(websocket:WebSocket):
await websocket.accept()
# -------------#-------------#--------------
try:
json = await websocket.receive_json()
# json = js.loads(json)
apitoken = json['supabase_authorisation']
result = sp.AuthUser_Validator(apitoken)
print(result)
if result == "Exceptional error":
await websocket.send_text("user unautheticated!")
await websocket.close()
else:
start = time.time()
# try:
# # json = await websocket.receive_text()
# # json = js.loads(json)
# json = await websocket.receive_json()
# except Exception as e:
# logging.info(f"Error {e}")
# print(f"Error {e}")
# await websocket.send_text(f"There is some error !!! {e}")
# return
access_token = json['access_token']
logging.info(f"access_token:{access_token}")
user_query = json['brand_name'] if json.get('brand_name') is not None else None
logging.info(f"brand_name: {user_query}")
if access_token is None:
await websocket.send_text("Access Token Invalid OR NULL !!!")
websocket.close()
# access_token = await websocket.receive_text()
brand_name = None
logging.info(f"brand_name: {user_query}")
logging.info(f"access_token : {access_token}")
if user_query is not None:
chat = ChatClient().create(conversation=[])
response = chat.send_message(content=f"{user_query}", stream=False)
if response.text == 'others':
brand_name = None
else:
brand_name = response.text
await w.websocket_main(access_token ,websocket,start,brand_name)
except HTTPException as e:
print("Inside exception")
await websocket.send_text(f"user unautheticated!{e.detail}")
#-----#----
# @router.websocket("/ws")
# async def get_data(websocket:WebSocket):
# await websocket.accept()
# #-------------#-------------#--------------
# # try:
# # result = AuthUser_Validator(apitoken)
# # if result == "Exceptional error":
# # await websocket.send_text("user unautheticated!")
# # else:
# # start = time.time()
# # try:
# # # json = await websocket.receive_text()
# # # json = js.loads(json)
# # json = await websocket.receive_json()
# # except Exception as e:
# # logging.info(f"Error {e}")
# # print(f"Error {e}")
# # await websocket.send_text(f"There is some error !!! {e}")
# # return
# # access_token = json['access_token']
# # logging.info(f"access_token:{access_token}")
# # user_query = json['brand_name'] if json.get('brand_name') is not None else None
# # logging.info(f"brand_name: {user_query}")
# # if access_token is None:
# # await websocket.send_text("Access Token Invalid OR NULL !!!")
# # websocket.close()
# # # access_token = await websocket.receive_text()
# # brand_name = ""
# # logging.info(f"brand_name: f{user_query}")
# # logging.info(f"access_token : {access_token}")
# # if user_query is not None:
# # chat = ChatClient().create(conversation=[])
# # response = chat.send_message(content=f"{user_query}", stream=False)
# # if response.text == 'others':
# # brand_name = None
# # else:
# # brand_name = response.text
# # await w.websocket_main(access_token ,websocket,start,brand_name)
# # except HTTPException as e:
# # print("Inside exception")
# # await websocket.send_text(f"user unautheticated!{e.detail}")
# #----------------#----------#------------
# start = time.time()
# try:
# #json = await websocket.receive_text()
# #json = js.loads(json)
# json = await websocket.receive_json()
# except Exception as e:
# logging.info(f"Error {e}")
# print(f"Error {e}")
# await websocket.send_text(f"There is some error !!! {e}")
# return
# access_token = json['access_token']
# logging.info(f"access_token:{access_token}")
# user_query = json['brand_name'] if json.get('brand_name') is not None else None
# logging.info(f"brand_name: {user_query}")
# if access_token is None:
# await websocket.send_text("Access Token Invalid OR NULL !!!")
# websocket.close()
# # access_token = await websocket.receive_text()
# brand_name = ""
# logging.info(f"brand_name: f{user_query}")
# logging.info(f"access_token : {access_token}")
# if user_query is not None:
# chat = ChatClient().create(conversation=[])
# response = chat.send_message(content=f"{user_query}", stream=False)
# if response.text == 'others':
# brand_name = None
# else:
# brand_name = response.text
# await w.websocket_main(access_token ,websocket,start,brand_name)
# # await w1.websocket_main(access_token , websocket , start , brand_name)
|