o thD@sddlZddlZddlmZmZmZddlmZddlm Z ddl m Z ddl m Z ejejdeeZeGdd d ZeGd d d ZGd d d ZddZedkr\edSdS)N)DictListOptional) dataclass) ChatOllama)PromptTemplate) HumanMessage)levelc@sveZdZUdZeed<eed<eed<eed<eed<eed<eed<eed <eed <eeed <eed <eed <dS)OnlineCourseRecommendationz+Data class for online course recommendation course_nameplatform instructorduration skill_level descriptionurlpriceratingskills_coveredmatch_percentagerecommendation_reasonN)__name__ __module__ __qualname____doc__str__annotations__rintrrL/var/www/eduai.edurigo.com/doc_train/edurigo_ai/TNA/testing/online_course.pyr s   r c@sFeZdZUdZeed<eed<eed<eeed<eed<e ed<dS) OnlineCourseResponsez4Data class for online course recommendation responsedomain field_prompttotal_recommendationscourse_recommendationssearch_summarysuccessN) rrrrrrrrr boolrrrrr s   r c @seZdZdZddefddZddeded ed efd d Zdeded ed e e fd dZ ded e e fddZ de d efddZd e e fddZded efddZdS) OnlineCourseRecommendationSystemz-Online Course Recommendation System using LLM gemma3:12b model_namec CsTzt|dddd|_td|WdSty)}z td|d}~ww)z2Initialize the online course recommendation systemig?i)modelnum_ctx temperature num_predictzr=r5rrrr;bs&< zAOnlineCourseRecommendationSystem._generate_course_recommendationsresponse_contentc Cs<zcddl}|d||j}|r_t|}g}|D]8}||rRt|d|d|d|d|d|d |d d |d |d |dt |d|dd }| |q|j dddd|WS| WStj y}ztd|| WYd}~Sd}~wty}ztd|gWYd}~Sd}~ww)z4Parse LLM response to extract course recommendationsrNz\[.*\]r r r rrrrrrrrr r r r rrrrrrrrrcSs|jS)N)r)xrrrszIOnlineCourseRecommendationSystem._parse_course_response..T)keyreversezJSON parsing failed: zError parsing course response: )researchDOTALLjsonloadsgroup_validate_course_itemr getrappendsort _create_fallback_recommendationsJSONDecodeErrorr0warningr2r3) r4rIrP json_matchparsed_responser=itemrecommendationr5rrrrFsH     z7OnlineCourseRecommendationSystem._parse_course_responser_c Csgd}|D]}||vrtd|dSqzt|d}|dks'|dkr2td|WdSWnttfyJtd|dYdSwt|d ts_td |d dSd S) z3Validate that a course item has all required fields) r r r rrrrrrrrzMissing required field: FrFdzInvalid match percentage: z!Invalid match percentage format: rzskills_covered must be a list: T)r0r\r ValueError TypeErrorrW isinstancelist)r4r_required_fieldsfield percentagerrrrVs( z6OnlineCourseRecommendationSystem._validate_course_itemcCs$tdddddddddd gd d d gS) z6Create fallback recommendations when LLM parsing failsz3Please check platform directly for relevant courseszMultiple PlatformszVarious InstructorsVariesz All LevelszpCourse recommendations could not be generated. Please search directly on platforms like Coursera, Udemy, or edX.rJzN/Azgeneral skillsKz0Fallback recommendation due to system limitationrK)r )r4rrrrZszAOnlineCourseRecommendationSystem._create_fallback_recommendations json_datac CsHz^t|}|dd}|dd}|dd}|r|s#ddgdWS||||}g}|jD] }||j|j|j|j |j |j |j |j |j|j|j|jd q/|j|j|j|j||jd WStjy} ztd | dd | gdWYd } ~ Sd } ~ wty} ztd| dt| gdWYd } ~ Sd } ~ ww)z Process JSON request and return JSON response Expected input format: { "domain": "Technology", "field_prompt": "Python Programming", "max_courses": 10 } r!rJr"r8r7Fz)Both domain and field_prompt are required)r&r3r$rK)r&r!r"r#r$r%zJSON decode error: zInvalid JSON format: NzProcessing error: )rSrTrWr?r$rXr r r rrrrrrrrrr&r!r"r#r%r[r0r3r2r) r4rl request_datar!r"r8r>course_recommendations_jsonrecr5rrrprocess_json_requestsb       z5OnlineCourseRecommendationSystem.process_json_requestN)r))r7)rrrrrr6rr r?rr r;rFdictr'rVrZrrprrrrr('s*R.r(cCsddddddddddd dd d d dg}td t}|D]}tddtd|dd|dtd|t|}td|dtdtd|dtd|dtd|dtdt|ddD]s\}}td|d|d td!|d"td#|d$td%|d&td'|d(td)|d*td+|d,td-|d.d/td0d1|d2td3|d4td5|d6d7d8d9qv|dstd:|dtd;|d<qd7S)=z,Test the online course recommendation system TechnologyzPython Programmingr@BusinesszDigital Marketingz!Data Science and Machine LearningCreativez UI/UX Designz3=== TESTING ONLINE COURSE RECOMMENDATION SYSTEM === zP================================================================================z TESTING: r!r:r"u ✅ SUCCESS: r&u 📊 STATS:z Total recommendations: r#z Domain: z Field: u 🎓 COURSE RECOMMENDATIONS:r$z u. 📚 r z Platform: r z Instructor: r z Duration: rz Level: rz Price: rz Rating: rz Match: r%z Skills: z, rz Reason: rz Description: rNrbz...z No recommendations found for u 📝 Summary: r%)printr(rprSdumps enumeratejoin) test_cases course_system test_caseresulticourserrr"test_online_course_recommendations]s\ r__main__)rSloggingtypingrrr dataclassesrlangchain_ollamarlangchain.promptsrlangchain.schemar basicConfigINFO getLoggerrr0r r r(rrrrrs(      8@