Update main.py

main.py

@@ -215,6 +215,38 @@ def get_education_profiles():
     education_profiles = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
     return jsonify({"ans": education_profiles})
 
+
+@app.route('/get_certificate', methods=['POST'])
+def get_certificate():
+    temperature = 0.9
+    max_new_tokens = 256
+    top_p = 0.95
+    repetition_penalty = 1.0
+
+    content = request.json
+    # user_degree = content.get('degree')  # Uncomment this line
+    user_stream = content.get('stream')
+
+    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 for certification recommendation for a student. Below are current details.
+    Stream: {user_stream}
+    Based on current details recommend the certification 
+    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)