o Zph@@sddlmZmZddlmZddlmZmZmZm Z ddl Z ddl Z ddl m Z ddlmZddlmZddlmZe je jd e eZed d d ZGd ddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZ GdddeZ!GdddeZ"Gdd d eZ#Gd!d"d"eZ$Gd#d$d$Z%e%Z&ej'd%ed&d'efd(d)Z(ej'd*e"d&d'e fd+d,Z)e*d-d.d/Z+e*d0d1d2Z,d3d4d5d6d7d8d6d9d:d6d;dd6d?d@d6gdAZ-dBdCd@dDdEdFdGdHgdIdJdKdLdMdNdLdOdPdLdQdEdLdRdSdLgdTdUdUdVgdWdXdYdYdZd[d\gdWd]d^d_d`gdadWgdbZ.edckrIddl/Z/e/j0edddedfdSdS)g)FastAPI HTTPException) BaseModel)ListDictOptionalAnyN)asdict) TNASystem)SkillAnalysisSystem)CourseFieldIdentifier)levelTNA System API1.0.0)titleversionc@eZdZUeed<eed<dS) DepartmentidnameN__name__ __module__ __qualname__int__annotations__strrrJ/var/www/eduai.edurigo.com/doc_train/edurigo_ai/Puru/puru_test_fast_api.pyr  rc@s"eZdZUeed<eeed<dS)DepartmentRequest user_promptorganization_departmentsN)rrrrrrrrrrrr s r c@r)DepartmentResponseidentified_department department_idN)rrrrrrrrrrr#rr#c@s>eZdZUeed<eed<dZeeed<dZeeed<dS) FilteredSkillskill_id skill_nameN skill_level skill_count) rrrrrrr)rr*rrrrr&!s r&c@r) AllSkillDatar'r(Nrrrrrr+'rr+c@s:eZdZUeed<eed<eed<eed<eeed<dS) OfflineCourse course_id course_nameshort_description descriptionskillsN)rrrrrrrrrrr,+s r,c@sReZdZUeed<eed<eed<eeed<eed<eeed<ee ed<dS) SkillsCourseRequest user_query admin_namedepartment_namefiltered_skillstotal_users_in_departmentall_skills_dataoffline_coursesN) rrrrrrr&rr+r,rrrrr22s   r2c@s.eZdZUeed<eed<eed<eed<dS)CourseRecommendationr-rreason percentageN)rrrrrfloatrrrrr:;s  r:c@2eZdZUeeefed<eeed<eed<dS)SkillsCourseResponseskill_analysiscourses_offlinestrategic_presentationN rrrrrrrrr:rrrrr?A   r?c@r>)!SkillsCourseResponseWithStrategicr@rArBNrCrrrrrEErDrEc@s:eZdZUeeefed<eeed<dZ e eed<dS)SkillsCourseResponseOptionsr@rANrB) rrrrrrrrr:rBrrrrrrFJs  rFc @seZdZdZddZdedefddZdee d e d e dee fd d Z d e dee de fddZd e de dede fddZdS)TNAIntegratedSystemzFIntegrated TNA System combining department, skill, and course analysiscCs&t|_t|_t|_tddS)zInitialize all three systemsz!TNA Integrated System initializedN)r department_systemr skill_systemr course_systemloggerinfo)selfrrr__init__SszTNAIntegratedSystem.__init__requestreturnc Csz|j|j|jdd|jDdd|jDd}|jt|}|d}|d}t d| dd d g}| d d ry| d ry|d dd|j Dd}|j t|}t d| dd d| d d ry||d|| dd}t| d d| dd | dg| dd| dd| dd| d d d||dWSty} zt d| tdd t| d!d"} ~ ww)#a> Process skill analysis and course recommendations with strategic presentation Args: request: SkillsCourseRequest with all necessary data Returns: SkillsCourseResponse with skill analysis, course recommendations, and strategic presentation cSs"g|] }|j|j|j|jdqS)r'r(r)r*rQ.0skillrrr jszBTNAIntegratedSystem.process_skills_and_courses..cSg|] }|j|jdqS)r'r(rWrRrrrrUss )r3r4r5r6r8 raw_analysisrBzSkill analysis completed: total_skills_recommendedrz skillssuccessFidentified_fieldc Ss6g|]}|j|j|j|j|jrd|jnddqS),)r-r.course_short_descriptioncourse_descriptionrT)r-r.r/r0r1join)rScourserrrrUs)r[courseszCourse analysis: total_matching_coursesz courses matchedmatching_coursesfiltering_reasoningr]Unknownrecommended_skillsanalysis_summary case_applieddepartment_coverageg)r[rYrgrhrirjrZ)r@rArBz(Error in skills and courses processing: z&Skills and courses processing failed:  status_codedetailN)r3r4r5r6r8rIprocess_with_both_formatsjsondumpsrKrLgetr9rJprocess_json_request _generate_course_recommendationsr? Exceptionerrorrr) rMrO skill_request both_formats skill_resultrBrAcourse_request course_resulterrrprocess_skills_and_coursesZsf            z.TNAIntegratedSystem.process_skills_and_coursesrdry reasoningc Csg}|dd}|dg}t|D]5\}}td|dd} |||} t| | d} |||| dk} |t|d |d | t| d d q|j d ddd|S)ac Generate course recommendations with reasons and percentages Args: matching_courses: List of courses that matched the field skill_result: Skill analysis result reasoning: LLM reasoning for course selection Returns: List of CourseRecommendation objects r[rfrg_<brr-r.)r-rr;r<cSs|jS)N)r<)xrrrszFTNAIntegratedSystem._generate_course_recommendations..T)keyreverse) rr enumeratemax_calculate_skill_match_bonusmin_generate_course_reasonappendr:roundsort) rMrdryr~recommendationsr[rgirabase_percentageskill_match_bonusfinal_percentager;rrrrts     z4TNAIntegratedSystem._generate_course_recommendationsrargcCs|sdS|ddd}dd|D}|sdSd}|D]}|dd}|D]}||vs4||vr:|d7}nq*qt|d d S) z1Calculate bonus percentage based on skill matchesrrTr]r\cSsg|] }|r|qSr)striprRrrrrUszDTNAIntegratedSystem._calculate_skill_match_bonus..r(r )rrlowersplitr)rMrarg course_skillsmatches skill_recr( course_skillrrrrs z0TNAIntegratedSystem._calculate_skill_match_bonusr[has_skill_matchcCs4|dd}|rd|d|dSd|d|dS) z8Generate explanation for why this course was recommendedr.zUnknown CoursezDirectly relevant to z1 field with matching skill requirements. Course 'zB' covers essential technologies and concepts needed for your goal.z Relevant to z8 field based on course content and description. Course 'zI' provides foundational knowledge that supports your learning objectives.)rr)rMrar[rr.rrrrs z+TNAIntegratedSystem._generate_course_reasonN)rrr__doc__rNr2r?r}rrrr:rtr=rboolrrrrrrGPs ^ *rGz/analyze-department)response_modelrOc sz\td|j|jdd|jDd}tjt|}td|| ddrA| ddrA| d d }| d d }n| d d }d }t ||d}td|d|d|WSt yy}zt d|t dt|dd}~ww)zy Department Analysis endpoint - FIRST CALL Identifies the most relevant department based on user prompt z$Processing department analysis for: cSrV)rrr)rSdeptrrrrUs z&analyze_department..r!r"zDepartment analysis result: rZFmatchedr5rfr%rinferred_department)r$r%zDepartment analysis completed: z (ID: )zDepartment analysis failed: rkrlN)rKrLr!r" tna_systemrHrsrprqrrr#rurvrr)rOdepartment_requestdepartment_resultr$r%resultr|rrranalyze_departments8   rz/analyze-skills-coursesc shztd|jt|}td|WSty3}ztd|tdt|dd}~ww)z Skills and Courses Analysis endpoint - SECOND CALL Processes skill analysis and course recommendations based on department context z,Processing skills and courses analysis for: z2Skills and courses analysis completed successfullyz$Skills and courses analysis failed: rkrlN) rKrLr3rr}rurvrr)rOrr|rrranalyze_skills_courses5s  rz/healthcs dddS)zHealth check endpointhealthyzTNA System API is running)statusmessagerrrrr health_checkFs r/csddddddddd S) z"Root endpoint with API informationrruITwo-step process: 1) Department Analysis → 2) Skills & Courses Analysisz@POST /analyze-department - FIRST CALL: Department identificationzVPOST /analyze-skills-courses - SECOND CALL: Skills analysis and course recommendationszGET /health - Health checkzGET /docs - API documentation)zanalyze-departmentzanalyze-skills-courseshealthdocs)rrflow endpointsrrrrrrootKsrz&Want to improve my teams sales ability OperationsrHRFinancezProduct Managementsales Developerrz$Want to improve my technical abilityAscentxPython IntermediaterQri_zcommunication skillrWi`ztechnical skillsy JavaScriptz{Angular32zGeneral Knowledger])r-r.r/r0r143z)Python Programming for Complete BeginnerszLearn Python from scratchpython programming63z Angular BasiczAngular framework coursezComplete Angular course)angular typescript javascript)r3r4r5r6r7r8r9__main__z0.0.0.0i@)hostport)1fastapirrpydanticrtypingrrrrrplogging dataclassesr tna_depr tna_skill_secondr tna_course_trialr basicConfigINFO getLoggerrrKapprr r#r&r+r,r2r:r?rErFrGrpostrrrrrrexample_department_requestexample_skills_courses_requestuvicornrunrrrrs        3  .     -