o
2åifº ã @ sT d dl Z d dlZd dlZej ej e¡¡ZG dd„ dƒZG dd„ dƒZ dd„ Z
dS )é Nc @ s4 e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd S ) ÚDocIDsc C s( | d¡| _| d¡| _| | j¡ d S )Nz doc_ids.jsonzall_files.json)ÚloadFileÚdoc_idsÚ
sugar_doc_idsÚ formatDocIDs)Úself© r ú+/home/shebes/code/rag_eval/src/apis/resp.pyÚ__init__ s zDocIDs.__init__c C sH t j t|¡}t j |¡r t|dƒ}t |¡}| ¡ |S t d
|¡ƒ‚)NÚrbz
{0} -- File does not exist
) ÚosÚpathÚjoinÚbasedirÚisfileÚopenÚjsonÚloadÚcloseÚ ExceptionÚformat)r Ú file_nameÚ file_pathÚfÚresultr r r r s
zDocIDs.loadFilec C sP | ¡ D ]!\}}t|tƒr%|D ]}| |d ¡}|d }| j ||i¡ qqd S )NÚfilenameÚid)ÚitemsÚ
isinstanceÚlistÚformatDocumentNamer Úupdate)r ÚvaluesÚ_keyÚ_valÚ theObjectÚdocNameÚdocIDr r r r s
€ûzDocIDs.formatDocIDsc C s$ | d¡}| d¡}| dd¡}|S )Nz.pdfz.txtÚ_Ú )Ú removesuffixÚreplace)r Úvalr r r r r & s
zDocIDs.formatDocumentNamec C s || j ¡ v r | j | S |S ©N)r Úkeys)r Údoc_idr r r ÚgetInfo- s
zDocIDs.getInfoN)Ú__name__Ú
__module__Ú __qualname__r
r r r r0 r r r r r s
r c @ s e Zd Zdd„ Zdd„ ZdS )Ú RespContextc C s || _ i | _d S r- )ÚcntxtÚtheObj)r Úcontextr r r r
4 s
zRespContext.__init__c C sV | ¡ D ]$\}}t|tƒr| |¡ nt|tƒs t|tƒr"|| j|< | j| j|< qd S r- )r r r ÚparseIntoObjectÚstrÚintr6 r, )r ÚinfoÚkeyr, r r r r8 8 s
ûzRespContext.parseIntoObjectN)r1 r2 r3 r
r8 r r r r r4 3 s r4 c C s< d}d}| ¡ D ]\}}d}|D ]
}|d|d 7 }qq|S )NÚ r ú Útext)r )Ú
theContextr Úndxr' Ú
summaryLstÚcomboTxtÚentryr r r ÚcreateAccordionFromJsonB s ÿrE ) Ústringr r r
ÚabspathÚdirnameÚ__file__r r r4 rE r r r r Ú<module> s (
