adzee17 commited on
Commit
d3cdf67
·
verified ·
1 Parent(s): 27dbfd3

Create utils.py

Browse files
Files changed (1) hide show
  1. utils.py +32 -0
utils.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import json
2
+ from transformers import pipeline
3
+
4
+ # Load local course data
5
+ def load_courses():
6
+ with open("data/courses.json") as f:
7
+ return json.load(f)
8
+
9
+ # Use LLM to recommend best course
10
+ def recommend_course(user_input, courses):
11
+ relevant = [c for c in courses if user_input.lower() in c["category"].lower() or user_input.lower() in c["title"].lower()]
12
+
13
+ if not relevant:
14
+ return "No relevant courses found."
15
+
16
+ sorted_courses = sorted(relevant, key=lambda x: x["rating"], reverse=True)
17
+
18
+ # Optional: Use LLM to summarize reviews
19
+ summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
20
+ summaries = []
21
+ for course in sorted_courses[:3]:
22
+ review_text = " ".join(course["reviews"])
23
+ summary = summarizer(review_text, max_length=60, min_length=10, do_sample=False)[0]["summary_text"]
24
+ summaries.append((course, summary))
25
+
26
+ result = ""
27
+ for course, summary in summaries:
28
+ result += f"**{course['title']}** ({course['platform']})\n"
29
+ result += f"⭐ Rating: {course['rating']}\n"
30
+ result += f"📝 Summary of reviews: {summary}\n\n"
31
+
32
+ return result