o g#l@sdZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZmZejd ddlZdd lmZmZmZmZmZdd lmZdd lm Z ddl!m"Z"m#Z#m$Z$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3eddgZ4eeedZ5e6e7e58dZ9e:Z;GdddZe>D]2Z?ddl@Z@e'jAe?dd ZBe@@ZCeefd?d@Z+dAdBZ,e-dCe.dDeefdEdFZ/dS)H PPTStructurerg333333?r" temperaturer#z:services/ppt_generator/content_slide_layout_choices_v5.txtNprojectclientcCs|jtvr td|j|jvrtdd|j||_||_ ||_ |j j dur1|j j n|j |_ |j rB|j j |jjdnd|_ t|j ddd }||_Wdn1s]wY|dS)zInitialize the service.z.public_status must be either public or privatezpitch_type must be , Nrzutf-8)encoding) public_status PUBLIC_STATUS ValueError pitch_typeadditional_instructionsjoinkeysr'get_main_company main_company client_configlayout_choicesLAYOUT_CHOICESstructure_recommendationgetopenreadcontent_slide_layout_choicesget_project_docs)selfr'r(filer@docswebsitesr@r@rAr=eszPPTStructure.get_project_docscCs |jS)z&Add or update the structure in the DB.)r'update_project_in_dbr>r@r@rAadd_or_update_project_in_dbss z(PPTStructure.add_or_update_project_in_dbreturnc Csgd}|jrd|j}|dd|f|jr*d|j}|dd|f|jr8|dd|jf|jjdkrD|dn|d |d t|}t t d }||j B|B}| |j j|j j|j jro|j jn|jd |j jr{|j jn|jd |j j|jd|jd|j j|jvr|j|j jnd|d }z dd|dD|j _Wnttfy|dd|j _Ynw|dS)zGet the structure of the PPT.))systemal You are an Investment Banking Managing Director creating a slide deck outline. The structure of the deck should be broken into sections which will contain different slides. Don't include an introduction section. Always include a company and market overview section. Each section should have multiple slides and a slide should only cover one topic. You want to ensure you have an executive overivew section to start and a financial overview at the end. {additional_instructions} )rUzYou will answer in json format which will be provided below. You are only being tasked to provide the structure (sections) of the PPT with the theme or content that should be included in each section.)humanzYou are pitching {company}, more information on the company is provided below, which is {public_status} in the {industry} and sector {sector} options for {pitch_type}.)rVzuCompany overview is below {company_overview} Company product and services overview is below {company_products} rVz2Here are the summaries of the uploaded documents: z'Here are the summaries of the website: z;Ensure the sections could cover the slides from this list: ib)rVz*You are going to limit your sections to 8.)rVz*You are going to limit your sections to 5.)rUz,You will answer in json format {json_format}pydantic_objectindustrysectoroverviewproducts) companyr,r[r\r/company_overviewcompany_productsr0 json_formatcSg|] }tdi|qSr@)r.0sectionr@r@rA sz.PPTStructure.get_structure..sections propertiesN)rHr1appendrJr8r5 client_typer from_messagesr r llm_model4invoker' company_namer,r[r4r\r/r0get_format_instructionsrjKeyErrorr.rS)r> llm_messagerHrJ chat_promptparserchainresponser@r@rA get_structurewsl            zPPTStructure.get_structurec sg}tjjD]G\}}dtdttdttffdd }|dkr,jjd|dng}|tjjdkrBjj|ddng}|t||||q tj |IdHj dS) z#Flesh out the structure of the PPT.rhprevious_sections next_sectionsc sztd|j|||IdHtd|jWdStyN}z(td|tjd|ddt|dddurCd|_WYd}~dSWYd}~dSd}~ww) zFlesh out the section.zfleshing sectionNzfleshed out sectionzError in fleshing out section: !Error in fleshing out section: %sTexc_infoslides)printtitleflesh_out_sectionAttributeError XCM_loggererrorgetattrr)rhrzr{erRr@rArs"  z:PPTStructure.flesh_out_sections..flesh_out_sectionrN) enumerater'rjrlistlenrlasyncio create_taskgatherrQ)r>tasksirhrrzr{r@rRrAflesh_out_sectionss,  zPPTStructure.flesh_out_sectionsrhrzr{c svgd}|rddd|D}|dd|df|r3ddd|D}|dd |df|jjd kr?|d n|d t|}ttd }tt dt d} || B|B} | |j j re|j j n|jd|j jrq|j jn|jd|j j|j j|jd|j|j|||d IdH} z dd| dD|_|WSttfy} ztd|j jd|j| d} ~ ww)z"Flesh out this section of the PPT.))rUzYou are an Investment Banking Managing Director who is an expert in the {industry}. Specifically, you are an expert in the {sector} sector.)rUzjYou are going to be given a section of a {pitch_type} presentation and you are going to help flesh it out.)rUz#One slide can only have one topic. )rUz.You will return a json object as {json_output})rVzTCompany you are working with is {company} with the products / services of {products})rVzThe section you are working on is: {section_title} and the content of the section should be: {section_content}. Don't overlap with other sections.)rVzYIn addition to your knowledge, you have the following documents to reference: {documents}r)cSg|]}|jqSr@rrfr@r@rAri.z2PPTStructure.flesh_out_section..rVz!The previous section titles are: z+. Don't overlap slides with other sections.cSrr@rrfr@r@rAri8rzThe next section titles are: rX)rVzTYou are going to limit your slides to 10 per section but you don't need all of them.)rVzWYou are going to limit your slides to 10 per section but you don't need to all of them.rYrr%r[r\r^) r[r\r/r`r^ section_titlesection_content documents json_outputNcSrdre)rrgslider@r@rAriosrz*Error in fleshing out section for project z , section )r1rlr5rmrrnr rr rrainvoker'r[r4r\r/rqrcontentfind_relevant_summaryrrrrsr.r project_id) r>rhrzr{prompt_messagesprevious_sections_strnext_sections_strpromptrvllm_rwrxexcr@r@rAr st          zPPTStructure.flesh_out_section sectionsLLMcsng}t|D]-\}}||r|||IdH}n||IdH}tdd|jDr/d|_||q|S)zWFlesh out new or updated sections and slides. Adds modification flags to track changes.Ncss|] }t|ddVqdS) isModifiedFNrrr@r@rA sz5PPTStructure.flesh_out_new_section..T)r_is_item_modified_process_modified_section_process_existing_sectionanyrrrl)r>rupdatedSectionsrrhr@r@rAflesh_out_new_sectionws  z"PPTStructure.flesh_out_new_sectionrDc sz:|dkr|jjd|ng}|t|jjdkr$|jj|ddng}|j|||dIdH}||d|_W|Stym}z&td|tj d|ddt |d ddurbd|_ WYd}~|SWYd}~|Sd}~ww) z!Process a new or updated section.rNr)rhrzr{TzError in fleshing out section:r|r}r) r'rjrr_clear_modification_flagsr Exceptionrrrrr)r>rhrDrzr{rr@r@rArs2    z&PPTStructure._process_modified_sectioncst|dds |Stfdd|jDdur|Sg}t|jD]\}}|r1|||dq ||IdH}|D]-}|ddurMd|j|d<q=|d |dkr^d |d_ |d|j|d<q=|S) z:Process existing section with potentially modified slides.rNc3s|]}|VqdS)N)rrrRr@rArsz9PPTStructure._process_existing_section..F)rDroutputrDrT) rrrrrrlflesh_out_slidesdictrr)r>rh slide_tasksrDr slide_taskr@rRrArs,   z&PPTStructure._process_existing_sectioncCst|ddp t|ddS)z!Check if slide is new or updated.isNewF isUpdatedr)r>itemr@r@rArszPPTStructure._is_item_modifiedcCs"dD] }t||rt||q|S)zFClear new/updated flags from an item but preserve modification status.)rr)hasattrdelattr)r>rflagr@r@rArs   z&PPTStructure._clear_modification_flagsrc s}t|}ttdttdtd}||BB}|fdd|DIdH}t |D]8\}} z td i||| d<Wq1t yi} zt j d| d j j jj| d d d| d<WYd} ~ q1d} ~ ww|S) z#Flesh out a batch of slides slides.rYg?)openai_api_keyr& model_namecs$g|]}|d|dqS)rrD)_get_slide_chain_inputs)rgrrvrhr>r@rAris z1PPTStructure.flesh_out_slides..NrzKError in fleshing out slide: %s in section %s in Project %s. Error was %srTr}r@)_build_slide_prompt_messagesrrnr rr rrabatchrrrrrr'r) r>rrhrrullmrwrxrrrr@rrArs>      zPPTStructure.flesh_out_slidescCs@dd|jjd|jjdfddg}|gd|d|S) z+Build prompt messages for slide generation.rUz,You are an investment banking expert in the z industry and the z sector.)rUzZYou are going to be provided with guidance on how to create a new slide in a presentation.)rVzThe section you are working on is: {section_title} and the content of the section is: {section_content}. Ensure this slide doesn't overlap with other slides in the section.))rVzYou are going to be adding a brand new slide which is titled: {slide_title} and should contain the following content: {slide_content})rVz6The slides before this one are titled: {slides_before})rVz4The slides after this one are titled: {slides_after})rVzFMake sure this slide doesn't overlap with other slides in the section.)rUz2You will respond in a json object as {json_output})r'r[r\extendrl)r>messagesr@r@rArsz)PPTStructure._build_slide_prompt_messagesrrvcCs|dkrdd|jd|Dng}|t|jdkr)dd|j|ddDng}|j|j|j|jd|d||dS) z*Get inputs for the slide generation chain.rcSrr@rrgsr@r@rAri'rz8PPTStructure._get_slide_chain_inputs..NrcSrr@rrr@r@rAri)rz; )rr slide_title slide_content slides_before slides_afterr)rrrrr1rr)r>rrhrDrvprevious_slides next_slidesr@r@rAr#s$z$PPTStructure._get_slide_chain_inputscCs4|jjd|jd|jdd}ddd|DS)zFind the relevant summary.zSection title: z and Section content: mmr)query search_typerWcSrr@) page_content)rgresultr@r@rAri@rz6PPTStructure.find_relevant_summary..)rLsearchrrr1)r>rhresultsr@r@rAr8s z"PPTStructure.find_relevant_summarycCst|j|j|j}|}|S)z"Create the PPT from the structure.)rr'r4r5convert_structure_to_ppt)r> ppt_creator deck_locationr@r@rA create_PPTBszPPTStructure.create_PPTcCst}t|jjj}||j|}||j|}tdi|}d|d<d|d<|j D]}|j dkr6|j |d<|j dkr@|j |d<q*|S)zGet the main company.r_r]r^Overview_StatsProductsNr@) rrr' company_urlurlget_itemcompany_info_table llm_tabler LLM_resultscategoryrx)r>dbrr4llm_info_itemr@r@rAr3Hs     zPPTStructure.get_main_companyi)ttlcCsHddlm}t}||j|jj}g}|D] }||dq||fS)z$Get the uploaded document summaries.rAttr doc_summary)boto3.dynamodb.conditionsrr query_items project_docsr'rrl)r>rrr doc_summariesdocr@r@rArG\s z-PPTStructure.get_uploaded_documents_summariescCsJddlm}t}|j|jd|jj|d|jjd}dd|D|fS)zGet the website summaries.rrzproject_id-indexr)table index_namesort_key_valueexpression_valuescSsg|]}|ddqS)summaryr_)r9)rgwebsiter@r@rArivsz6PPTStructure.get_website_summaries..)rrr query_index web_pagesr'req)r>rrrPr@r@rArIjs z"PPTStructure.get_website_summariescCs@|r|jjdd|Dd|r|jjdd|DddSdS)zCreate the vector store.cSs*g|]}t|d|d|dddqS)r file_namedoc_pathrrrmetadatar)rgrr@r@rAri|sz5PPTStructure._create_vector_store..)rcSs.g|]}t|dd|d|dddqS)rr_rrr)rr9)rgsiter@r@rAris N)rL add_documentsrNr@r@rArMxs  z!PPTStructure._create_vector_storecCsHtd|jjt|j|j|j}||td|jj|jS)z&Conduct research for the presentation.z'Conducting research for presentation %sz&Research for presentation %s completed) rinfor'rrr4r5research_presentation_update_project)r>rr@r@rArsz"PPTStructure.research_presentationsection_number slide_numbercCsH|jj|d}|j|d}t|j|j|j}|||_||S)zResearch the slide.r) r'rjrrr4r5conduct_researchresearchr)r>rrrhrrr@r@rAresearch_slides  zPPTStructure.research_slidecCstdi|j|_|jS)z)Update the project into the project classNr@)rr'rrRr@r@rArszPPTStructure._update_projectr new_sectionscCs tj|d}|durtddS)z"Check if there are any new updatesrNz&Project does not exist in the database)rcheck_project_in_dbr.)r>rrr'r@r@rA_check_new_updatess zPPTStructure._check_new_updates)NN)0__name__ __module__ __qualname____doc__r rrror llm_model3r7rr0rrrBr=rSrrryrrrintrrrrrrrrr rrrr3rrGrIrMrrr staticmethodstrrr@r@r@rAr$1sl$k) l((       r$__main__z0iti-digital-1b7107ab-9e73-4590-90ce-1ba6c6b6a791tequity)rr(rrr()r()Ir rjsonsysfaissdotenvrlangchain.promptsr&langchain_community.docstore.in_memoryr langchain_community.vectorstoresrlangchain_core.documentsrlangchain_core.output_parsersr langchain_openair r pathrlloggingconfigs.configr rrrr1services.company_profile.data_classes.llm_resultsr!services.ppt_generator.create_pptr8services.ppt_generator.data_classes.presentation_outlinerrrr+services.ppt_generator.data_classes.projectrrr)services.ppt_generator.presentation_typesrutils.client_checkrutils.dynamo_dbrutils.redis_cacherutils.url_parserrr-rK IndexFlatL2r embed_queryrD getLoggerrr$r projectsproject_detailstimerr'startget_client_config ppt_structurerendrr@r@r@rAsn