Update main.py

main.py

@@ -196,18 +196,19 @@ def get_streams():
 
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
     return jsonify({"ans": stream})
+    
 
-
-@app.route('/get_streams_degree', methods=['POST'])
-def get_streams():
-    data = request.get_json()
-    degree = data.get('degree')
-
+@app.route('/get_stream_degree', 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('degree')
+
     generate_kwargs = dict(
         temperature=temperature,
         max_new_tokens=max_new_tokens,
@@ -217,11 +218,11 @@ def get_streams():
         seed=42,
     )
     prompt = f""" prompt: 
-    You need to act like as recommendation engine.
-    List 5 streams/branches based on the degree provided.
-    For example, if the degree is 'bachelors', list 5 streams related to bachelor's degree.
+    You need to act like as recommendation engine for streams/branches recommendation for a student. Below are current details.
+    Stream: {user_stream}
+    Based on current details recommend the streams/branches 
     Note: Output should be list in below format:
-    [branch1, branch2, branch3,...]
+    [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)
@@ -231,6 +232,7 @@ def get_streams():
 
 
 
+
 @app.route('/get_education_profiles', methods=['GET'])
 def get_education_profiles():
     temperature = 0.9