Update main.py

main.py

@@ -53,6 +53,7 @@ def recommend():
     Based on above details recommend the courses that relate to the above details 
     Note: Output should be valid json format in below format:
     {{"course1:course_name, course2:course_name, course3:course_name,...}}
+    Return only answer not prompt and unnecessary stuff
     """
     formatted_prompt = format_prompt(prompt)
 
@@ -95,6 +96,7 @@ 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:mentor_name,mentor2:mentor_name,mentor3:mentor_name,...}}
+    Return only answer not prompt and unnecessary stuff
     """
     formatted_prompt = format_prompt(prompt)