o KiC@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__rintrr7/var/www/eduai.edurigo.com/tna/staging/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 @seZdZdZddefddZddZdd ed ed ed efd dZ d ed ed ed e e fddZ ded e e fddZ ded efddZd e e fddZded efddZdS) OnlineCourseRecommendationSystemz-Online Course Recommendation System using LLM gemma3:12b model_namec CsZzt|dddd|_d|_td|WdSty,}z td|d}~ww) z2Initialize the online course recommendation systemig?i)modelnum_ctx temperature num_predictrzrJ_parse_course_responser1r2r9r3r4) r5r!r"r@prompt_templateformatted_promptrErDr6rrrrCjs(< zAOnlineCourseRecommendationSystem._generate_course_recommendationsr;c 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_recommendationsJSONDecodeErrorr1warningr3r4) r5r;rV json_matchparsed_responserDitemrecommendationr6rrrrMsH     z7OnlineCourseRecommendationSystem._parse_course_responserec 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)r1rbr ValueError TypeErrorr] isinstancelist)r5rerequired_fieldsfield percentagerrrr\s( 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.rPzN/Azgeneral skillsKz0Fallback recommendation due to system limitationrQ)r )r5rrrr` szAOnlineCourseRecommendationSystem._create_fallback_recommendations json_datac CsRzcd|_t|}|dd}|dd}|dd}|r|s&ddgd WS||||}g}|jD] }||j|j|j |j |j |j |j |j|j|j|j|jd q2|j|j|j|j||j|jd WStjy} ztd | dd | gd WYd} ~ Sd} ~ wty} ztd| dt| gd WYd} ~ Sd} ~ ww)z Process JSON request and return JSON response Expected input format: { "domain": "Technology", "field_prompt": "Python Programming", "max_courses": 10 } rr!rPr"r@r?Fz)Both domain and field_prompt are required)r&r4r$rQ)r&r!r"r#r$r% token_countzJSON decode error: zInvalid JSON format: NzProcessing error: )r0rYrZr]rFr$r^r r r rrrrrrrrrr&r!r"r#r%rar1r4r3r) r5rr request_datar!r"r@rEcourse_recommendations_jsonrecr6rrrprocess_json_requestsf        z5OnlineCourseRecommendationSystem.process_json_requestN)r))r?)rrrrrr7r>rr rFrr rCrMdictr'r\r`rrwrrrrr('s*S.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 ProgrammingrGBusinesszDigital Marketingz!Data Science and Machine LearningCreativez UI/UX Designz3=== TESTING ONLINE COURSE RECOMMENDATION SYSTEM === zP================================================================================z TESTING: r!rBr"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: rNrhz...z No recommendations found for u 📝 Summary: r%)printr(rwrYdumps enumeratejoin) test_cases course_system test_caseresulticourserrr"test_online_course_recommendationsjs\ r__main__)rYloggingtypingrrr dataclassesrlangchain_ollamarlangchain_core.promptsrlangchain_core.messagesr basicConfigINFO getLoggerrr1r r r(rrrrrs(      E@