Update main.py

main.py

@@ -20,7 +20,7 @@ def recommendation():
     user_stream = content.get('stream')
     user_semester = content.get('semester')
     prompt = """
-    You need to act like as recommendataion engine for course recommendation based on below details.
+    You need to act like as recommendataion engine for course recommendation for student based on below details.
 
     Degree: {user_degree}
     Stream: {user_stream}
@@ -37,5 +37,30 @@ def recommendation():
     prompt = f"{prefix}{user.replace('{prompt}', prompt)}{suffix}"
     return jsonify({"ans":llm(prompt)})
 
+
+@app.route('/get_mentor', methods=['POST'])
+def recommendation():
+    content = request.json
+    user_degree = content.get('degree')
+    user_stream = content.get('stream')
+    user_semester = content.get('semester')
+    prompt = """
+    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}
+
+
+    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,...}}
+    
+    """
+    suffix="[/INST]"
+    prefix="[INST] <<SYS>> You are a helpful assistant <</SYS>>"
+    prompt = f"{prefix}{user.replace('{prompt}', prompt)}{suffix}"
+    return jsonify({"ans":llm(prompt)})
+
 if __name__ == '__main__':
     app.run(debug=True)