Update main.py

main.py

@@ -9,6 +9,10 @@ llm = AutoModelForCausalLM.from_pretrained("TheBloke/Mistral-7B-v0.1-GGUF", mode
 
 app = Flask(__name__)
 
+file_path = "mentor.txt"
+with open(file_path, "r") as file:
+    mentors_data = file.read()
+
 @app.route('/')
 def home():
     return jsonify({"message": "Welcome to the Recommendation API!"})
@@ -44,14 +48,16 @@ def recommendation():
     user_degree = content.get('degree')
     user_stream = content.get('stream')
     user_semester = content.get('semester')
-    prompt = """
+    courses = content.get('courses')
+    prompt = f"""
     You need to act like as recommendataion engine for mentor recommendation for student based on below details also the list of mentors with their experience is attached.
 
     Degree: {user_degree}
     Stream: {user_stream}
     Current Semester: {user_semester}
+    courses opted:{courses}
 
-
+    Mentor list= {mentors_data}
     Based on above details recommend the mentor that realtes to above details 
     Note: Output should bevalid json format in below format:
     {{"mentor1:ABC,mentor2:DEF,mentor3:XYZ,...}}