Update main.py

main.py

@@ -30,7 +30,7 @@ def format_prompt(message):
     return f"<s>[INST] {user_prompt} [/INST] {bot_response}</s> [INST] {message} [/INST]"
 
 
-@app.route('/ai_mentor', methods=['POST'])
+@app.route('/ai_mentor', methods=['GET'])
 def ai_mentor():
     data = request.get_json()
     message = data.get('message')
@@ -68,7 +68,7 @@ def ai_mentor():
         return jsonify({"message": f"Failed to process request: {str(e)}"}), 500
 
 
-@app.route('/get_course', methods=['POST'])
+@app.route('/get_course', methods=['GET'])
 def get_course():
     temperature = 0.9
     max_new_tokens = 256
@@ -104,7 +104,7 @@ def get_course():
     return jsonify({"ans": stream})
 
 
-@app.route('/get_mentor', methods=['POST'])
+@app.route('/get_mentor', methods=['GET'])
 def get_mentor():
     temperature = 0.9
     max_new_tokens = 256
@@ -159,6 +159,40 @@ def get_mentor():
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
     return jsonify({"ans": stream})
 
+@app.route('/get_streams', methods=['GET'])
+def get_streams():
+    temperature = 0.9
+    max_new_tokens = 256
+    top_p = 0.95
+    repetition_penalty = 1.0
+
+
+    content = request.json
+    user_degree = content.get('degree')
+    user_stream = content.get('stream')
+    #user_semester = content.get('semester')
+
+    generate_kwargs = dict(
+        temperature=temperature,
+        max_new_tokens=max_new_tokens,
+        top_p=top_p,
+        repetition_penalty=repetition_penalty,
+        do_sample=True,
+        seed=42,
+    )
+    prompt = f""" prompt: 
+    You need to act like as recommendation engine.
+    List all 40+ streams/branches in below degree 
+    Degree: {user_degree}
+    Note: Output should be list in below format:
+    [course1, course2, course3,...]
+    Return only answer not prompt and unnecessary stuff, also dont add any special characters or punctuation marks
+    """
+    formatted_prompt = format_prompt(prompt)
 
+    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
+    return jsonify({"ans": stream})
+
+    
 if __name__ == '__main__':
     app.run(debug=True)