4g( j d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl m
Z
m Z m Z m
Z
mZmZmZmZmZmZ d dlmZ d dlmZ d dlmZ d dlmZ d d l m Z m Z d d
l m Z m!Z!m"Z" e e
Z#e#jI ed gd d gd g
i Z% e jL d Z' e jL d Z( e jL d Z)e(re)re's e*d e e( Z+ G d de Z,de-de,fdZ. G d de Z/ ed fde-fdZ0de-fd Z1d e,d e-d e-fd Z2e#jg d! d"e/fd# Z4e#jg d$ ed%d&' ed%d(' ed%d)' fd e-d ee- d ee- fd* Z5e#jm d+ d,e de-fd- Z7e#jg d. e e0 ed% fde-d/e
fd0 Z8y)1 N)uuid4)ListOptional)
FastAPI WebSocketDepends
UploadFileFile
HTTPExceptionHeaderstatusWebSocketDisconnectBody) HTMLResponse)CORSMiddleware) BaseModel) load_dotenv)PineconeServerlessSpec)extract_text_from_pdfextract_text_from_fileextract_text_from_image*T)
allow_originsallow_credentials
allow_methods
allow_headersOPENAI_API_KEYPINECONE_API_KEYPINECONE_ENVIRONMENTzBOPENAI_API_KEY, PINECONE_API_KEY, or PINECONE_ENVIRONMENT not set.)api_keyc @ e Zd ZU eed< g Zee ed< G d d Zy)UserDatausername chat_historyc e Zd ZdZy)UserData.ConfigTN)__name__
__module__ __qualname__arbitrary_types_allowed /opt/aibudy/backend/main.pyConfigr' / s "& r- r/ N)r( r) r* str__annotations__r% r r/ r, r- r. r# r# + s M L$s) ' 'r- r# r$ returnc J | t vrt | t | < t | S )Nr$ )usersr# r4 s r. get_userr6 2 s" u "H5h ?r- c e Zd ZU eed< y) LoginRequestr$ N)r( r) r* r0 r1 r, r- r. r8 r8 7 s Mr- r8
x_usernamec H K | st t j d | S w)NzInvalid username status_codedetail)r r
HTTP_401_UNAUTHORIZED)r9 s r. get_current_usernamer? : s* 44%
s "c d| }t j j }||vr6t d| t j |ddt dt t j | S )z,Get or create a Pinecone index for the user.z user-index-z&Creating new Pinecone index for user: i euclideanaws)cloudregion)name dimensionmetricspec)pc list_indexesnamesprint create_indexr r Index)r$
index_name
index_lists r. get_or_create_user_indexrQ B sx xj)J "((*J#
6xjAB
+
88J r- userpayloadURIc t | j }t j j d|gddi }t t
}| j |d}|j ||d j |fg t d| j d | y # t $ r }t d
| t d d }~ww xY w) Nmultilingual-e5-large
input_typepassage)inputs
parameters)rR rT r )vectorszVector upserted for user z with URI: "Error updating user vector store: 7An error occurred while updating the user vector store.) rQ r$ rI inferenceembedr0 r upsertvaluesrL Exception)rR rS rT index
embeddings vector_idmetadataes r. update_user_vector_storerh X s S( 7 \\'' #9 i (
L MM
y*Q-*>*> I J K
)$-- C5IJ S
21#67QRRSs BB C&B??Cz/login
login_requestc x t dt j | j }t | dd| diS )Nzpython version: messagezUser z logged in.)rL sysversionr$ r6 )ri r$ s r. loginrn z s= S[[M
*+ %%H X xj 4 55r- z/consume_browser.test)examplezYour payload data herezhttp://example.com/your-uric K t | }t ||| ddiS # t $ r }t d| t dd d }~ww xY ww)Nrk zUser vector updated.r\ r] r; )r6 rh rb rL r )rR rS rT user_objrg s r. consume_browserrt s\ oD> 7C8122 o
21#674mnnos$ A A A AA A z/ws/{username} websocketc K | j d { t | } | j d { }|dk( r| j d { y t |j
} t j j d|gddi } |j |d j d d
d }g }|d D ]& } |j d| j ddh i ( d}
|D ]8 }|d d j! d r|
d|d d z dz z
}
|
d|d d z z
}
: | j# |
d { 7 17 7 # t $ r }t d| t dd
d }~ww xY w# t $ r }t d| t dd
d }~ww xY w7 f# t$ $ r t d Y y w xY w# t $ r }t d| t dd
d }~ww xY ww) NTcloserV rW query)modelrY rZ zError embedding data: rr z+An error occurred while embedding the data.r; r F)vectortop_kinclude_valuesinclude_metadataz Error querying Pinecone index: z4An error occurred while querying the Pinecone index.matchesrf rT httpz3
z
zWebSocket connection closedzGENERAL PINECONE FAIL z5An error occurred during the WebSocket communication.)acceptr6 receive_textrw rQ r$ rI r^ r_ rb rL r rx ra appendget
startswith send_textr ) ru r$ rR datarc xrg resultsserializable_resultsmatch
return_strings r. websocket_endpointr sY
H DAm;
& 3 3 55 7?#//+++ 1 ? o ** 5 $v ('$ + A
x #kk t{{ ',)- * G (*$ $Y / E(// #EIIj5'$B1 !#
- DA}U+66v>%+^abcmanotau)ux|)||
!tq}U/C%CCMD ))-888q 6 , ! o 21#67'C@mnno ! x ;A3?@'C@vwwx, 9&
34
m
&qc*+4kllms G/EG/G F) E
F) E
F) G/ F) 5%E "E; =BF) F'F) G G/
F)
F) E8E33E88F) ; F$F F$$F) )G =G >G/?G G G, G''G,,G/z/uploadfilec K t | } |j d { }t | dkD r
t dd dt
}t |d 5 }|j | d d d h d
}|j }|j d d j } | |vr
t dd
| dv r t | d { }
n-| dv r t | d { }
n t | d { }
t j | t# ||
| ddiS 7 # t $ r }t d| t dd d }~ww xY w# 1 sw Y xY w7 # t $ r }t d| t dd d }~ww xY w7 # t $ r }t d| t dd d }~ww xY w7 # t $ r }t d| t dd d }~ww xY w# t $ r }t d| t dd d }~ww xY w# t $ r#}t d| t dd| d }~ww xY ww)Ni P i z#File size exceeds the limit of 5MB.r; zError reading the file: rr z)An error occurred while reading the file.z/tmp/wb> jpgpdfpngtxtdocxjpeg.zUnsupported file type.)r r r z&Error extracitng text from the image: z7An error occurred while extracting text from the image.)r z!Error extracting text from file: z6An error occurred while extracting text from the file.r\ r] rk z)File uploaded and processed successfully.zError processing file: )r6 readlenr rb rL r openwritefilenamesplitlowerr r r osremoverh ) r$ r rs file_to_tmprg filename_tmpfallowed_extensionsr extensionextracted_texts r. upload_filer s
:SH% e $ +K; /1#|'L!L '
!
v'<\'J!J
v'=l'K!K , s $X~x H
FGGc , e ,QC0 1 C8cd d e
! ! "M
w>qcBC###<G H
G, #H &I'D* * E3EEH E H "E$ $ F
-FF
H F F;F66F;;H >G G) G$$G))H , H5HHH I! H??II)9r rl jsonhtmluuidr typingr r fastapir r r r r
r r r
r r fastapi.responsesr fastapi.middleware.corsr pydanticr dotenvr pineconer r
extract_toolsr r r appadd_middlewarer5 getenvr r r
ValueErrorrI r# r0 r6 r8 r? rQ rh postrn rt ru r r r, r- r. r s:
! + 2 - ` `
i %%% +, 299 /0 ryy!78 3>
Y
ZZ &''y 's x
9 28 3 s , S8 Sc S SD (6 6 6
S&) !#/G H c+HI
o
o
c]
o
#
o
o Em EmS Em !EmN ) 01Cy>S>S
>S >Sr-