Update main.py

main.py

@@ -23,7 +23,7 @@ def recommendation():
     user_degree = content.get('degree')
     user_stream = content.get('stream')
     user_semester = content.get('semester')
-    prompt = """
+    prompt = """ prompt: 
     You need to act like as recommendataion engine for course recommendation for student based on below details.
 
     Degree: {user_degree}
@@ -38,7 +38,7 @@ def recommendation():
     """
     suffix="[/INST]"
     prefix="[INST] <<SYS>> You are a helpful assistant <</SYS>>"
-    prompt = f"{prefix}{user.replace('{prompt}', prompt)}{suffix}"
+    prompt = f"{prefix}{prompt}{suffix}"
     return jsonify({"ans":llm(prompt)})
 
 
@@ -49,7 +49,7 @@ def mentor():
     user_stream = content.get('stream')
     user_semester = content.get('semester')
     courses = content.get('courses')
-    prompt = f"""
+    prompt = f""" 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}
@@ -61,11 +61,10 @@ def mentor():
     Based on above details recommend the mentor that realtes to above details 
     Note: Output should be valid 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}"
+    prompt = f"{prefix}{prompt}{suffix}"
     return jsonify({"ans":llm(prompt)})
 
 if __name__ == '__main__':